mirror of
https://github.com/Dvorinka/SpotifyRecAlg.git
synced 2026-06-04 04:23:02 +00:00
Tomas Dvorak
34 lines
967 B
Python
34 lines
967 B
Python
from sqlalchemy import Integer, insert, select, update
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from swingmusic.db import Base
|
|
from swingmusic.db.engine import DbEngine
|
|
|
|
|
|
class MigrationTable(Base):
|
|
__tablename__ = "dbmigration"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
version: Mapped[int] = mapped_column(Integer())
|
|
|
|
@classmethod
|
|
def set_version(cls, version: int):
|
|
with DbEngine.manager(commit=True) as conn:
|
|
result = conn.execute(
|
|
update(cls).where(cls.id == 1).values(version=version)
|
|
)
|
|
|
|
if result.rowcount == 0:
|
|
conn.execute(insert(cls).values(id=1, version=version))
|
|
|
|
@classmethod
|
|
def get_version(cls):
|
|
with DbEngine.manager() as conn:
|
|
result = conn.execute(select(cls.version).where(cls.id == 1))
|
|
result = result.fetchone()
|
|
|
|
if result:
|
|
return result[0]
|
|
|
|
return -1
|