Moonlit
commited on
Commit
·
4a90706
1
Parent(s):
182a723
Refactor: Use TaskStatus enum for task status handling (#646)
Browse files### What problem does this PR solve?
This commit changes the status 'not started' from being hard-coded to
being maintained by the TaskStatus enum. This enhancement ensures
consistency across the codebase and improves maintainability.
### Type of change
- [x] Refactoring
- api/apps/document_app.py +2 -1
- api/db/__init__.py +1 -0
api/apps/document_app.py
CHANGED
|
@@ -367,7 +367,8 @@ def change_parser():
|
|
| 367 |
return get_data_error_result(retmsg="Not supported yet!")
|
| 368 |
|
| 369 |
e = DocumentService.update_by_id(doc.id,
|
| 370 |
-
{"parser_id": req["parser_id"], "progress": 0, "progress_msg": "",
|
|
|
|
| 371 |
if not e:
|
| 372 |
return get_data_error_result(retmsg="Document not found!")
|
| 373 |
if "parser_config" in req:
|
|
|
|
| 367 |
return get_data_error_result(retmsg="Not supported yet!")
|
| 368 |
|
| 369 |
e = DocumentService.update_by_id(doc.id,
|
| 370 |
+
{"parser_id": req["parser_id"], "progress": 0, "progress_msg": "",
|
| 371 |
+
"run": TaskStatus.UNSTART.value})
|
| 372 |
if not e:
|
| 373 |
return get_data_error_result(retmsg="Document not found!")
|
| 374 |
if "parser_config" in req:
|
api/db/__init__.py
CHANGED
|
@@ -64,6 +64,7 @@ class ChatStyle(StrEnum):
|
|
| 64 |
|
| 65 |
|
| 66 |
class TaskStatus(StrEnum):
|
|
|
|
| 67 |
RUNNING = "1"
|
| 68 |
CANCEL = "2"
|
| 69 |
DONE = "3"
|
|
|
|
| 64 |
|
| 65 |
|
| 66 |
class TaskStatus(StrEnum):
|
| 67 |
+
UNSTART = "0"
|
| 68 |
RUNNING = "1"
|
| 69 |
CANCEL = "2"
|
| 70 |
DONE = "3"
|