fix: added name for primary key constraint

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>

.gitignore

@@ -158,3 +158,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+.DS_*

creyPY/const/__init__.py

@@ -1,3 +1,4 @@
+from .fields import *  # noqa
 from .groups import *  # noqa
 from .i18n import *  # noqa
 from .stripe import *  # noqa

creyPY/const/fields.py

@@ -0,0 +1,17 @@
+from sqlalchemy import types
+
+
+class LowerCaseString(types.TypeDecorator):
+    """Converts strings to lower case on the way in."""
+
+    impl = types.String
+    cache_ok = True
+
+    def process_bind_param(self, value, dialect):
+        if value is None:
+            return value
+        return value.lower()
+
+    @property
+    def python_type(self):
+        return str

creyPY/fastapi/crud.py

@@ -67,7 +67,6 @@ def get_object_or_404(
             row = result.scalar_one_or_none()
             if row is None:
                 raise HTTPException(status_code=404, detail="The object does not exist.")
-
             obj_dict = row
         if expunge:
             await db.expunge(obj_dict)

creyPY/fastapi/models/base.py

@@ -1,7 +1,7 @@
 import uuid
 from datetime import datetime
 
-from sqlalchemy import Column, DateTime, String
+from sqlalchemy import Column, DateTime, PrimaryKeyConstraint, String
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.ext.declarative import declared_attr
 from sqlalchemy.orm import as_declarative
@@ -23,6 +23,11 @@ class Base(AutoAnnotateMixin, AutoInitMixin):
 
     # TODO: Add automated foreign key resolution
 
+    # Add name to primary key constraint to ensure alembic can pick it up later
+    @declared_attr
+    def __table_args__(cls):
+        return (PrimaryKeyConstraint("id", name=f"pk_{cls.__tablename__}"),)
+
     # Generate __tablename__ automatically
     @declared_attr
     def __tablename__(cls) -> str:

creyPY/fastapi/schemas/__init__.py

@@ -1,2 +1,2 @@
 from .base import *  # noqa
-from .response_schema import * #noqa
+from .schema_optional import * #noqa

creyPY/fastapi/schemas/base.py

@@ -11,6 +11,6 @@ class BaseSchemaModelIN(BaseModel):
 
 
 class BaseSchemaModelOUT(BaseSchemaModelIN):
-    id: UUID | str
+    id: UUID
     created_at: datetime
     updated_at: datetime

creyPY/fastapi/schemas/response_schema.py

@@ -1,43 +0,0 @@
-from typing import List, Optional, Type
-
-from fastapi import Query
-from pydantic import BaseModel, create_model
-
-
-class ResponseModelDependency:
-    def __init__(self, model_class: Type[BaseModel]):
-        self.model_class = model_class
-
-    def __call__(self, response_fields: Optional[List[str]] = Query(None)) -> Type[BaseModel]:
-        def process_result(result, fields=None, async_session=False):
-            if not fields:
-                if async_session:
-                    return {k: v for k, v in result.__dict__.items() if not k.startswith("_")}
-                return result
-
-            if hasattr(result, "_fields"):
-                row_fields = result._fields
-                return dict(zip(row_fields, result))
-            elif isinstance(result, tuple):
-                return dict(zip(fields, result))
-            elif isinstance(result, dict):
-                return result
-            else:
-                return {field: getattr(result, field) for field in fields if hasattr(result, field)}
-
-        if not response_fields:
-            return self.model_class, None, process_result
-
-        all_annotations = {}
-        for cls in self.model_class.__mro__:
-            if hasattr(cls, "__annotations__"):
-                all_annotations.update(cls.__annotations__)
-
-        fields = {}
-        for field in response_fields:
-            if field in all_annotations:
-                fields[field] = (all_annotations[field], None)
-
-        dynamic_model = create_model(f"Dynamic{self.model_class.__name__}", **fields)
-
-        return dynamic_model, response_fields, process_result

creyPY/fastapi/schemas/schema_optional.py

@@ -0,0 +1,19 @@
+from typing import Optional, Type, Union, get_args, get_origin, get_type_hints
+
+from pydantic import BaseModel, create_model
+
+
+def optional_fields(cls: Type[BaseModel]) -> Type[BaseModel]:
+    fields = {}
+    for name, hint in get_type_hints(cls).items():
+        if name.startswith("_"):
+            continue
+
+        if get_origin(hint) is not Union or type(None) not in get_args(hint):
+            hint = Optional[hint]
+
+        fields[name] = (hint, None)
+
+    new_model = create_model(cls.__name__, __base__=cls, **fields)
+
+    return new_model

creyPY/services/__init__.py

@@ -1,2 +1,3 @@
 from .auth0 import *  # noqa
 from .stripe import *  # noqa
+from .aws import *  # noqa

creyPY/services/auth0/utils.py

@@ -145,3 +145,21 @@ def password_change_mail(email: str) -> bool:
     if re.status_code != 200:
         raise HTTPException(re.status_code, re.json())
     return True
+
+
+def user_password_change_ticket(user_id: str) -> str:
+    re = requests.post(
+        f"https://{AUTH0_DOMAIN}/api/v2/tickets/password-change",
+        headers={"Authorization": f"Bearer {get_management_token()}"},
+        json={
+            "user_id": user_id,
+            "client_id": AUTH0_CLIENT_ID,
+            "ttl_sec": 0,
+            "mark_email_as_verified": False,
+            "includeEmailInRedirect": False,
+        },
+        timeout=5,
+    )
+    if re.status_code != 201:
+        raise HTTPException(re.status_code, re.json())
+    return re.json()["ticket"]

creyPY/services/aws/__init__.py

@@ -0,0 +1 @@
+from .email import *  # noqa

creyPY/services/aws/email.py

@@ -0,0 +1,32 @@
+import os
+
+import boto3
+from botocore.exceptions import ClientError
+
+AWS_CLIENT_ID = os.getenv("AWS_CLIENT_ID")
+AWS_CLIENT_SECRET = os.getenv("AWS_CLIENT_SECRET")
+AWS_SENDER_EMAIL = os.getenv("AWS_SENDER_EMAIL")
+AWS_REGION = os.getenv("AWS_REGION", "eu-central-1")
+
+
+async def send_email_ses(recipient_email, subject, html_body):
+    ses_client = boto3.client(
+        "ses",
+        aws_access_key_id=AWS_CLIENT_ID,
+        aws_secret_access_key=AWS_CLIENT_SECRET,
+        region_name=AWS_REGION,
+    )
+    email_message = {
+        "Source": AWS_SENDER_EMAIL,
+        "Destination": {"ToAddresses": [recipient_email]},
+        "Message": {
+            "Subject": {"Data": subject, "Charset": "UTF-8"},
+            "Body": {"Html": {"Data": html_body, "Charset": "UTF-8"}},
+        },
+    }
+
+    try:
+        response = ses_client.send_email(**email_message)
+        return response["MessageId"]
+    except ClientError as e:
+        return None

requirements.stripe.txt

@@ -1 +1 @@
-stripe==11.6.0 # Stripe
+stripe==12.3.0 # Stripe