Skip to content

Commit 3d761af

Browse files
AMZheleva21AMZheleva21
committed
Add migrations to the project
1 parent c9afa00 commit 3d761af

File tree

7 files changed

+273
-0
lines changed

7 files changed

+273
-0
lines changed

‎My_path_team9/__init__.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,17 +2,20 @@
22
from flask_bootstrap import Bootstrap
33
from flask_login import LoginManager
44
from flask_sqlalchemy import SQLAlchemy
5+
from flask_migrate import Migrate
56

67

78
db = SQLAlchemy()
89
bootstrap = Bootstrap()
910
login_manager = LoginManager()
11+
migrate = Migrate()
1012

1113

1214
def create_app(config):
1315
app = Flask(__name__)
1416
app.config.from_object(config)
1517
db.init_app(app)
18+
migrate.init_app(app, db)
1619
bootstrap.init_app(app)
1720
login_manager.init_app(app)
1821

‎My_path_team9/migrations/README‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Single-database configuration for Flask.

‎My_path_team9/migrations/alembic.ini‎

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
# A generic, single database configuration.
2+
3+
[alembic]
4+
# template used to generate migration files
5+
# file_template = %%(rev)s_%%(slug)s
6+
7+
# set to 'true' to run the environment during
8+
# the 'revision' command, regardless of autogenerate
9+
# revision_environment = false
10+
11+
12+
# Logging configuration
13+
[loggers]
14+
keys = root,sqlalchemy,alembic,flask_migrate
15+
16+
[handlers]
17+
keys = console
18+
19+
[formatters]
20+
keys = generic
21+
22+
[logger_root]
23+
level = WARN
24+
handlers = console
25+
qualname =
26+
27+
[logger_sqlalchemy]
28+
level = WARN
29+
handlers =
30+
qualname = sqlalchemy.engine
31+
32+
[logger_alembic]
33+
level = INFO
34+
handlers =
35+
qualname = alembic
36+
37+
[logger_flask_migrate]
38+
level = INFO
39+
handlers =
40+
qualname = flask_migrate
41+
42+
[handler_console]
43+
class = StreamHandler
44+
args = (sys.stderr,)
45+
level = NOTSET
46+
formatter = generic
47+
48+
[formatter_generic]
49+
format = %(levelname)-5.5s [%(name)s] %(message)s
50+
datefmt = %H:%M:%S

‎My_path_team9/migrations/env.py‎

Lines changed: 113 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,113 @@
1+
import logging
2+
from logging.config import fileConfig
3+
4+
from flask import current_app
5+
6+
from alembic import context
7+
8+
# this is the Alembic Config object, which provides
9+
# access to the values within the .ini file in use.
10+
config = context.config
11+
12+
# Interpret the config file for Python logging.
13+
# This line sets up loggers basically.
14+
fileConfig(config.config_file_name)
15+
logger = logging.getLogger('alembic.env')
16+
17+
18+
def get_engine():
19+
try:
20+
# this works with Flask-SQLAlchemy<3 and Alchemical
21+
return current_app.extensions['migrate'].db.get_engine()
22+
except (TypeError, AttributeError):
23+
# this works with Flask-SQLAlchemy>=3
24+
return current_app.extensions['migrate'].db.engine
25+
26+
27+
def get_engine_url():
28+
try:
29+
return get_engine().url.render_as_string(hide_password=False).replace(
30+
'%', '%%')
31+
except AttributeError:
32+
return str(get_engine().url).replace('%', '%%')
33+
34+
35+
# add your model's MetaData object here
36+
# for 'autogenerate' support
37+
# from myapp import mymodel
38+
# target_metadata = mymodel.Base.metadata
39+
config.set_main_option('sqlalchemy.url', get_engine_url())
40+
target_db = current_app.extensions['migrate'].db
41+
42+
# other values from the config, defined by the needs of env.py,
43+
# can be acquired:
44+
# my_important_option = config.get_main_option("my_important_option")
45+
# ... etc.
46+
47+
48+
def get_metadata():
49+
if hasattr(target_db, 'metadatas'):
50+
return target_db.metadatas[None]
51+
return target_db.metadata
52+
53+
54+
def run_migrations_offline():
55+
"""Run migrations in 'offline' mode.
56+
57+
This configures the context with just a URL
58+
and not an Engine, though an Engine is acceptable
59+
here as well. By skipping the Engine creation
60+
we don't even need a DBAPI to be available.
61+
62+
Calls to context.execute() here emit the given string to the
63+
script output.
64+
65+
"""
66+
url = config.get_main_option("sqlalchemy.url")
67+
context.configure(
68+
url=url, target_metadata=get_metadata(), literal_binds=True
69+
)
70+
71+
with context.begin_transaction():
72+
context.run_migrations()
73+
74+
75+
def run_migrations_online():
76+
"""Run migrations in 'online' mode.
77+
78+
In this scenario we need to create an Engine
79+
and associate a connection with the context.
80+
81+
"""
82+
83+
# this callback is used to prevent an auto-migration from being generated
84+
# when there are no changes to the schema
85+
# reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html
86+
def process_revision_directives(context, revision, directives):
87+
if getattr(config.cmd_opts, 'autogenerate', False):
88+
script = directives[0]
89+
if script.upgrade_ops.is_empty():
90+
directives[:] = []
91+
logger.info('No changes in schema detected.')
92+
93+
conf_args = current_app.extensions['migrate'].configure_args
94+
if conf_args.get("process_revision_directives") is None:
95+
conf_args["process_revision_directives"] = process_revision_directives
96+
97+
connectable = get_engine()
98+
99+
with connectable.connect() as connection:
100+
context.configure(
101+
connection=connection,
102+
target_metadata=get_metadata(),
103+
**conf_args
104+
)
105+
106+
with context.begin_transaction():
107+
context.run_migrations()
108+
109+
110+
if context.is_offline_mode():
111+
run_migrations_offline()
112+
else:
113+
run_migrations_online()

