Merge branch 'main' of gitlab.prodcontest.com:team-39/backend

src/template_project/adapters/data_gateways/tables.py

@@ -1,4 +1,5 @@
-from typing import Final
+import json
+from typing import Any, Final, override
 
 from pgvector.sqlalchemy import Vector
 from sqlalchemy import (
@@ -13,9 +14,11 @@ from sqlalchemy import (
     Numeric,
     String,
     Table,
+    TypeDecorator,
     UniqueConstraint,
+    text,
 )
-from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.dialects.postgresql import JSONB, UUID
 from sqlalchemy.orm import registry
 
 from template_project.application.access_token.entity import AccessToken
@@ -28,6 +31,37 @@ from template_project.application.user.profile.entity import Profile
 meta_data: Final = MetaData()
 mapper_registry: Final = registry()
 
+
+class StringArrayType(TypeDecorator[list[str]]):
+    impl: Any = JSONB
+    cache_ok: bool | None = True
+
+    @override
+    def process_bind_param(self, value: Any, dialect: Any) -> Any:
+        if value is None:
+            return []
+        if isinstance(value, list):
+            return value
+        return []
+
+    @override
+    def process_result_value(self, value: Any, dialect: Any) -> list[str]:
+        if value is None:
+            return []
+        if isinstance(value, list):
+            return [str(item) for item in value]
+        if isinstance(value, str):
+            try:
+                parsed = json.loads(value)
+                if isinstance(parsed, list):
+                    return [str(item) for item in parsed]
+            except (json.JSONDecodeError, TypeError):
+                pass
+        if isinstance(value, dict):
+            return []
+        return []
+
+
 user_table: Final = Table(
     "users",
     meta_data,
@@ -94,7 +128,7 @@ resume_table: Final = Table(
     Column("user_id", UUID, ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
     Column("position", String, nullable=False),
     Column("about_me", String, nullable=False),
-    Column("key_skills", ARRAY(String, as_tuple=True), nullable=False),
+    Column("key_skills", StringArrayType(), nullable=False, server_default=text("'[]'::jsonb")),
     Column("experience_type", String, nullable=False),
     Column("down_resume_id", UUID, ForeignKey("resume.id", ondelete="CASCADE"), nullable=True, default=None),
     Column("up_resume_id", UUID, ForeignKey("resume.id", ondelete="CASCADE"), nullable=True, default=None),
@@ -118,7 +152,7 @@ resume_prediction_table: Final = Table(
     Column("resume_id", UUID, ForeignKey("resume.id", ondelete="CASCADE"), nullable=False),
     Column("from_salary", Numeric, nullable=False),
     Column("to_salary", Numeric, nullable=False),
-    Column("recommended_skills", ARRAY(String, as_tuple=True), nullable=False),
+    Column("recommended_skills", StringArrayType(), nullable=False, server_default=text("'[]'::jsonb")),
 )
 key_skills_table: Final = Table(
     "key_skills",
@@ -133,6 +167,18 @@ mapper_registry.map_imperatively(AccessToken, access_token_table)
 mapper_registry.map_imperatively(AuthIdentity, auth_identity_table)
 mapper_registry.map_imperatively(Profile, profile_table)
 mapper_registry.map_imperatively(NotificationDevice, notification_device_table)
-mapper_registry.map_imperatively(Resume, resume_table)
+mapper_registry.map_imperatively(
+    Resume,
+    resume_table,
+    properties={
+        "key_skills": resume_table.c.key_skills,
+    },
+)
 mapper_registry.map_imperatively(ResumeEmbedding, resume_embedding_table)
-mapper_registry.map_imperatively(ResumePrediction, resume_prediction_table)
+mapper_registry.map_imperatively(
+    ResumePrediction,
+    resume_prediction_table,
+    properties={
+        "recommended_skills": resume_prediction_table.c.recommended_skills,
+    },
+)