mirror of
https://github.com/harry0703/MoneyPrinterTurbo.git
synced 2026-02-22 00:47:22 +08:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
# State Management
|
|
# This module is responsible for managing the state of the application.
|
|
import math
|
|
|
|
# 如果你部署在分布式环境中，你可能需要一个中心化的状态管理服务，比如 Redis 或者数据库。
|
|
# 如果你的应用程序是单机的，你可以使用内存来存储状态。
|
|
|
|
# If you are deploying in a distributed environment, you might need a centralized state management service like Redis or a database.
|
|
# If your application is single-node, you can use memory to store the state.
|
|
|
|
from app.models import const
|
|
from app.utils import utils
|
|
|
|
_tasks = {}
|
|
|
|
|
|
def update_task(task_id: str, state: int = const.TASK_STATE_PROCESSING, progress: int = 0, **kwargs):
|
|
"""
|
|
Set the state of the task.
|
|
"""
|
|
progress = int(progress)
|
|
if progress > 100:
|
|
progress = 100
|
|
|
|
_tasks[task_id] = {
|
|
"state": state,
|
|
"progress": progress,
|
|
**kwargs,
|
|
}
|
|
|
|
def get_task(task_id: str):
|
|
"""
|
|
Get the state of the task.
|
|
"""
|
|
return _tasks.get(task_id, None)
|