Fix migration error when migrating from scratch

This commit is contained in:
rubenwardy
2020-09-01 14:57:03 +01:00
parent 934d581737
commit 5b086bb559
4 changed files with 68 additions and 32 deletions

View File

@@ -21,16 +21,6 @@ depends_on = None
def upgrade():
op.add_column('package', sa.Column('downloads', sa.Integer(), nullable=False, server_default="0"))
bind = op.get_bind()
session = orm.Session(bind=bind)
for package in session.query(Package).all():
downloads_result = session.query(func.sum(PackageRelease.downloads)).filter_by(package_id=package.id).one_or_none()
downloads = 0 if not downloads_result or not downloads_result[0] else downloads_result[0]
package.downloads = downloads
session.commit()
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###

View File

@@ -31,8 +31,7 @@ def upgrade():
bind = op.get_bind()
session = orm.Session(bind=bind)
for package in session.query(Package).all():
package.maintainers.append(package.author)
op.execute('INSERT INTO maintainers (package_id, user_id) SELECT id, author_id FROM package;')
session.commit()