add myoffice scripts

myoffice_projects/co_scripts/pgs_avatar_sync/modules/flask_model.py

@@ -0,0 +1,48 @@
+import base64
+import os
+import python_freeipa
+import requests
+import warnings
+
+warnings = warnings.filterwarnings("ignore")
+
+class PGSAvatarModule:
+    def __init__(self):
+        self.CO_API_URL      = os.getenv("COAPI_URL", "https://coapi.hyperus.team/api/v1/auth")
+        self.IPA_ADDRESS     = os.getenv("IPA_ADDRESS", "ipa01.hyperus.team")
+        self.IPA_LOGIN       = os.getenv("IPA_LOGIN", "automated.carbon")
+        self.IPA_PASSWORD    = os.getenv("IPA_PASSWORD")
+
+    def get_avatar_ipa(self, login) -> str:
+        ipa_user = {}
+        ipa_user_avatar = ""
+        ipa_client = python_freeipa.ClientMeta(host=self.IPA_ADDRESS, verify_ssl=False)
+        ipa_client.login(self.IPA_LOGIN, self.IPA_PASSWORD)
+        ipa_users = ipa_client.user_find(o_uid=login)
+        ipa_client.logout()
+        if ipa_users["count"] == 1:
+            ipa_user = ipa_users["result"][0]
+            if "jpegphoto" in ipa_user:
+                ipa_user_avatar = ipa_user["jpegphoto"][-1]["__base64__"]
+            else:
+                return "User doesnt have avatar photo", 406
+        else:
+            return "User not found", 404
+        return base64.b64decode(ipa_user_avatar), 200
+
+    def get_avatar_pgs(self, token) -> str:
+        header = {"X-co-auth-token": token}
+        request = requests.get(url=f"{self.CO_API_URL}/profile/picture", headers=header)
+        if request.status_code == 200:
+            return request.content, 200
+        elif request.status_code == 204:
+            return "Avatar not exist", 204
+        return "Bad response from COAPI", 400
+
+    def update_avatar(self, login, token, photo) -> list:
+        header = {"X-co-auth-token": token}
+        file = [("file", ("avatar.jpg", photo, "image/jpeg"))]
+        request = requests.post(url=f"{self.CO_API_URL}/profile/picture", headers=header, data={}, files=file)
+        if request.status_code == 200:
+            return f"Avatar has been updated for user <{login}>", 200
+        return "Something bad with update process", 401

myoffice_projects/co_scripts/pgs_avatar_sync/modules/flask_view.py

@@ -0,0 +1,30 @@
+from flask_restful import reqparse, Resource
+from .flask_model import PGSAvatarModule
+
+class PGSAvatarListener(Resource):
+  parser = reqparse.RequestParser()
+  def get(self):
+      return "Method not allowed", 405
+
+  def delete(self):
+      return "Method not allowed", 405
+
+  def post(self):
+        self.parser.add_argument("login")
+        self.parser.add_argument("token")
+        args = self.parser.parse_args()
+        if not args["login"] or not args["token"]:
+            return "Not correct query", 400
+        login = args["login"].split("@")[0]
+        token = args["token"]
+        avatar_module = PGSAvatarModule()
+        ipa_photo = avatar_module.get_avatar_ipa(login)
+        pgs_photo = avatar_module.get_avatar_pgs(token)
+        if 400 in ipa_photo:
+            return ipa_photo
+        if 400 in pgs_photo:
+            return pgs_photo
+        if ipa_photo != pgs_photo:
+            return avatar_module.update_avatar(login, token, ipa_photo[0])
+        else:
+            return f"Avatar photo of user <{login}> is actual", 208