DocStringにAnnoFab API DocumentのURLを記載する (#189)

* [genearted_api] API DocumentのURLを記載する

* versionup

* 予定稼働時間取得用のメソッドを追加

* update poetry

Author: yuji38kwmt

annofabapi/wrapper.py

@@ -1442,3 +1442,93 @@ def get_latest_job() -> Optional[JobInfo]:
                 else:
                     logger.debug("job_id = %s のジョブに %d 回アクセスしましたが、完了しませんでした。", job["job_id"], job_access_count)
                     return False
+
+    #########################################
+    # Public Method : Labor Control
+    #########################################
+    @staticmethod
+    def _get_actual_worktime_hour_from_labor(labor: Dict[str, Any]) -> Optional[float]:
+        working_time_by_user = labor["values"]["working_time_by_user"]
+        if working_time_by_user is None:
+            return None
+
+        actual_worktime = working_time_by_user.get("results")
+        if actual_worktime is None:
+            return None
+        else:
+            return actual_worktime / 3600 / 1000
+
+    @staticmethod
+    def _get_plan_worktime_hour_from_labor(labor: Dict[str, Any]) -> Optional[float]:
+        working_time_by_user = labor["values"]["working_time_by_user"]
+        if working_time_by_user is None:
+            return None
+
+        actual_worktime = working_time_by_user.get("plans")
+        if actual_worktime is None:
+            return None
+        else:
+            return actual_worktime / 3600 / 1000
+
+    def get_labor_control_worktime(
+        self,
+        organization_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        account_id: Optional[str] = None,
+        from_date: Optional[str] = None,
+        to_date: Optional[str] = None,
+    ) -> List[Dict[str, Any]]:
+        """
+        実績作業時間(actual_worktime)と予定作業時間(plan_worktime)を取得する。
+
+        Args:
+            query_params:
+
+        Returns:
+
+        """
+
+        def _to_new_data(labor: Dict[str, Any]) -> Dict[str, Any]:
+            labor["actual_worktime"] = self._get_actual_worktime_hour_from_labor(labor)
+            labor["plan_worktime"] = self._get_plan_worktime_hour_from_labor(labor)
+            labor.pop("values", None)
+            return labor
+
+        query_params = {
+            "organization_id": organization_id,
+            "project_id": project_id,
+            "account_id": account_id,
+            "from_date": from_date,
+            "to_date": to_date,
+        }
+        labor_list, _ = self.api.get_labor_control(query_params)
+        return [_to_new_data(e) for e in labor_list]
+
+    def get_labor_control_availability(
+        self, account_id: str = None, from_date: Optional[str] = None, to_date: Optional[str] = None
+    ) -> List[Dict[str, Any]]:
+        """
+        労務管理の予定稼働時間を取得する。
+
+        Args:
+            account_id:
+            from_date:
+            to_date:
+
+        Returns:
+            予定稼働時間情報
+        """
+
+        def _to_new_data(labor: Dict[str, Any]) -> Dict[str, Any]:
+            labor["availability"] = self._get_plan_worktime_hour_from_labor(labor)
+            labor.pop("values", None)
+            return labor
+
+        query_params = {
+            "organization_id": "___plannedWorktime___",
+            "account_id": account_id,
+            "from_date": from_date,
+            "to_date": to_date,
+        }
+        labor_list, _ = self.api.get_labor_control(query_params)
+        return [_to_new_data(e) for e in labor_list]