Files
TheGrapefruitsDuo/server/app/db/musicians.py
2024-05-03 12:55:08 -07:00

44 lines
1.4 KiB
Python

from icecream import ic
from app.constants import MUSICIAN_TABLE
from app.db.base_queries import BaseQueries
from app.db.conn import connect_db
from app.models.musician import Musician
class MusicianQueries(BaseQueries):
def __init__(self) -> None:
super().__init__()
self.table = MUSICIAN_TABLE
async def update_bio(self, musician: Musician, bio: str) -> None:
"""Updates a musician's biography in the database.
Args:
musician (Musician): The musician object to update
bio (str): The new biography for the musician
"""
db = connect_db()
cursor = db.cursor()
query = f"UPDATE {self.table} SET bio = %s WHERE id = %s"
cursor.execute(query, (bio, musician.id))
db.commit()
cursor.close()
db.close()
async def update_headshot(self, musician: Musician, headshot_id: str) -> None:
"""Updates a musician's headshot ID in the database.
The image itself is stored with Cloudinary.
Args:
musician (Musician): The musician object to update
headshot_id (str): The public ID of the new headshot (as determined by Cloudinary)
"""
db = connect_db()
cursor = db.cursor()
query = f"UPDATE {self.table} SET headshot_id = %s WHERE id = %s"
cursor.execute(query, (headshot_id, musician.id))
db.commit()
cursor.close()
db.close()