Migrated links, lyrics and properties
This commit is contained in:
48
scripts/02_migrate_properties.py
Normal file
48
scripts/02_migrate_properties.py
Normal file
@@ -0,0 +1,48 @@
|
||||
import json
|
||||
from collections.abc import Iterable
|
||||
from typing import TypedDict
|
||||
|
||||
from folkugat_web.dal.sql import get_connection
|
||||
from folkugat_web.dal.sql.temes import properties as properties_dal
|
||||
from folkugat_web.model.temes import Property, PropertyField
|
||||
|
||||
query = """ SELECT id, properties FROM temes """
|
||||
PropertyRowTuple = tuple[int, str]
|
||||
|
||||
|
||||
class OldPropertyRowDict(TypedDict):
|
||||
field: str
|
||||
value: str
|
||||
|
||||
|
||||
def _build_property(tema_id: int, property_dict: OldPropertyRowDict) -> Property:
|
||||
return Property(
|
||||
id=None,
|
||||
tema_id=tema_id,
|
||||
field=PropertyField(property_dict["field"]),
|
||||
value=property_dict["value"],
|
||||
)
|
||||
|
||||
|
||||
def _build_properties(row: PropertyRowTuple) -> list[Property]:
|
||||
tema_id = row[0]
|
||||
property_dicts: list[OldPropertyRowDict] = json.loads(row[1])
|
||||
return [_build_property(tema_id, property_dict) for property_dict in property_dicts]
|
||||
|
||||
|
||||
with get_connection() as con:
|
||||
cur = con.cursor()
|
||||
_ = cur.execute(query)
|
||||
iter_rows: Iterable[PropertyRowTuple] = cur.fetchall()
|
||||
rows = list(map(_build_properties, iter_rows))
|
||||
|
||||
_ = cur.execute("DELETE FROM tema_properties")
|
||||
for properties in rows:
|
||||
for property in properties:
|
||||
_ = properties_dal.insert_property(property=property, con=con)
|
||||
|
||||
drop_query = """ ALTER TABLE temes DROP COLUMN properties"""
|
||||
_ = cur.execute(drop_query)
|
||||
|
||||
|
||||
print("DONE!")
|
||||
Reference in New Issue
Block a user