feat:#183

This commit is contained in:
Lan
2024-09-24 23:09:13 +08:00
parent 6c2f2caee0
commit 2bef407373
+7 -1
View File
@@ -3,12 +3,13 @@
# @File : tasks.py
# @Software: PyCharm
import asyncio
import os
from tortoise.expressions import Q
from apps.base.models import FileCodes
from apps.base.utils import ip_limit
from core.settings import settings
from core.settings import settings, data_root
from core.storage import FileStorageInterface, storages
from core.utils import get_now
@@ -17,6 +18,11 @@ async def delete_expire_files():
file_storage: FileStorageInterface = storages[settings.file_storage]()
while True:
try:
# 遍历 share目录下的所有文件夹,删除空的文件夹,并判断父目录是否为空,如果为空也删除
if settings.file_storage == 'local':
for root, dirs, files in os.walk(f"{data_root}/share/data"):
if not dirs and not files:
os.rmdir(root)
await ip_limit['error'].remove_expired_ip()
await ip_limit['upload'].remove_expired_ip()
expire_data = await FileCodes.filter(Q(expired_at__lt=await get_now()) | Q(expired_count=0)).all()