2024-01-23 20:55:20 +08:00
|
|
|
from __future__ import annotations
|
|
|
|
|
|
|
|
|
|
import os
|
|
|
|
|
import shutil
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
required_files = {
|
2025-04-29 17:24:07 +08:00
|
|
|
'data/config.yaml': 'templates/config.yaml',
|
2024-01-23 20:55:20 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
required_paths = [
|
2025-04-29 17:24:07 +08:00
|
|
|
'temp',
|
|
|
|
|
'data',
|
|
|
|
|
'data/metadata',
|
|
|
|
|
'data/logs',
|
|
|
|
|
'data/labels',
|
2024-01-23 20:55:20 +08:00
|
|
|
]
|
|
|
|
|
|
2025-04-29 17:24:07 +08:00
|
|
|
|
2024-01-23 20:55:20 +08:00
|
|
|
async def generate_files() -> list[str]:
|
|
|
|
|
global required_files, required_paths
|
|
|
|
|
|
|
|
|
|
for required_paths in required_paths:
|
|
|
|
|
if not os.path.exists(required_paths):
|
|
|
|
|
os.mkdir(required_paths)
|
|
|
|
|
|
|
|
|
|
generated_files = []
|
|
|
|
|
for file in required_files:
|
|
|
|
|
if not os.path.exists(file):
|
|
|
|
|
shutil.copyfile(required_files[file], file)
|
|
|
|
|
generated_files.append(file)
|
|
|
|
|
|
|
|
|
|
return generated_files
|