feat(lists): adds redirector data format

This commit is contained in:
Iain Learmonth 2022-12-21 19:27:46 +00:00
parent c16a80dc16
commit 60255afe3f
8 changed files with 117 additions and 4 deletions

View file

@ -0,0 +1,45 @@
"""Adds api key field to pool table
Revision ID: a08ce5e7246a
Revises: 6a59928efeb7
Create Date: 2022-12-20 18:10:19.540534
"""
import secrets
from alembic import op
import sqlalchemy as sa
from sqlalchemy.orm import Session
from app.models.base import Pool
revision = 'a08ce5e7246a'
down_revision = '6a59928efeb7'
branch_labels = None
depends_on = None
# 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():
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))
session = Session(bind=op.get_bind())
for pool in session.query(Pool).all():
pool.api_key = secrets.token_urlsafe(nbytes=32)
session.commit()
with op.batch_alter_table('pool', schema=None) as batch_op:
batch_op.alter_column('api_key', nullable=False, unique=True)
def downgrade():
with op.batch_alter_table('pool', schema=None) as batch_op:
batch_op.drop_column('api_key')