rommapp/romm
1
0
Fork 0
mirror of https://github.com/rommapp/romm.git synced 2026-02-19 07:50:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

rb connection autoreconnect

Browse Source
This commit is contained in:
zurdi zurdo
2023-03-13 18:16:18 +01:00
parent 1f37b85668
commit 568e4f73c5
1 changed files with 29 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
backend/src/handler/db_handler.py
View File

@@ -12,20 +12,22 @@ from logger.logger import log
class DBHandler:
def __init__(self) -> None:
self.DATABASE: str = 'romm'
self.PLATFORM_TABLE: str = 'platform'
self.ROM_TABLE: str = 'rom'
try:
self.conn = db.connect(
host=DB_HOST,
port=DB_PORT,
user=DB_USER,
password=DB_PASSWD,
database=self.DATABASE,
autocommit=True
)
self.conn.auto_reconnect = True
except db.Error as e:
log.error(f"error connecting to the database: {e}")
sys.exit(1)
self.DATABASE: str = 'romm'
self.PLATFORM_TABLE: str = 'platform'
self.ROM_TABLE: str = 'rom'
self.cur = self.conn.cursor()
@@ -33,20 +35,20 @@ class DBHandler:
slug: str, name: str, path_logo: str) -> None:
try:
self.cur.execute(f"""
create table if not exists {self.DATABASE}.{self.PLATFORM_TABLE}
create table if not exists {self.PLATFORM_TABLE}
({igdb_id} varchar(20), {sgdb_id} varchar(20),
{slug} varchar(50), {name} varchar(100),
{path_logo} varchar(500))
""")
log.info(f"{self.DATABASE}.{self.PLATFORM_TABLE} table created")
log.info(f"{self.PLATFORM_TABLE} table created")
except Exception as e:
log.error(f"{self.DATABASE}.{self.PLATFORM_TABLE} table can't be created: {e}")
log.error(f"{self.PLATFORM_TABLE} table can't be created: {e}")
raise HTTPException(status_code=500, detail=f"Can't create platform table. {e}")
try:
self.cur.execute(f"truncate {self.DATABASE}.{self.PLATFORM_TABLE}")
log.info(f"{self.DATABASE}.{self.PLATFORM_TABLE} table truncated")
self.cur.execute(f"truncate {self.PLATFORM_TABLE}")
log.info(f"{self.PLATFORM_TABLE} table truncated")
except Exception as e:
log.error(f"{self.DATABASE}.{self.PLATFORM_TABLE} table can't be truncated: {e}")
log.error(f"{self.PLATFORM_TABLE} table can't be truncated: {e}")
raise HTTPException(status_code=500, detail=f"Can't truncate platform table. {e}")
@@ -54,63 +56,63 @@ class DBHandler:
rom_filename: str, rom_name: str, rom_slug: str, summary: str, platform_slug: str, path_cover_big: str, path_cover_small: str) -> None:
try:
self.cur.execute(f"""
create table if not exists {self.DATABASE}.{self.ROM_TABLE}
create table if not exists {self.ROM_TABLE}
({igdb_id} varchar(20), {sgdb_id} varchar(20),
{platform_igdb_id} varchar(20), {platform_sgdb_id} varchar(20),
{rom_filename} varchar(200), {rom_name} varchar(100),
{rom_slug} varchar(100), {summary} text, {platform_slug} varchar(100),
{path_cover_big} varchar(500), {path_cover_small} varchar(500))
""")
log.info(f"{self.DATABASE}.{self.ROM_TABLE} table created")
log.info(f"{self.ROM_TABLE} table created")
except Exception as e:
log.error(f"{self.DATABASE}.{self.ROM_TABLE} table can't be created: {e}")
log.error(f"{self.ROM_TABLE} table can't be created: {e}")
raise HTTPException(status_code=500, detail=f"Can't create rom table. {e}")
try:
self.cur.execute(f"truncate {self.DATABASE}.{self.ROM_TABLE}")
log.info(f"{self.DATABASE}.{self.ROM_TABLE} table truncated")
self.cur.execute(f"truncate {self.ROM_TABLE}")
log.info(f"{self.ROM_TABLE} table truncated")
except Exception as e:
log.error(f"{self.DATABASE}.{self.ROM_TABLE} table can't be truncated: {e}")
log.error(f"{self.ROM_TABLE} table can't be truncated: {e}")
raise HTTPException(status_code=500, detail=f"Can't truncate rom table. {e}")
def write_platforms(self, platforms: list) -> None:
values: list = [{k: str(v) for k, v in asdict(p).items()} for p in platforms]
try:
self.cur.executemany(f"insert into {self.DATABASE}.{self.PLATFORM_TABLE} (igdb_id, sgdb_id, slug, name, path_logo) \
self.cur.executemany(f"insert into {self.PLATFORM_TABLE} (igdb_id, sgdb_id, slug, name, path_logo) \
values (%(igdb_id)s, %(sgdb_id)s, %(slug)s, %(name)s, %(path_logo)s)", values)
log.info(f"{self.DATABASE}.{self.PLATFORM_TABLE} table populated")
log.info(f"{self.PLATFORM_TABLE} table populated")
except Exception as e:
log.error(f"{self.DATABASE}.{self.PLATFORM_TABLE} can't be populated: {e}")
log.error(f"{self.PLATFORM_TABLE} can't be populated: {e}")
raise HTTPException(status_code=500, detail=f"Can't write in platform table. {e}")
def write_roms(self, roms: list) -> None:
values: list = [{k: str(v) for k, v in asdict(p).items()} for p in roms]
try:
self.cur.executemany(f"insert into {self.DATABASE}.{self.ROM_TABLE} (igdb_id, sgdb_id, platform_igdb_id, platform_sgdb_id, filename, name, slug, summary, platform_slug, path_cover_big, path_cover_small) \
self.cur.executemany(f"insert into {self.ROM_TABLE} (igdb_id, sgdb_id, platform_igdb_id, platform_sgdb_id, filename, name, slug, summary, platform_slug, path_cover_big, path_cover_small) \
values (%(igdb_id)s, %(sgdb_id)s, %(platform_igdb_id)s, %(platform_sgdb_id)s, %(filename)s, %(name)s, %(slug)s, %(summary)s, %(platform_slug)s, %(path_cover_big)s, %(path_cover_small)s)", values)
log.info(f"{self.DATABASE}.{self.ROM_TABLE} table populated")
log.info(f"{self.ROM_TABLE} table populated")
except Exception as e:
log.error(f"{self.DATABASE}.{self.ROM_TABLE} can't be populated: {e}")
log.error(f"{self.ROM_TABLE} can't be populated: {e}")
raise HTTPException(status_code=500, detail=f"Can't write in platform table. {e}")
def get_platforms(self) -> list:
try:
self.cur.execute(f"select igdb_id, sgdb_id, slug, name, path_logo from {self.DATABASE}.{self.PLATFORM_TABLE} order by name asc")
log.info(f"platforms details fetch from {self.DATABASE}.{self.PLATFORM_TABLE}")
self.cur.execute(f"select igdb_id, sgdb_id, slug, name, path_logo from {self.PLATFORM_TABLE} order by name asc")
log.info(f"platforms details fetch from {self.PLATFORM_TABLE}")
except Exception as e:
log.error(f"platforms details can't be fetch from {self.DATABASE}.{self.PLATFORM_TABLE}")
log.error(f"platforms details can't be fetch from {self.PLATFORM_TABLE}")
raise HTTPException(status_code=500, detail=f"Can't read platform table. {e}")
return self.cur
def get_roms(self, platform_slug: str) -> list:
try:
self.cur.execute(f"select igdb_id, sgdb_id, platform_igdb_id, platform_sgdb_id, filename, name, slug, summary, platform_slug, path_cover_big, path_cover_small from {self.DATABASE}.{self.ROM_TABLE} where platform_slug = '{platform_slug}' order by filename asc")
log.info(f"platforms details fetch from {self.DATABASE}.{self.ROM_TABLE}")
self.cur.execute(f"select igdb_id, sgdb_id, platform_igdb_id, platform_sgdb_id, filename, name, slug, summary, platform_slug, path_cover_big, path_cover_small from {self.ROM_TABLE} where platform_slug = '{platform_slug}' order by filename asc")
log.info(f"platforms details fetch from {self.ROM_TABLE}")
except Exception as e:
log.error(f"platforms details can't be fetch from {self.DATABASE}.{self.ROM_TABLE}")
log.error(f"platforms details can't be fetch from {self.ROM_TABLE}")
raise HTTPException(status_code=500, detail=f"Can't read rom table. {e}")
return self.cur