Moznost nahravat fotky copilotem

2024-03-07 06:45:00 +01:00
parent c0ae73440d
commit 400ad8b75a
15 changed files with 242 additions and 44 deletions
@@ -0,0 +1,53 @@
+"""add flight_has_copilot.token
+
+Revision ID: 7dd139549851
+Revises: a410455fb3bb
+Create Date: 2024-03-02 15:02:26.594830
+
+"""
+import random
+import string
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import text
+
+# revision identifiers, used by Alembic.
+revision = '7dd139549851'
+down_revision = 'a410455fb3bb'
+branch_labels = None
+depends_on = None
+
+
+def rand_str():
+    alphabet = string.ascii_letters + string.digits
+    return ''.join([random.choice(alphabet) for _ in range(64)])
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('flight_has_copilot', sa.Column('token', sa.String(length=64), nullable=True))
+
+    conn = op.get_bind()
+    data = conn.execute(text("SELECT flight_id, copilot_id FROM flight_has_copilot"))
+    for flight_id, copilot_id in data:
+        print(flight_id, copilot_id)
+        conn.execute(text(
+            f"UPDATE flight_has_copilot SET token='{rand_str()}'"
+            f" WHERE flight_id={flight_id} AND copilot_id={copilot_id}"
+        ))
+
+
+    op.create_unique_constraint(None, 'flight_has_copilot', ['token'])
+    op.add_column('photo', sa.Column('created_by_copilot_id', sa.Integer(), nullable=True))
+    op.create_foreign_key(None, 'photo', 'copilot', ['created_by_copilot_id'], ['id'])
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('photo_ibfk_5', 'photo', type_='foreignkey')
+    op.drop_column('photo', 'created_by_copilot_id')
+    op.drop_constraint('token', 'flight_has_copilot', type_='unique')
+    op.drop_column('flight_has_copilot', 'token')
+    # ### end Alembic commands ###