"""任务相关数据模型"""
from datetime import datetime

class Task:
    def __init__(self, beton_task_id, beton_volume, artifact_id, block_number):
        self.beton_task_id = beton_task_id
        self.beton_volume = beton_volume
        self.artifact_id = artifact_id
        self.block_number = block_number

class SendTask:
    def __init__(self, beton_task_id, beton_volume, artifact_id, block_number,
                 beton_grade, mix_id, timestamp=None):
        self.beton_task_id = beton_task_id
        self.beton_volume = beton_volume
        self.artifact_id = artifact_id
        self.block_number = block_number
        self.beton_grade = beton_grade
        self.mix_id = mix_id
        self.timestamp = timestamp or datetime.now()