feat(application-form-fill): 新增模板配置骨架
This commit is contained in:
55
review_agent/application_form_fill/storage.py
Normal file
55
review_agent/application_form_fill/storage.py
Normal file
@@ -0,0 +1,55 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import hashlib
|
||||
from pathlib import Path
|
||||
|
||||
from django.conf import settings
|
||||
|
||||
from review_agent.models import ApplicationFormFillArtifact, ApplicationFormFillBatch
|
||||
|
||||
|
||||
def build_batch_work_dir(batch: ApplicationFormFillBatch | None = None, *, batch_no: str = "") -> Path:
|
||||
if batch:
|
||||
return Path(settings.MEDIA_ROOT) / "application_form_fill" / str(batch.user_id) / str(batch.conversation_id) / batch.batch_no
|
||||
return Path(settings.MEDIA_ROOT) / "application_form_fill" / batch_no
|
||||
|
||||
|
||||
def compute_file_sha256(path: str | Path) -> str:
|
||||
file_path = Path(path)
|
||||
digest = hashlib.sha256()
|
||||
with file_path.open("rb") as handle:
|
||||
for chunk in iter(lambda: handle.read(1024 * 1024), b""):
|
||||
digest.update(chunk)
|
||||
return digest.hexdigest()
|
||||
|
||||
|
||||
def ensure_batch_subdir(batch: ApplicationFormFillBatch, name: str) -> Path:
|
||||
root = Path(batch.work_dir) if batch.work_dir else build_batch_work_dir(batch)
|
||||
target = root / Path(name).name
|
||||
target.mkdir(parents=True, exist_ok=True)
|
||||
return target
|
||||
|
||||
|
||||
def create_artifact_for_file(
|
||||
batch: ApplicationFormFillBatch,
|
||||
*,
|
||||
path: str | Path,
|
||||
artifact_type: str,
|
||||
file_format: str,
|
||||
name: str = "",
|
||||
metadata: dict | None = None,
|
||||
created_by_node: str = "",
|
||||
) -> ApplicationFormFillArtifact:
|
||||
file_path = Path(path)
|
||||
return ApplicationFormFillArtifact.objects.create(
|
||||
batch=batch,
|
||||
artifact_type=artifact_type,
|
||||
file_format=file_format,
|
||||
name=name or file_path.stem,
|
||||
file_name=file_path.name,
|
||||
storage_path=str(file_path),
|
||||
file_size=file_path.stat().st_size if file_path.exists() else 0,
|
||||
content_hash=compute_file_sha256(file_path) if file_path.exists() else "",
|
||||
metadata=metadata or {},
|
||||
created_by_node=created_by_node,
|
||||
)
|
||||
Reference in New Issue
Block a user