LobotomyLabs/classquiz-ai
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🚑 Fixed migration

Browse Source
This commit is contained in:
Mawoka
2023-06-12 20:22:39 +02:00
parent ce0754e7a0
commit 23ddde9659
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
migrations/versions/89c4b5d547aa_new_storage.py
+3 -1
View File
@@ -12,6 +12,7 @@ import re
import json import json
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from sqlalchemy.sql import text
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
revision = "89c4b5d547aa" revision = "89c4b5d547aa"
@@ -99,13 +100,14 @@ def upgrade() -> None:
session.execute(f"UPDATE quiz SET cover_image = '{new_bg_image}' WHERE id='{id}';") session.execute(f"UPDATE quiz SET cover_image = '{new_bg_image}' WHERE id='{id}';")
except AttributeError: except AttributeError:
continue continue
s = text("UPDATE quiz SET questions = ':q' WHERE id=':i';")
all_questions = session.execute("SELECT questions, id from quiz;") all_questions = session.execute("SELECT questions, id from quiz;")
question_image_regex = rf"{settings.root_address}/api/v1/storage/download/(?=.{{36}}--.{{36}})" question_image_regex = rf"{settings.root_address}/api/v1/storage/download/(?=.{{36}}--.{{36}})"
for question, id in all_questions: for question, id in all_questions:
question_as_json = json.dumps(question) question_as_json = json.dumps(question)
result = re.sub(question_image_regex, "", question_as_json) result = re.sub(question_image_regex, "", question_as_json)
session.execute(f"UPDATE quiz SET questions = '{result}' WHERE id='{id}';") session.execute(s, q=result, i=id)
## ADDED THUMBHASH AND SERVER STORAGE ITEM ## ADDED THUMBHASH AND SERVER STORAGE ITEM
op.add_column("storage_items", sa.Column("thumbhash", sa.Text(), nullable=True)) op.add_column("storage_items", sa.Column("thumbhash", sa.Text(), nullable=True))