Added Alembic for migrations

2023-05-27 23:14:06 +02:00
parent adda3e6f25
commit de1bf05922
14 changed files with 346 additions and 47 deletions
--- a/migrations/versions/2023_05_27_2146-684d32cf7e4d_initial.py
+++ b/migrations/versions/2023_05_27_2146-684d32cf7e4d_initial.py
@@ -0,0 +1,83 @@
+"""initial.
+
+Revision ID: 684d32cf7e4d
+Create Date: 2023-05-27 21:46:24.361353
+"""
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.engine.reflection import Inspector
+
+# revision identifiers, used by Alembic.
+revision = "684d32cf7e4d"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Perform the upgrade."""
+    conn = op.get_bind()
+    inspector = Inspector.from_engine(conn)  # type: ignore[arg-type]
+    tables = inspector.get_table_names()
+    if "vendor" in tables:
+        # If the vendor table exists, we assume that the initial migration has already been performed.
+        return
+
+    op.create_table(
+        "vendor",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("registered", sa.DateTime(), nullable=False, server_default=sa.func.now()),
+        sa.Column("name", sa.String(length=64), nullable=False),
+        sa.Column("comment", sa.String(length=1024), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_vendor_id"), "vendor", ["id"], unique=False)
+    op.create_table(
+        "filament",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("registered", sa.DateTime(), nullable=False, server_default=sa.func.now()),
+        sa.Column("name", sa.String(length=64), nullable=True),
+        sa.Column("vendor_id", sa.Integer(), nullable=True),
+        sa.Column("material", sa.String(length=64), nullable=True),
+        sa.Column("price", sa.Float(), nullable=True),
+        sa.Column("density", sa.Float(), nullable=False),
+        sa.Column("diameter", sa.Float(), nullable=False),
+        sa.Column("weight", sa.Float(), nullable=True, comment="The filament weight of a full spool (net weight)."),
+        sa.Column("spool_weight", sa.Float(), nullable=True, comment="The weight of an empty spool."),
+        sa.Column("article_number", sa.String(length=64), nullable=True),
+        sa.Column("comment", sa.String(length=1024), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["vendor_id"],
+            ["vendor.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_filament_id"), "filament", ["id"], unique=False)
+    op.create_table(
+        "spool",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("registered", sa.DateTime(), nullable=False, server_default=sa.func.now()),
+        sa.Column("first_used", sa.DateTime(), nullable=True),
+        sa.Column("last_used", sa.DateTime(), nullable=True),
+        sa.Column("filament_id", sa.Integer(), nullable=False),
+        sa.Column("used_weight", sa.Float(), nullable=False),
+        sa.Column("location", sa.String(length=64), nullable=True),
+        sa.Column("lot_nr", sa.String(length=64), nullable=True),
+        sa.Column("comment", sa.String(length=1024), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["filament_id"],
+            ["filament.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_spool_id"), "spool", ["id"], unique=False)
+
+
+def downgrade() -> None:
+    """Perform the downgrade."""
+    op.drop_index(op.f("ix_spool_id"), table_name="spool")
+    op.drop_table("spool")
+    op.drop_index(op.f("ix_filament_id"), table_name="filament")
+    op.drop_table("filament")
+    op.drop_index(op.f("ix_vendor_id"), table_name="vendor")
+    op.drop_table("vendor")