Adjusted files for isort & autopep

fix: add company id to invite user

creyPY/fastapi/db/__init__.py

@@ -1,3 +1,8 @@
-from .async_session import *  # noqa
-from .helpers import *  # noqa
-from .session import *  # noqa
+try:
+    import sqlalchemy
+
+    from .async_session import *
+    from .helpers import *
+    from .session import *
+except ImportError:
+    print("SQLAlchemy not installed. Database functionality will be disabled.")

creyPY/fastapi/testing_unit.py

@@ -140,6 +140,7 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
         pagination: bool = True,
         id_field: str = "id",
         created_at_check: bool = True,
+        patch: dict | None = None,
     ):
         # GET LIST
         re = await self.get(url)
@@ -164,6 +165,14 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
         re = await self.get(f"{url}{obj_id}/")
         self.assertEqual(re[id_field], obj_id)
 
+        # PATCH
+        if patch:
+            for key, value in patch.items():
+                input_obj[key] = value
+            re = await self.patch(f"{url}{obj_id}/", obj=input_obj)
+            for key, value in patch.items():
+                self.assertEqual(re[key], value)
+
         # GET LIST
         re = await self.get(url)
         if pagination:

creyPY/services/auth0/utils.py

@@ -101,7 +101,7 @@ def request_verification_mail(sub: str) -> None:
     return re.json()
 
 
-def create_user_invite(email: str) -> dict:
+def create_user_invite(email: str, company_id: str) -> dict:
     re = requests.post(
         f"https://{AUTH0_DOMAIN}/api/v2/users",
         headers={"Authorization": f"Bearer {get_management_token()}"},
@@ -111,6 +111,7 @@ def create_user_invite(email: str) -> dict:
             "password": create_random_password(),
             "verify_email": False,
             "app_metadata": {"invitedToMyApp": True},
+            "user_metadata": {"company_ids": [company_id]},
         },
         timeout=5,
     )