63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from db.sql_db import DBTASK, DBRESUME, SqliteSqlalchemy
|
|
import uuid
|
|
from datetime import datetime
|
|
from decouple import config
|
|
import pathlib
|
|
from fastapi import File, UploadFile
|
|
from typing import List
|
|
import os
|
|
|
|
BASE_PATH = config('BASE_PATH', default='E://pyptoject//yj_resume//')
|
|
|
|
|
|
def check_and_create_directory(files):
|
|
# 先创建一个task
|
|
if not files or len(files) == 0:
|
|
return None
|
|
id = str(uuid.uuid4())
|
|
current_time = datetime.now()
|
|
# 格式化时间为字符串
|
|
formatted_time = current_time.strftime("%Y-%m-%d-%H-%M-%S")
|
|
task = DBTASK(id=id, create_time=datetime.now(), status=0, success_num=0, total_num=len(files),
|
|
fail_num=0,name=f"解析任务({formatted_time})")
|
|
|
|
session = SqliteSqlalchemy().session
|
|
try:
|
|
session.add(task)
|
|
session.commit()
|
|
except Exception as e:
|
|
print(f"Failed to save DBTASK info error {e}")
|
|
session.rollback()
|
|
return None
|
|
finally:
|
|
session.close()
|
|
return id
|
|
|
|
|
|
async def upload_and_save_file(dir_id, files: List[UploadFile]) -> (bool, str):
|
|
pathxx = pathlib.Path(BASE_PATH).joinpath(dir_id)
|
|
pathxx.mkdir(parents=True, exist_ok=True)
|
|
data = []
|
|
i = 0
|
|
for file in files:
|
|
name, fix = os.path.splitext(file.filename)
|
|
if fix not in ['.doc', '.docx']:
|
|
continue
|
|
i = i + 1
|
|
with open(pathxx.joinpath(str(i) + fix), 'wb') as f:
|
|
file_content = await file.read()
|
|
f.write(file_content)
|
|
data.append(DBRESUME(id=str(uuid.uuid4()), task_id=dir_id, status=0, file_name=str(i) + fix))
|
|
session = SqliteSqlalchemy().session
|
|
try:
|
|
session.bulk_save_objects(data)
|
|
session.commit()
|
|
except Exception as e:
|
|
print(f"Failed to save DBRESUME error {e}")
|
|
session.rollback()
|
|
return False, f"Failed to save DBRESUME error {e}"
|
|
finally:
|
|
session.close()
|
|
return True, "success"
|
|
|