refactor: 定时清理数据库中没关联应用的图片

Author: liuruibin

apps/common/job/__init__.py

@@ -6,12 +6,10 @@
     @date：2024/3/14 11:54
     @desc:
 """
-from .clean_orphaned_file_job import *
 from .client_access_num_job import *
 from .clean_chat_job import *
 
 
 def run():
     client_access_num_job.run()
     clean_chat_job.run()
-    clean_orphaned_file_job.run()

apps/common/job/clean_chat_job.py

@@ -10,6 +10,7 @@
 from application.models import Application, Chat
 from django.db.models import Q
 from common.lock.impl.file_lock import FileLock
+from dataset.models import File
 
 scheduler = BackgroundScheduler()
 scheduler.add_jobstore(DjangoJobStore(), "default")
@@ -38,6 +39,8 @@ def clean_chat_log_job():
             if count == 0:
                 break
             deleted_count, _ = Chat.objects.filter(id__in=logs_to_delete).delete()
+            # 删除对应的文件
+            File.objects.filter(~Q(meta__chat_id__in=logs_to_delete)).delete()
             if deleted_count < batch_size:
                 break
 

apps/common/job/clean_orphaned_file_job.py

@@ -9,6 +9,8 @@
 import uuid
 
 from django.db import models
+from django.db.models.signals import pre_delete
+from django.dispatch import receiver
 
 from common.db.sql_execute import select_one
 from common.mixins.app_model_mixin import AppModelMixin
@@ -157,3 +159,9 @@ def save(
     def get_byte(self):
         result = select_one(f'SELECT lo_get({self.loid}) as "data"', [])
         return result['data']
+
+
+
+@receiver(pre_delete, sender=File)
+def on_delete_file(sender, instance, **kwargs):
+    select_one(f'SELECT lo_unlink({instance.loid})', [])