awesome-fastapi-projects/migrations/versions/90eb9d1f9267_set_up_the_database.py

74 lines
2.3 KiB
Python

"""Set up the database
Revision ID: 90eb9d1f9267
Revises:
Create Date: 2023-08-15 14:13:30.562069
"""
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision = "90eb9d1f9267"
down_revision = None
branch_labels = None
depends_on = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"dependency",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("name", sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint("id", name=op.f("pk_dependency")),
sa.UniqueConstraint("name", name=op.f("uq_dependency_name ")),
)
op.create_table(
"repo",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("url", sa.String(), nullable=False),
sa.Column("description", sa.Text(), nullable=False),
sa.Column("stars", sa.BigInteger(), nullable=False),
sa.Column("source_graph_repo_id", sa.BigInteger(), nullable=True),
sa.PrimaryKeyConstraint("id", name=op.f("pk_repo")),
sa.UniqueConstraint(
"source_graph_repo_id", name=op.f("uq_repo_source_graph_repo_id ")
),
sa.UniqueConstraint(
"url",
"source_graph_repo_id",
name=op.f("uq_repo_url_source_graph_repo_id "),
),
sa.UniqueConstraint("url", name=op.f("uq_repo_url ")),
)
op.create_table(
"repo_dependency",
sa.Column("repo_id", sa.Integer(), nullable=False),
sa.Column("dependency_id", sa.Integer(), nullable=False),
sa.ForeignKeyConstraint(
["dependency_id"],
["dependency.id"],
name=op.f("fk_repo_dependency_dependency_id_dependency"),
ondelete="CASCADE",
),
sa.ForeignKeyConstraint(
["repo_id"],
["repo.id"],
name=op.f("fk_repo_dependency_repo_id_repo"),
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint(
"repo_id", "dependency_id", name=op.f("pk_repo_dependency")
),
)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table("repo_dependency")
op.drop_table("repo")
op.drop_table("dependency")
# ### end Alembic commands ###