retro-back/setup_db.py
2025-01-22 17:18:24 +01:00

32 lines
1.2 KiB
Python

from sqlalchemy.orm import Session
from db_classes import *
from sqlalchemy import create_engine
from initial_data import CONSOLE_DATA, LANGUAGES
#db pass MumMTK3bwjYJRfe
def create_db():
engine = create_engine('mariadb+mariadbconnector://videogamedb:MumMTK3bwjYJRfe@simailadjalim.fr/videogamedb')
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
session = Session(engine)
consoles = []
rom_file_extensions = {}
languages = []
for console_name, console_data in CONSOLE_DATA.items():
console = Console(console_name,console_data["console_core"] ,console_data["playable"], [])
consoles.append(console)
for extension in console_data["extensions"]:
if extension not in rom_file_extensions:
rom_file_extensions[extension] = RomFileExtensions(extension, [])
rom_file_extensions[extension].consoles.append(console)
for language_name, language_data in LANGUAGES.items():
language = Language(language_name, [])
languages.append(language)
session.add_all(consoles)
session.add_all(rom_file_extensions.values())
session.add_all(languages)
session.commit()
create_db()