Skip to content

Commit

Permalink
make winner not nullable in favor of Team.NONE
Browse files Browse the repository at this point in the history
  • Loading branch information
spookybear0 committed Jan 18, 2025
1 parent c08f915 commit 9e54049
Show file tree
Hide file tree
Showing 3 changed files with 29 additions and 2 deletions.
2 changes: 1 addition & 1 deletion db/laserball.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@

class LaserballGame(Model):
id = fields.IntField(pk=True)
winner = fields.CharEnumField(Team, null=True)
winner = fields.CharEnumField(Team)
winner_color = fields.CharField(20)
tdf_name = fields.CharField(100)
file_version = fields.CharField(20) # version is a decimal number, we can just store it as a string
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
return """
ALTER TABLE `laserballgame` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `laserballgame` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacylaserballgame` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacylaserballgameplayer` MODIFY COLUMN `team` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacysm5game` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacysm5gameplayer` MODIFY COLUMN `team` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `sm5game` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `sm5game` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `sm5game` MODIFY COLUMN `last_team_standing` VARCHAR(7) COMMENT 'Neutral Team: neutral\nNone Team: none\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';"""


async def downgrade(db: BaseDBAsyncClient) -> str:
return """
ALTER TABLE `sm5game` MODIFY COLUMN `winner` VARCHAR(7) COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `sm5game` MODIFY COLUMN `winner` VARCHAR(7) COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `sm5game` MODIFY COLUMN `last_team_standing` VARCHAR(7) COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `laserballgame` MODIFY COLUMN `winner` VARCHAR(7) COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `laserballgame` MODIFY COLUMN `winner` VARCHAR(7) COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacysm5game` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacylaserballgame` MODIFY COLUMN `winner` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacysm5gameplayer` MODIFY COLUMN `team` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';
ALTER TABLE `legacylaserballgameplayer` MODIFY COLUMN `team` VARCHAR(7) NOT NULL COMMENT 'Neutral Team: neutral\nFire Team: red\nEarth Team: green\nIce Team: blue\nYellow Team: yellow\nPurple Team: purple';"""
2 changes: 1 addition & 1 deletion db/sm5.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

class SM5Game(Model):
id = fields.IntField(pk=True)
winner = fields.CharEnumField(Team, null=True) # null if the game ended early
winner = fields.CharEnumField(Team)
winner_color = fields.CharField(20)
tdf_name = fields.CharField(100)
file_version = fields.CharField(20) # version is a decimal number, we can just store it as a string
Expand Down

0 comments on commit 9e54049

Please sign in to comment.