Ffile ttpew

app/api/services/file_upload.py

@@ -3,11 +3,12 @@ from flask import jsonify
 from app.api.enums import KafkaMessage
 from app.api.services import MyProductsService
 from app.config import settings
-from app.models import Members
+from app.models import Members, MembersWebfiles
 from app.utils.logger import logger
 from app.api.services.base_service import BaseService
 from sqlalchemy import func, desc
 from app.extensions import db
+from app.api.types.data_types import MemberFileData
 import os
 from werkzeug.utils import secure_filename
 import time
@@ -31,9 +32,10 @@ class FileUploadService(BaseService):
             # Who is this
             member_data = Members.get_member_by_uid(member_uid)
             if not member_data:
-                return []
+                return jsonify({"message": "User not found"}), 500
 
-            final_folder = f"W{member_data.id:010d}"
+            member_id=member_data.id
+            final_folder = f"W{member_id:010d}"
 
             save_path = FileUploadService.UPLOAD_FOLDER + "/WEBSITE"
             if not os.path.isdir(save_path):
@@ -52,6 +54,16 @@ class FileUploadService(BaseService):
             logger.info(f'POST CALLED 6 {final_save_path}')
 
             file.save(final_save_path)
+            save_file_props  = MemberFileData()
+            save_file_props.member_uid = member_uid
+            save_file_props.filename = filename
+            save_file_props.save_filename = save_filename
+            save_file_props.file_group="WEBSITE"
+            save_file_props.member_id = member_id
+            save_file_props.file_size = 0
+            save_file_props.file_size = filename.rsplit('.', 1)[1].lower()
+
+            save_result = MembersWebfiles.create_file_profile(save_file_props)
 
             response_data = {
                 "file_uid": file_uid,
@@ -62,3 +74,7 @@ class FileUploadService(BaseService):
         except Exception as e:
             logger.error(f"An error occurred while uploading file: {str(e)}", exc_info=True)
             return jsonify({"message": "Internal Server Error"}), 500
+
+    def allowed_file(filename):
+        return '.' in filename and \
+            filename.rsplit('.', 1)[1].lower() in FileUploadService.ALLOWED_EXTENSIONS

app/api/types/data_types.py

@@ -0,0 +1,8 @@
+class MemberFileData:
+    member_id: int
+    member_uid: str
+    file_group: str
+    filename: str
+    save_filename: str
+    file_type: str
+    file_size: str