Initial commit

utilities/openproject.py

@@ -0,0 +1,107 @@
+import json
+import requests
+from requests.auth import HTTPBasicAuth
+from datetime import datetime, timezone
+import time
+import logging
+
+
+class OpenProject:
+
+    def __init__(self):
+        # self.server_url = "http://localhost:8080"
+        self.server_url = "http://17.114.221.240:8080"
+        # self.api_key = "bb5897eb1daf9bdc4b400675de8e1e52bd64e1e8bce95b341a61a036431c850e"
+        self.api_key = "b902d975fcf6a29558e611e665145282acffa1e7109bfb462ef25266f7f9ed6e"
+
+    def create_shot(self, scene, shot, project, sequence=None):
+        url = self.server_url + "/api/v3/work_packages"
+
+        project_url = 1
+        attributes = {
+            "subject": "SC{}_{}".format(scene, shot),
+            "customField2": scene,
+            "customField1": shot,
+            "_links": {
+                "project": {"href": "/api/v3/projects/{}".format(project_url)},
+                "type": {"href": "/api/v3/types/1"}
+            }
+        }
+
+        return self._send_command(url, attributes)
+
+    def add_comment(self, work_project_id, comment, notify=False):
+
+        url = self.server_url + "/api/v3/work_packages/{}/activities?notify={}".format(str(work_project_id), str(notify))
+        attributes = {"comment": {"raw": comment}}
+
+        return self._send_command(url, attributes)
+
+    def get_work_package(self, identifier=None, attribute=None):
+        url = self.server_url + "/api/v3/work_packages/"
+        if identifier:
+            url = url + str(identifier)
+        return self._send_command(url, attribute)
+
+    def get_projects(self, identifier=None):
+        url = self.server_url + "/api/v3/projects/"
+        if identifier:
+            url = url + str(identifier)
+        return self._send_command(url, None)
+
+    def _send_command(self, url, body):
+
+        if body:
+            response = requests.post(url, json=body,
+                                     auth=HTTPBasicAuth('apikey', self.api_key))
+        else:
+            response = requests.get(url, auth=HTTPBasicAuth('apikey', self.api_key))
+
+        if not response.ok:
+            logging.error('Response error: {}'.format(response.reason))
+
+        return response.json()
+
+
+class OpenProjectWatcher:
+
+    def __init__(self, op_instance, interval=30):
+        self.op = OpenProject()
+        self.interval = interval
+        self.last_check = None
+
+    def _check_projects(self):
+        projects = self.op.get_projects()
+        for project in projects['_embedded']['elements']:
+            # last_update = datetime.datetime.fromisoformat(project['updatedAt'])
+            last_update = datetime.strptime(project['updatedAt'], "%Y-%m-%dT%H:%M:%S%z")
+            if not self.last_check or last_update > self.last_check:
+                logging.info("Update found for project: {}".format(project['name']))
+            #     todo: do something with updated info
+
+    def _check_work_projects(self):
+        packages = self.op.get_work_package()
+        for pkg in packages['_embedded']['elements']:
+            # print(pkg.keys())
+            last_update = datetime.strptime(pkg['updatedAt'], "%Y-%m-%dT%H:%M:%S%z")
+            if not self.last_check or last_update > self.last_check:
+                logging.info("Update found for shot: {}".format(pkg['subject']))
+            #     todo: do something with updated info
+
+    def watch(self):
+        while True:
+            now = datetime.now(timezone.utc)
+            self._check_projects()
+            self._check_work_projects()
+            self.last_check = now
+            time.sleep(self.interval)
+
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.INFO)
+
+    op = OpenProject()
+    op.add_comment(42, "After Effects Render completed successfully. Log available here.", True)
+    # print(op.get_projects())
+    watcher = OpenProjectWatcher(OpenProject())
+    watcher.watch()