删除ai临时文件

Controller/AiController.py

@@ -15,6 +15,7 @@ import base64
 import json
 import os
 import time
+import glob
 import urllib
 from urllib.parse import quote, parse_qs, unquote
 
@@ -678,6 +679,9 @@ class AiView(View):
                                              aiEndTime=ai_end_time, pushTime=apns_start_time,
                                              pushEndTime=apns_end_time, serverCountTime=count_time,picUploadTime=cover_start_time,
                                                                    picUploadEndTime = endTime, endTime=endTime)
+                #删除临时文件
+                delf = os.path.join(BASE_DIR, 'static/ai')
+                self.delfile(path=delf)
                 return JsonResponse(status=200, data='success', safe=False)
             return JsonResponse(status=500, data='fail', safe=False)
 
@@ -685,6 +689,25 @@ class AiView(View):
             print(e)
             return response.json(500, repr(e))
 
+    def delfile(self, path):
+
+        #   read all the files under the folder
+        fileNames = glob.glob(path + r'\*')
+
+        for fileName in fileNames:
+            try:
+                #           delete file
+                os.remove(fileName)
+            except:
+                try:
+                    #               delete empty folders
+                    os.rmdir(fileName)
+                except:
+                    #               Not empty, delete files under folders
+                    delfile(fileName)
+                    #               now, folders are empty, delete it
+                    os.rmdir(fileName)
+
 
     ## 检查是否有符合条件的标签
     def checkLabels(self, user_detect_group, labels):