24 lines
696 B
Python
24 lines
696 B
Python
import os
|
|
from sqlalchemy import text
|
|
|
|
def generate_alembic(session):
|
|
|
|
try:
|
|
result = session.execute(
|
|
text(
|
|
"SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = "
|
|
"'alembic_version') AS table_existence;"
|
|
)
|
|
)
|
|
if result.first()[0]:
|
|
session.execute(text("delete from alembic_version;"))
|
|
session.commit()
|
|
except Exception as e:
|
|
print(e)
|
|
finally:
|
|
run_command = "python -m alembic stamp head;"
|
|
run_command += (
|
|
"python -m alembic revision --autogenerate;python -m alembic upgrade head;"
|
|
)
|
|
os.system(run_command)
|