56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from datetime import datetime
|
|
from uuid import UUID, uuid4
|
|
from pony.orm import *
|
|
|
|
db = Database()
|
|
|
|
STATE_QUEUED = 0
|
|
STATE_STARTED = 1
|
|
STATE_ERROR = -1
|
|
|
|
|
|
class Patch(db.Entity):
|
|
id = PrimaryKey(int, auto=True)
|
|
player = Required(int)
|
|
team = Required(int, default=0)
|
|
data = Required(buffer, lazy=True)
|
|
seed = Optional('Seed')
|
|
|
|
|
|
class Room(db.Entity):
|
|
id = PrimaryKey(UUID, default=uuid4)
|
|
last_activity = Required(datetime, default=lambda: datetime.utcnow(), index=True)
|
|
creation_time = Required(datetime, default=lambda: datetime.utcnow())
|
|
owner = Required(UUID, index=True)
|
|
commands = Set('Command')
|
|
seed = Required('Seed', index=True)
|
|
multisave = Optional(Json, lazy=True)
|
|
show_spoiler = Required(int, default=0) # 0 -> never, 1 -> after completion, -> 2 always
|
|
timeout = Required(int, default=lambda: 6 * 60 * 60) # seconds since last activity to shutdown
|
|
tracker = Optional(UUID, index=True)
|
|
last_port = Optional(int, default=lambda: 0)
|
|
|
|
|
|
class Seed(db.Entity):
|
|
id = PrimaryKey(UUID, default=uuid4)
|
|
rooms = Set(Room)
|
|
multidata = Optional(Json, lazy=True)
|
|
owner = Required(UUID, index=True)
|
|
creation_time = Required(datetime, default=lambda: datetime.utcnow())
|
|
patches = Set(Patch)
|
|
spoiler = Optional(LongStr, lazy=True)
|
|
|
|
|
|
class Command(db.Entity):
|
|
id = PrimaryKey(int, auto=True)
|
|
room = Required(Room)
|
|
commandtext = Required(str)
|
|
|
|
|
|
class Generation(db.Entity):
|
|
id = PrimaryKey(UUID, default=uuid4)
|
|
owner = Required(UUID)
|
|
options = Required(bytes, lazy=True) # these didn't work as JSON on mariaDB, so they're getting pickled now
|
|
meta = Required(bytes, lazy=True) # if state is -1 (error) this will contain an utf-8 encoded error message
|
|
state = Required(int, default=0, index=True)
|