first commit

2026-07-13 15:37:05 +08:00
commit 8b11aa2efc
299 changed files with 206339 additions and 0 deletions
--- a/code/system/apis/vdes/chat/contacts.py
+++ b/code/system/apis/vdes/chat/contacts.py
@@ -0,0 +1,102 @@
+from system.models import Contacts, HistorySession
+from ninja import Router, Schema, Field
+
+router = Router()
+
+class addContactSchema(Schema):
+    MMSI: str = Field(None, alias="mmsi")
+    name: str = Field(None, alias="name")
+    phone: str = Field(None, alias="phone")
+    region: str = Field(None, alias="region")
+
+class deleteContactSchema(Schema):
+    MMSI: str = Field(None, alias="mmsi")
+
+class editContactSchema(Schema):
+    oldMMSI: str = Field(None, alias="old_mmsi")
+    newMMSI: str = Field(None, alias="new_mmsi")
+    name: str = Field(None, alias="name")
+    phone: str = Field(None, alias="phone")
+    region: str = Field(None, alias="region")
+
+
+@router.get('/contactsList')
+def get_contacts_list(request):
+    try:
+        contacts = Contacts.objects.all().values()
+        return {
+            "code": 200,
+            "message": "成功",
+            "data": list(contacts)
+        }
+    except Exception as e:
+        return {'code': 500, 'message': str(e)}
+        
+@router.post('/deleteContact')
+def delete_contact(request, data: deleteContactSchema):
+    try:
+        if not data.MMSI:
+            return {'code': 400, 'message': 'MMSI不能为空'} 
+        contact = Contacts.objects.filter(MMSI=data.MMSI).first()
+        if not contact:
+            return {'code': 404, 'message': '未找到对应联系人'}
+        # 删除关联的历史会话记录
+        HistorySession.objects.filter(FormId=contact.Id).delete()
+        contact.delete()
+        return {'code': 200, 'message': '删除成功'}
+    except Exception as e:
+        return {'code': 500, 'message': str(e)}
+
+
+@router.post('/editContact')
+def edit_contact(request, data: editContactSchema):
+    try:
+        if not data.oldMMSI:
+            return {'code': 400, 'message': '原MMSI不能为空'}
+            
+        contact = Contacts.objects.filter(MMSI=data.oldMMSI).first()
+        if not contact:
+            return {'code': 404, 'message': '未找到对应联系人'}
+        # 更新联系人信息    
+        contact.MMSI = data.newMMSI if data.newMMSI else contact.MMSI
+        contact.Name = data.name if data.name else contact.Name
+        contact.Mobile = data.phone if data.phone else contact.Mobile
+        contact.Region = data.region if data.region else contact.Region
+        contact.save()
+
+        # 更新关联的历史会话记录
+        if data.newMMSI or data.name:
+            HistorySession.objects.filter(FormId=contact.Id).update(
+                MMSI=data.newMMSI if data.newMMSI else contact.MMSI,
+                Name=data.name if data.name else contact.Name
+            )
+        
+        return {
+            'code': 200,
+            'message': '修改成功',
+            'data': {
+                'id': contact.Id,
+                'MMSI': contact.MMSI
+            }
+        }
+    except Exception as e:
+        return {'code': 500, 'message': str(e)}
+
+@router.post('/addContact')
+def add_contact(request, data: addContactSchema):
+    try:
+        if not data.MMSI:
+            return {'code': 400, 'message': 'MMSI不能为空'}
+        name = data.name if data.name else data.MMSI
+        contact = Contacts.objects.create(
+            MMSI=data.MMSI,
+            Name=name,
+            Mobile=data.phone,
+            Region=data.region
+        )
+        return {
+            'code': 200,
+            'message': '成功',
+        }
+    except Exception as e:
+        return {'code': 500, 'message': str(e)}