Fixes database table creation and adds profile

image column
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-05 18:15:30 -03:00
commit 811da0323d
2 changed files with 6 additions and 6 deletions

View file

@ -23,7 +23,7 @@ def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind() conn = op.get_bind()
inspector = Inspector.from_engine(conn) inspector = Inspector.from_engine(conn) # type: ignore
# List existing tables # List existing tables
existing_tables = inspector.get_table_names() existing_tables = inspector.get_table_names()
# Drop 'flowstyle' table if it exists # Drop 'flowstyle' table if it exists
@ -145,8 +145,8 @@ def upgrade() -> None:
def downgrade() -> None: def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind() conn = op.get_bind()
inspector = Inspector.from_engine(conn) inspector = Inspector.from_engine(conn) # type: ignore
# List existing tables # List existing tables
existing_tables = inspector.get_table_names() existing_tables = inspector.get_table_names()
if "flow" in existing_tables: if "flow" in existing_tables:

View file

@ -7,9 +7,9 @@ Create Date: 2023-09-08 07:36:13.387318
""" """
from typing import Sequence, Union from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
import sqlmodel import sqlmodel
from alembic import op
from sqlalchemy.engine.reflection import Inspector from sqlalchemy.engine.reflection import Inspector
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
@ -22,7 +22,7 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None: def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind() conn = op.get_bind()
inspector = Inspector.from_engine(conn) inspector = Inspector.from_engine(conn) # type: ignore
if "user" in inspector.get_table_names() and "profile_image" not in [ if "user" in inspector.get_table_names() and "profile_image" not in [
column["name"] for column in inspector.get_columns("user") column["name"] for column in inspector.get_columns("user")
]: ]:
@ -39,7 +39,7 @@ def upgrade() -> None:
def downgrade() -> None: def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind() conn = op.get_bind()
inspector = Inspector.from_engine(conn) inspector = Inspector.from_engine(conn) # type: ignore
if "user" in inspector.get_table_names() and "profile_image" in [ if "user" in inspector.get_table_names() and "profile_image" in [
column["name"] for column in inspector.get_columns("user") column["name"] for column in inspector.get_columns("user")
]: ]: