Add prices for lower quantities (500-999L)

Don't forget to run the migratedb command
This commit is contained in:
2022-08-16 00:39:22 +02:00
parent b5e981291f
commit 10d288f3c1
3 changed files with 95 additions and 16 deletions

View File

@@ -35,8 +35,27 @@ class Price(BaseModel):
id = AutoField()
date = DateField(unique=True)
price = IntegerField()
price500 = IntegerField()
def init():
db.connect()
db.create_tables([Price])
def migrate_001():
backend = conf["db"].get("backend", None)
if backend == "sqlite":
from playhouse.migrate import SqliteMigrator
migrator = SqliteMigrator(db)
elif backend == "mysql":
from playhouse.migrate import MySQLMigrator
migrator = MysqlMigrator(db)
from playhouse.migrate import migrate
price500_field = IntegerField(null=True)
with db.atomic():
migrate(migrator.add_column("price", "price500", price500_field))