‎My_path_team9/migrations/script.py.mako‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
"""${message}
2+
3+
Revision ID: ${up_revision}
4+
Revises: ${down_revision | comma,n}
5+
Create Date: ${create_date}
6+
7+
"""
8+
from alembic import op
9+
import sqlalchemy as sa
10+
${imports if imports else ""}
11+
12+
# revision identifiers, used by Alembic.
13+
revision = ${repr(up_revision)}
14+
down_revision = ${repr(down_revision)}
15+
branch_labels = ${repr(branch_labels)}
16+
depends_on = ${repr(depends_on)}
17+
18+
19+
def upgrade():
20+
${upgrades if upgrades else "pass"}
21+
22+
23+
def downgrade():
24+
${downgrades if downgrades else "pass"}

‎My_path_team9/migrations/versions/27f77abad8e5_add_user_id_to_survey_model.py‎

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
"""Add user_id column as nullable to survey table
2+
3+
Revision ID: 27f77abad8e5
4+
Revises:
5+
Create Date: 2025-06-16 22:15:58.008582
6+
"""
7+
from alembic import op
8+
import sqlalchemy as sa
9+
10+
revision = '27f77abad8e5'
11+
down_revision = None
12+
branch_labels = None
13+
depends_on = None
14+
15+
def upgrade():
16+
with op.batch_alter_table('survey') as batch_op:
17+
batch_op.add_column(sa.Column('user_id', sa.Integer(), nullable=True))
18+
batch_op.create_foreign_key(None, 'users', ['user_id'], ['id'])
19+
20+
def downgrade():
21+
with op.batch_alter_table('survey') as batch_op:
22+
batch_op.drop_constraint(None, type_='foreignkey')
23+
batch_op.drop_column('user_id')
24+
25+
26+
27+
def downgrade():
28+
# ### commands auto generated by Alembic - please adjust! ###
29+
with op.batch_alter_table('survey', schema=None) as batch_op:
30+
batch_op.drop_constraint(None, type_='foreignkey')
31+
batch_op.drop_column('user_id')
32+
33+
op.create_table('messages',
34+
sa.Column('id', sa.INTEGER(), sa.Identity(always=False, start=1, increment=1), autoincrement=True, nullable=False),
35+
sa.Column('sender_id', sa.INTEGER(), autoincrement=False, nullable=False),
36+
sa.Column('receiver_id', sa.INTEGER(), autoincrement=False, nullable=False),
37+
sa.Column('content', sa.VARCHAR(collation='SQL_Latin1_General_CP1_CI_AS'), autoincrement=False, nullable=False),
38+
sa.Column('timestamp', sa.DATETIME(), autoincrement=False, nullable=True),
39+
sa.ForeignKeyConstraint(['receiver_id'], ['users.id'], name=op.f('FK__messages__receiv__70DDC3D8')),
40+
sa.ForeignKeyConstraint(['sender_id'], ['users.id'], name=op.f('FK__messages__sender__6FE99F9F')),
41+
sa.PrimaryKeyConstraint('id', name=op.f('PK__messages__3213E83FD2330A77'))
42+
)
43+
op.create_table('posts',
44+
sa.Column('id', sa.INTEGER(), sa.Identity(always=False, start=1, increment=1), autoincrement=True, nullable=False),
45+
sa.Column('content', sa.VARCHAR(collation='SQL_Latin1_General_CP1_CI_AS'), autoincrement=False, nullable=False),
46+
sa.Column('timestamp', sa.DATETIME(), autoincrement=False, nullable=True),
47+
sa.Column('user_id', sa.INTEGER(), autoincrement=False, nullable=False),
48+
sa.ForeignKeyConstraint(['user_id'], ['users.id'], name=op.f('FK__posts__user_id__02FC7413')),
49+
sa.PrimaryKeyConstraint('id', name=op.f('PK__posts__3213E83FF61DD0E1'))
50+
)
51+
# ### end Alembic commands ###

‎My_path_team9/migrations/versions/a66a06943692_make_user_id_non_nullable.py‎

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
"""Make user_id non-nullable
2+
3+
Revision ID: a66a06943692
4+
Revises: 27f77abad8e5
5+
Create Date: 2025-06-16 22:34:19.641758
6+
7+
"""
8+
from alembic import op
9+
import sqlalchemy as sa
10+
11+
12+
# revision identifiers, used by Alembic.
13+
revision = 'a66a06943692'
14+
down_revision = '27f77abad8e5'
15+
branch_labels = None
16+
depends_on = None
17+
18+
19+
def upgrade():
20+
with op.batch_alter_table('survey') as batch_op:
21+
batch_op.alter_column('user_id',
22+
existing_type=sa.Integer(),
23+
nullable=False)
24+
25+
def downgrade():
26+
with op.batch_alter_table('survey') as batch_op:
27+
batch_op.alter_column('user_id',
28+
existing_type=sa.Integer(),
29+
nullable=True)
30+
31+

0 commit comments

Comments
 (0)