Rename KYC models (Application/Profile) and add public schema with MongoDB

kyc_app/schemas/user_schema.py

@@ -1,13 +1,14 @@
 import graphene
 from graphene_django import DjangoObjectType
 from django.contrib.contenttypes.models import ContentType
-from ..models import KYCRequest, KYCRequestRussia
+from ..models import KYCApplication, KYCDetailsRussia
 from ..temporal import KycWorkflowClient
 
 
-class KYCRequestType(DjangoObjectType):
+class KYCApplicationType(DjangoObjectType):
+    """GraphQL type for KYC Application (заявка)."""
     class Meta:
-        model = KYCRequest
+        model = KYCApplication
         fields = '__all__'
 
     country_data = graphene.JSONString()
@@ -17,13 +18,15 @@ class KYCRequestType(DjangoObjectType):
         return self.get_country_data()
 
 
-class KYCRequestRussiaType(DjangoObjectType):
+class KYCDetailsRussiaType(DjangoObjectType):
+    """GraphQL type for Russia-specific KYC details."""
     class Meta:
-        model = KYCRequestRussia
+        model = KYCDetailsRussia
         fields = '__all__'
 
 
-class KYCRequestRussiaInput(graphene.InputObjectType):
+class KYCApplicationRussiaInput(graphene.InputObjectType):
+    """Input for creating KYC Application for Russia."""
     companyName = graphene.String(required=True)
     companyFullName = graphene.String(required=True)
     inn = graphene.String(required=True)
@@ -38,11 +41,12 @@ class KYCRequestRussiaInput(graphene.InputObjectType):
     contactPhone = graphene.String(required=True)
 
 
-class CreateKYCRequestRussia(graphene.Mutation):
+class CreateKYCApplicationRussia(graphene.Mutation):
+    """Create KYC Application for Russian company."""
     class Arguments:
-        input = KYCRequestRussiaInput(required=True)
+        input = KYCApplicationRussiaInput(required=True)
 
-    kyc_request = graphene.Field(KYCRequestType)
+    kyc_application = graphene.Field(KYCApplicationType)
     success = graphene.Boolean()
 
     def mutate(self, info, input):
@@ -52,7 +56,7 @@ class CreateKYCRequestRussia(graphene.Mutation):
             raise Exception("Not authenticated")
 
         # 1. Create Russia details
-        russia_details = KYCRequestRussia.objects.create(
+        russia_details = KYCDetailsRussia.objects.create(
             company_name=input.companyName,
             company_full_name=input.companyFullName,
             inn=input.inn,
@@ -64,49 +68,67 @@ class CreateKYCRequestRussia(graphene.Mutation):
             correspondent_account=input.correspondentAccount or '',
         )
 
-        # 2. Create main KYCRequest with reference to details
-        kyc_request = KYCRequest.objects.create(
+        # 2. Create main KYC Application with reference to details
+        kyc_application = KYCApplication.objects.create(
             user_id=user_id,
             team_name=input.companyName,
             country_code='RU',
             contact_person=input.contactPerson,
             contact_email=input.contactEmail,
             contact_phone=input.contactPhone,
-            content_type=ContentType.objects.get_for_model(KYCRequestRussia),
+            content_type=ContentType.objects.get_for_model(KYCDetailsRussia),
             object_id=russia_details.id,
         )
 
         # 3. Start Temporal workflow
-        KycWorkflowClient.start(kyc_request)
+        KycWorkflowClient.start(kyc_application)
 
-        return CreateKYCRequestRussia(kyc_request=kyc_request, success=True)
+        return CreateKYCApplicationRussia(kyc_application=kyc_application, success=True)
 
 
 class UserQuery(graphene.ObjectType):
     """User schema - ID token authentication"""
-    kyc_requests = graphene.List(KYCRequestType)
-    kyc_request = graphene.Field(KYCRequestType, uuid=graphene.String(required=True))
+    # Keep old names for backwards compatibility
+    kyc_requests = graphene.List(KYCApplicationType, description="Get user's KYC applications")
+    kyc_request = graphene.Field(KYCApplicationType, uuid=graphene.String(required=True))
+    # New names
+    kyc_applications = graphene.List(KYCApplicationType, description="Get user's KYC applications")
+    kyc_application = graphene.Field(KYCApplicationType, uuid=graphene.String(required=True))
 
     def resolve_kyc_requests(self, info):
-        # Filter by user_id from JWT token
+        return self._get_applications(info)
+
+    def resolve_kyc_applications(self, info):
+        return self._get_applications(info)
+
+    def _get_applications(self, info):
         user_id = getattr(info.context, 'user_id', None)
         if not user_id:
             return []
-        return KYCRequest.objects.filter(user_id=user_id)
+        return KYCApplication.objects.filter(user_id=user_id)
 
     def resolve_kyc_request(self, info, uuid):
+        return self._get_application(info, uuid)
+
+    def resolve_kyc_application(self, info, uuid):
+        return self._get_application(info, uuid)
+
+    def _get_application(self, info, uuid):
         user_id = getattr(info.context, 'user_id', None)
         if not user_id:
             return None
         try:
-            return KYCRequest.objects.get(uuid=uuid, user_id=user_id)
-        except KYCRequest.DoesNotExist:
+            return KYCApplication.objects.get(uuid=uuid, user_id=user_id)
+        except KYCApplication.DoesNotExist:
             return None
 
 
 class UserMutation(graphene.ObjectType):
     """User mutations - ID token authentication"""
-    create_kyc_request_russia = CreateKYCRequestRussia.Field()
+    # Keep old name for backwards compatibility
+    create_kyc_request_russia = CreateKYCApplicationRussia.Field()
+    # New name
+    create_kyc_application_russia = CreateKYCApplicationRussia.Field()
 
 
 user_schema = graphene.Schema(query=UserQuery, mutation=UserMutation)