semaphoreui-client 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

semaphoreui_client/__about__.py

@@ -1 +1 @@
-__version__ = "0.1.2"
+__version__ = "0.1.3"

semaphoreui_client/client.py

@@ -7,8 +7,12 @@ import requests
 
 class SemaphoreUIClient:
 
-    def __init__(self, host, path="/api"):
+    def __init__(self, host: str, path: str="/api"):
         self.http = requests.Session()
+        if host.endswith("/"):
+            host = host.strip("/")
+        if not path.startswith("/"):
+            path = f"/{path}"
         self.api_endpoint = f"{host}{path}"
 
     def login(self, user, password):
@@ -186,11 +190,13 @@ class SemaphoreUIClient:
         ssh: typing.Optional[typing.Tuple[str, str, str]],
     ) -> "Key":
         if key_type not in ("ssh", "login_password"):
-            raise ValueError(f"Invalid key_type: {key_type}. Acceptable values are: ssh, login_password")
+            raise ValueError(
+                f"Invalid key_type: {key_type}. Acceptable values are: ssh, login_password"
+            )
         if key_type == "ssh":
             if ssh is None:
                 raise ValueError("ssh parameter must be set on key_type: ssh")
-            json_data =             json={
+            json_data = {
                 "id": 0,
                 "project_id": project_id,
                 "name": name,
@@ -208,7 +214,9 @@ class SemaphoreUIClient:
             }
         elif key_type == "login_password":
             if login_password is None:
-                raise ValueError("login_password parameter must be set on key_type: login_password")
+                raise ValueError(
+                    "login_password parameter must be set on key_type: login_password"
+                )
             json_data = {
                 "id": 0,
                 "project_id": project_id,
@@ -219,11 +227,10 @@ class SemaphoreUIClient:
                     "login": login_password[0],
                     "password": login_password[1],
                 },
-                "ssh": {"login": "", "passphrase": "", "private_key": ""}
+                "ssh": {"login": "", "passphrase": "", "private_key": ""},
             }
         response = self.http.post(
-            f"{self.api_endpoint}/project/{project_id}/keys",
-            json=json_data
+            f"{self.api_endpoint}/project/{project_id}/keys", json=json_data
         )
         assert response.status_code == 204
 
@@ -402,7 +409,7 @@ class SemaphoreUIClient:
         type: str,
         start_version: str,
         autorun: bool,
-        build_template_id: typing.Optional[int]=None,
+        build_template_id: typing.Optional[int] = None,
     ) -> "Template":
         response = self.http.post(
             f"{self.api_endpoint}/project/{project_id}/templates",
@@ -434,13 +441,69 @@ class SemaphoreUIClient:
             response.status_code == 201
         ), f"Expected response code 201, got {response.status_code}"
         return Template(**response.json(), client=self)
-    
+
     def delete_project_template(self, project_id: int, id: int):
         response = self.http.delete(
             f"{self.api_endpoint}/project/{project_id}/templates/{id}"
         )
         assert response.status_code == 204
 
+    def get_project_schedules(self, project_id: int) -> typing.List["Schedule"]:
+        response = self.http.get(f"{self.api_endpoint}/project/{project_id}/schedules")
+        assert response.status_code == 200
+        return [Schedule(**schedule, client=self) for schedule in response.json()]
+
+    def create_project_schedule(
+        self,
+        project_id: int,
+        template_id: int,
+        name: str,
+        cron_format: str,
+        active: bool = True,
+    ):
+        response = self.http.post(
+            f"{self.api_endpoint}/project/{project_id}/schedules",
+            json={
+                "id": 0,
+                "project_id": project_id,
+                "template_id": template_id,
+                "name": name,
+                "cron_format": cron_format,
+                "active": active,
+            },
+        )
+        assert response.status_code == 201
+        return Schedule(**response.json(), client=self)
+
+    def update_project_schedule(
+        self,
+        project_id: int,
+        schedule_id: int,
+        template_id: int,
+        name: str,
+        cron_format: str,
+        active: bool,
+    ):
+        response = self.http.post(
+            f"{self.api_endpoint}/project/{project_id}/schedules",
+            json={
+                "id": schedule_id,
+                "project_id": project_id,
+                "template_id": template_id,
+                "name": name,
+                "cron_format": cron_format,
+                "active": active,
+            },
+        )
+        assert response.status_code == 201
+        return Schedule(**response.json(), client=self)
+
+    def delete_project_schedule(self, project_id: int, schedule_id: int):
+        response = self.http.get(
+            f"{self.api_endpoint}/project/{project_id}/schedules/{schedule_id}"
+        )
+        assert response.status_code == 204
+
 
 @dataclass
 class Integration:
