jasima/majuna
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(db): embed pool model in migration

Browse source
This commit is contained in:
Iain Learmonth 2023-01-21 13:38:24 +00:00
parent fa3e4665d0
commit 962f90b384
1 changed files with 7 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

19
migrations/versions/a08ce5e7246a_adds_api_key_field_to_pool_table.py
View file

@ -6,33 +6,28 @@ Create Date: 2022-12-20 18:10:19.540534
""" """
import secrets import secrets
from datetime import datetime
from alembic import op from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from sqlalchemy.ext.automap import automap_base
from app.models.base import Pool
revision = 'a08ce5e7246a' revision = 'a08ce5e7246a'
down_revision = '6a59928efeb7' down_revision = '6a59928efeb7'
branch_labels = None branch_labels = None
depends_on = None depends_on = None
Base = automap_base()
# class Pool(db.Model):
# id = db.Column(db.Integer, primary_key=True)
# description = db.Column(db.String(255), nullable=False)
# added = db.Column(db.DateTime(), default=datetime.utcnow, nullable=False)
# updated = db.Column(db.DateTime(), default=datetime.utcnow, nullable=False)
# destroyed = db.Column(db.DateTime(), nullable=True)
# pool_name = db.Column(db.String(80), unique=True, nullable=False)
# api_key = db.Column(db.String(80), nullable=False)
def upgrade(): def upgrade():
with op.batch_alter_table('pool', schema=None) as batch_op: with op.batch_alter_table('pool', schema=None) as batch_op:
batch_op.add_column(sa.Column('api_key', sa.String(length=80), nullable=True, unique=False)) batch_op.add_column(sa.Column('api_key', sa.String(length=80), nullable=True, unique=False))
session = Session(bind=op.get_bind()) bind = op.get_bind()
Base.prepare(autoload_with=bind)
Pool = Base.classes.pool
session = Session(bind=bind)
for pool in session.query(Pool).all(): for pool in session.query(Pool).all():
pool.api_key = secrets.token_urlsafe(nbytes=32) pool.api_key = secrets.token_urlsafe(nbytes=32)
session.commit() session.commit()