image_superres_fastapi_example/datatype.py at main · luixiao0/image_superres_fastapi_example · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from pydantic import BaseModel
from typing import Optional
import os
import json
import config

class Token(BaseModel):
    access_token: str
    token_type: str


class TokenData(BaseModel):
    username: Optional[str] = None


class Task_front:
    def __init__(self, parent):
        self.id = parent.taskid
        self.s = parent.state
        self.p = parent.args
        self.date = parent.date


class Task:
    def __init__(self, filename, args, uid):
        self.img = filename
        self.args = json.loads(args)
        self.state = 0
        self.date = ""
        self.uid = uid
        self.taskid = 0

    def finished(self):
        if self.state == 1:
            return True
        else:
            return False

    def commit(self):
        task = config.db.newTask(self.img, self.uid, self.args)
        self.taskid = task.taskid
        self.date = task.date

    def set(self, state):
        if state != self.state:
            self.state = state
            config.db.toState(self.taskid, state)

    def front_dump(self):
        return Task_front(self)

    def get(self, c):
        filename, ext = self.img.split('.')
        root = os.path.join(config.storage, str(self.uid))
        if c == 'root':
            return root
        if c == 'input':
            return os.path.join(root, c, self.img)
        elif c == 'output':
            return os.path.join(root, c, "{}.png".format(filename))
        else:
            return os.path.join(root, c, "{}.jpg".format(filename))


class regItem(BaseModel):
    uname:str
    psw:str


class User(BaseModel):
    id = 0
    name = ""
    t_num = 0
    hashedkey = ""
    tasks = ""