mirror of
https://github.com/xhongc/music-tag-web.git
synced 2026-05-11 10:41:08 +08:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import os
|
|
import time
|
|
import uuid
|
|
|
|
from django.conf import settings
|
|
|
|
from applications.music.models import Folder
|
|
from applications.subsonic.constants import AUDIO_EXTENSIONS_AND_MIMETYPE, COVER_TYPE
|
|
from applications.task.services.scan_utils import ScanMusic
|
|
from django_vue_cli.celery_app import app
|
|
|
|
|
|
def get_uuid():
|
|
return str(uuid.uuid4())
|
|
|
|
|
|
@app.task
|
|
def full_scan_folder():
|
|
music_folder = os.path.join(settings.MEDIA_ROOT, "music")
|
|
bulk_create = []
|
|
stack = [(None, music_folder)]
|
|
while len(stack) != 0:
|
|
# 从栈里取出数据
|
|
parent_uid, dir_data = stack.pop(0)
|
|
if os.path.isdir(dir_data):
|
|
sub_path = os.listdir(dir_data)
|
|
my_uuid = get_uuid()
|
|
sub_path = [(my_uuid, f"{dir_data}/{i}") for i in sub_path]
|
|
stack.extend(sub_path)
|
|
bulk_create.append(
|
|
Folder(**{
|
|
"name": dir_data.split("/")[-1],
|
|
"path": dir_data,
|
|
"file_type": "folder",
|
|
"uid": my_uuid,
|
|
"parent_id": parent_uid
|
|
})
|
|
)
|
|
else:
|
|
suffix = dir_data.split(".")[-1]
|
|
if suffix in dict(AUDIO_EXTENSIONS_AND_MIMETYPE):
|
|
print(dir_data)
|
|
my_uuid = get_uuid()
|
|
bulk_create.append(
|
|
Folder(**{
|
|
"name": dir_data.split("/")[-1],
|
|
"path": dir_data,
|
|
"file_type": "music",
|
|
"uid": my_uuid,
|
|
"parent_id": parent_uid
|
|
})
|
|
)
|
|
elif suffix in COVER_TYPE:
|
|
my_uuid = get_uuid()
|
|
bulk_create.append(
|
|
Folder(**{
|
|
"name": dir_data.split("/")[-1],
|
|
"path": dir_data,
|
|
"file_type": "image",
|
|
"uid": my_uuid,
|
|
"parent_id": parent_uid
|
|
})
|
|
)
|
|
Folder.objects.bulk_create(bulk_create, batch_size=500)
|
|
|
|
|
|
@app.task
|
|
def scan_music_id3():
|
|
a = time.time()
|
|
ScanMusic("/").scan()
|
|
print(time.time() - a)
|