✨ Added working user-avatars

migrations/env.py

@@ -6,6 +6,8 @@ from logging.config import fileConfig
 from classquiz.db.models import metadata
 from classquiz.config import settings
 from sqlalchemy import create_engine
+from sqlalchemy.ext.asyncio import async_engine_from_config
+import asyncio
 
 from alembic import context
 

migrations/versions/ff573859eb32_user_avatar.py

@@ -0,0 +1,43 @@
+"""user avatar
+
+Revision ID: ff573859eb32
+Revises: da778d551bf4
+Create Date: 2022-07-05 20:36:44.858963
+
+"""
+import asyncio
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.orm import Session
+from classquiz.helpers.avatar import gzipped_user_avatar
+import ormar
+from classquiz.db.models import User
+from classquiz.db import metadata, database
+
+# revision identifiers, used by Alembic.
+revision = "ff573859eb32"
+down_revision = "da778d551bf4"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("users", sa.Column("avatar", sa.LargeBinary(length=25000), nullable=True))
+    conn = op.get_bind()
+    session = Session(bind=conn)
+    res = session.execute("SELECT id from users;")
+    for row in res:
+        user_id = str(row).strip(",.'()")
+        avatar = gzipped_user_avatar().hex()
+        session.execute(f"UPDATE users SET avatar = (decode('{avatar}', 'hex')) WHERE users.id='{user_id}'")
+    op.alter_column("users", "avatar", nullable=False)
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("users", "avatar")
+    # ### end Alembic commands ###