Skip to content

Commit d881db0

Browse files
peterjczzzeek
authored andcommitted
Avoid D103 linter warnings via script.py.mako
Added a basic docstring to the migration template files so that the upgrade/downgrade methods pass the D103 linter check which requires a docstring for public functions. Pull request courtesy Peter Cock. Fixes: #1567 Closes: #1568 Pull-request: #1568 Pull-request-sha: fa3bcd5 Change-Id: I14b7863dfef01edca9c94d8dff4109efd274df3c
1 parent 3d33cfa commit d881db0

File tree

5 files changed

+18
-2
lines changed

5 files changed

+18
-2
lines changed

alembic/templates/async/script.py.mako

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,10 @@ depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
1919

2020

2121
def upgrade() -> None:
22+
"""Upgrade schema."""
2223
${upgrades if upgrades else "pass"}
2324

2425

2526
def downgrade() -> None:
27+
"""Downgrade schema."""
2628
${downgrades if downgrades else "pass"}

alembic/templates/generic/script.py.mako

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,10 @@ depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
1919

2020

2121
def upgrade() -> None:
22+
"""Upgrade schema."""
2223
${upgrades if upgrades else "pass"}
2324

2425

2526
def downgrade() -> None:
27+
"""Downgrade schema."""
2628
${downgrades if downgrades else "pass"}

alembic/templates/multidb/script.py.mako

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,12 @@ depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
2222

2323

2424
def upgrade(engine_name: str) -> None:
25+
"""Upgrade schema."""
2526
globals()["upgrade_%s" % engine_name]()
2627

2728

2829
def downgrade(engine_name: str) -> None:
30+
"""Downgrade schema."""
2931
globals()["downgrade_%s" % engine_name]()
3032

3133
<%
@@ -38,10 +40,12 @@ def downgrade(engine_name: str) -> None:
3840
% for db_name in re.split(r',\s*', db_names):
3941

4042
def upgrade_${db_name}() -> None:
43+
"""Upgrade ${db_name} schema."""
4144
${context.get("%s_upgrades" % db_name, "pass")}
4245

4346

4447
def downgrade_${db_name}() -> None:
48+
"""Downgrade ${db_name} schema."""
4549
${context.get("%s_downgrades" % db_name, "pass")}
4650

4751
% endfor

docs/build/unreleased/1567.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
.. change::
2+
:tags: bug, environment
3+
:tickets: 1567
4+
5+
Added a basic docstring to the migration template files so that the
6+
upgrade/downgrade methods pass the D103 linter check which requires a
7+
docstring for public functions. Pull request courtesy Peter Cock.

tests/test_script_production.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -657,16 +657,17 @@ def process_revision_directives(context, rev, generate_revisions):
657657
result = handle.read()
658658
assert (
659659
(
660-
"""
660+
'''
661661
def upgrade() -> None:
662+
"""Upgrade schema."""
662663
# ### commands auto generated by Alembic - please adjust! ###
663664
op.create_table('test_table',
664665
sa.Column('id', sa.Integer(), nullable=False),
665666
sa.Column('name', sa.String(length=50), nullable=False),
666667
sa.PrimaryKeyConstraint('id')
667668
)
668669
# ### end Alembic commands ###
669-
"""
670+
'''
670671
)
671672
in result
672673
)

0 commit comments

Comments
 (0)