@@ -532,9 +595,9 @@ class Project:
         self,
         name: str,
         key_type: str,
-        override_secret: bool=False,
-        login_password: typing.Optional[typing.Tuple[str, str]]=None,
-        ssh: typing.Optional[typing.Tuple[str, str, str]]=None,
+        override_secret: bool = False,
+        login_password: typing.Optional[typing.Tuple[str, str]] = None,
+        ssh: typing.Optional[typing.Tuple[str, str, str]] = None,
     ):
         return self.client.create_project_key(
             self.id, name, key_type, override_secret, login_password, ssh
@@ -610,7 +673,7 @@ class Project:
         type: str,
         start_version: str,
         autorun: bool,
-        build_template_id: typing.Optional[int]=None,
+        build_template_id: typing.Optional[int] = None,
     ) -> "Template":
         return self.client.create_project_template(
             self.id,
@@ -635,6 +698,16 @@ class Project:
             build_template_id,
         )
 
+    def schedules(self) -> typing.List["Schedule"]:
+        return self.client.get_project_schedules(self.id)
+
+    def create_schedule(
+        self, template_id: int, name: str, cron_format: str, active: bool = True
+    ) -> "Schedule":
+        return self.client.create_project_schedule(
+            self.id, template_id, name, cron_format, active
+        )
+
 
 @dataclass
 class Permissions:
@@ -806,8 +879,32 @@ class Template:
     last_task: int
     tasks: int
 
-
     client: SemaphoreUIClient
 
     def delete(self):
         self.client.delete_project_template(self.project_id, self.id)
+
+
+@dataclass
+class Schedule:
+    id: int
+    cron_format: str
+    project_id: int
+    template_id: int
+    name: str
+    active: bool
+
+    client: SemaphoreUIClient
+
+    def save(self):
+        self.client.update_project_schedule(
+            self.project_id,
+            self.id,
+            self.template_id,
+            self.name,
+            self.cron_format,
+            self.active,
+        )
+
+    def delete(self):
+        self.client.delete_project_schedule(self.project_id, self.id)

{semaphoreui_client-0.1.2.dist-info → semaphoreui_client-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: semaphoreui-client
-Version: 0.1.2
+Version: 0.1.3
 Summary: An api client for interacting with Semaphore UI
 Project-URL: Documentation, https://github.com/rockstar/semaphoreui-client#readme
 Project-URL: Issues, https://github.com/rockstar/semaphoreui-client/issues

semaphoreui_client-0.1.3.dist-info/RECORD

@@ -0,0 +1,6 @@
+semaphoreui_client/__about__.py,sha256=R5TtpJu7Qu6sOarfDpp-5Oyy8Pi2Ir3VewCvsCQiAgo,21
+semaphoreui_client/__init__.py,sha256=r2j1tTgwFzp3AKAktr8izDIXllRf-Y0SdHAjInYDTP0,55
+semaphoreui_client/client.py,sha256=AvZ1v4_aMFzRDM-YKb7CQd6Od9EgoKty-YXrf51ecsA,27612
+semaphoreui_client-0.1.3.dist-info/METADATA,sha256=VW8KWzF_RjLzD1D7ooiLIb8kP8nI0MzTlWmPQ7wMoIw,1632
+semaphoreui_client-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+semaphoreui_client-0.1.3.dist-info/RECORD,,

semaphoreui_client-0.1.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-semaphoreui_client/__about__.py,sha256=K5SiDdEGYMpdqXThrqwTqECJJBOQNTQDrnpc2K5mzKs,21
-semaphoreui_client/__init__.py,sha256=r2j1tTgwFzp3AKAktr8izDIXllRf-Y0SdHAjInYDTP0,55
-semaphoreui_client/client.py,sha256=CBw2BUELSuwguRGt1hczE0yEUMTG63uHkWvh9VO4_4Y,24716
-semaphoreui_client-0.1.2.dist-info/METADATA,sha256=D7_PSIOOZichIN1JmhIMxuwiuv0HbV7z6wGoDsoeJj8,1632
-semaphoreui_client-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-semaphoreui_client-0.1.2.dist-info/RECORD,,