PoseCorrection/backend_refactor/db/task.py

import time

from db import db


class Task(db.Model):
    uuid = db.Column(db.String(64), primary_key=True)
    type = db.Column(db.Enum('PHOTO_STANDING', 'VIDEO_HIGHKNEES',
                             'VIDEO_CRUNCH', 'VIDEO_STANDINGLONGJUMP', 'VIDEO_MULTIJUMP',
                             'VIDEO_SITANDREACH', 'VIDEO_RUNBACKANDFORTH',
                             'VIDEO_TENNISTHROW', 'VIDEO_BALANCEBEAM',
                             'VIDEO_SINGLELEGSTAND', 'VIDEO_PLANK', 'VIDEO_SQUAT'))
    create_time = db.Column(db.BigInteger)
    status = db.Column(db.Enum('QUEUEING', 'RUNNING', 'FINISHED', 'ERROR'))
    progress = db.Column(db.Integer)
    result = db.Column(db.Text(16777216))

    def __init__(self, uuid, type, create_time=time.time() * 1000, status='QUEUEING', progress=0, result='{}'):
        self.uuid = uuid
        self.type = type
        self.create_time = create_time
        self.status = status
        self.progress = progress
        self.result = result