First Elections PR

src/alembic/env.py

@@ -3,6 +3,7 @@
 
 import auth.models
 import database
+import elections.models
 import officers.models
 from alembic import context
 from sqlalchemy import pool

src/alembic/versions/cb5df398547c_create_election_tables.py

@@ -0,0 +1,59 @@
+"""create election tables
+
+Revision ID: cb5df398547c
+Revises: 43f71e4bd6fc
+Create Date: 2024-08-09 09:36:22.745234
+
+"""
+from collections.abc import Sequence
+from typing import Union
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "cb5df398547c"
+down_revision: str | None = "43f71e4bd6fc"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table("election",
+    sa.Column("slug", sa.String(length=32), nullable=False),
+    sa.Column("name", sa.String(length=32), nullable=False),
+    sa.Column("type", sa.String(length=64), nullable=True),
+    sa.Column("date", sa.DateTime(), nullable=True),
+    sa.Column("end_date", sa.DateTime(), nullable=True),
+    sa.Column("websurvey", sa.String(length=300), nullable=True),
+    sa.PrimaryKeyConstraint("slug")
+    )
+    op.create_table("election_nominee",
+    sa.Column("computing_id", sa.String(length=32), nullable=False),
+    sa.Column("full_name", sa.String(length=64), nullable=False),
+    sa.Column("facebook", sa.String(length=64), nullable=True),
+    sa.Column("instagram", sa.String(length=64), nullable=True),
+    sa.Column("email", sa.String(length=64), nullable=True),
+    sa.Column("discord", sa.String(length=32), nullable=True),
+    sa.Column("discord_id", sa.String(length=18), nullable=True),
+    sa.Column("discord_username", sa.String(length=32), nullable=True),
+    sa.PrimaryKeyConstraint("computing_id")
+    )
+    op.create_table("nominee_speech",
+    sa.Column("computing_id", sa.String(length=32), nullable=False),
+    sa.Column("nominee_election", sa.String(length=32), nullable=False),
+    sa.Column("speech", sa.Text(), nullable=True),
+    sa.ForeignKeyConstraint(["computing_id"], ["election_nominee.computing_id"], ),
+    sa.ForeignKeyConstraint(["nominee_election"], ["election.slug"], ),
+    sa.PrimaryKeyConstraint("computing_id", "nominee_election")
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("nominee_speech")
+    op.drop_table("election_nominee")
+    op.drop_table("election")
+    # ### end Alembic commands ###

src/elections/models.py

@@ -0,0 +1,62 @@
+from datetime import datetime
+
+from constants import (
+    COMPUTING_ID_LEN,
+    DISCORD_ID_LEN,
+    DISCORD_NAME_LEN,
+    DISCORD_NICKNAME_LEN,
+)
+from database import Base
+from sqlalchemy import (
+    Column,
+    DateTime,
+    ForeignKeyConstraint,
+    PrimaryKeyConstraint,
+    String,
+    Text,
+)
+
+
+# Each row represents an instance of an election
+class Election(Base):
+    __tablename__ = "election"
+
+    # Slugs are unique identifiers
+    slug = Column(String(32), primary_key=True)
+    name = Column(String(32), nullable=False)
+    # Can be one of (general_election: General Election, by_election: By-Election, council_rep_election: Council Rep Election)
+    type = Column(String(64), default="general_election")
+    date = Column(DateTime, default=datetime.now())
+    end_date = Column(DateTime)
+    websurvey = Column(String(300))
+
+# Each row represents a nominee of a given election
+class Nominee(Base):
+    __tablename__ = "election_nominee"
+
+    # Previously named sfuid
+    computing_id = Column(String(COMPUTING_ID_LEN), primary_key=True)
+    full_name = Column(String(64), nullable=False)
+    facebook = Column(String(64))
+    instagram = Column(String(64))
+    email = Column(String(64))
+    discord = Column(String(DISCORD_NAME_LEN))
+    discord_id = Column(String(DISCORD_ID_LEN))
+    discord_username = Column(String(DISCORD_NICKNAME_LEN))
+
+class NomineeSpeech(Base):
+    __tablename__ = "nominee_speech"
+
+    computing_id = Column(String(COMPUTING_ID_LEN), primary_key=True)
+    nominee_election = Column(String(32), primary_key=True)
+    speech = Column(Text)
+
+    __table_args__ = (
+        PrimaryKeyConstraint(computing_id, nominee_election),
+        ForeignKeyConstraint(
+            ["computing_id"], ["election_nominee.computing_id"]
+        ),
+        ForeignKeyConstraint(
+            ["nominee_election"], ["election.slug"]
+        )
+    )