structure refactoring

Author: Masterain98

base_logger.py

@@ -0,0 +1,16 @@
+import logging
+import os
+
+
+logger = logging
+if os.getenv("DEBUG") == "1":
+    logging.basicConfig(
+        level=logging.DEBUG,
+        format='%(levelname)s %(asctime)s -> %(message)s',
+
+        datefmt='%Y-%m-%dT%H:%M:%S')
+else:
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(levelname)s %(asctime)s -> %(message)s',
+        datefmt='%Y-%m-%dT%H:%M:%S')

db/__init__.py

@@ -0,0 +1,84 @@
+import json
+import logging
+from sqlalchemy.orm import Session
+from sqlalchemy import exc
+
+from db.models import I18nDict, GenericDict
+from typing import Dict, List, Optional
+from api_config import game_name_id_map, ACCEPTED_LANGUAGES, LANGUAGE_PAIRS
+
+
+def get_game_id_by_name(this_game_name: str) -> Optional[int]:
+    return game_name_id_map.get(this_game_name)
+
+
+def get_lang_column(lang: str):
+    """
+    Map from short-code to I18nDict property.
+    """
+    mapper = {
+        "chs": I18nDict.chs_text,
+        "cht": I18nDict.cht_text,
+        "de": I18nDict.de_text,
+        "en": I18nDict.en_text,
+        "es": I18nDict.es_text,
+        "fr": I18nDict.fr_text,
+        "id": I18nDict.id_text,
+        "jp": I18nDict.jp_text,
+        "kr": I18nDict.kr_text,
+        "pt": I18nDict.pt_text,
+        "ru": I18nDict.ru_text,
+        "th": I18nDict.th_text,
+        "vi": I18nDict.vi_text,
+    }
+    return mapper.get(lang)
+
+
+def clear_game_data(db: Session, game_id: int):
+    """ Delete old data from i18n_dict & generic_dict for a given game_id. """
+    db.query(I18nDict).filter(I18nDict.game_id == game_id).delete()
+    db.query(GenericDict).filter(GenericDict.game_id == game_id).delete()
+    db.commit()
+
+
+def insert_localization_data(
+        db: Session,
+        game_id: int,
+        localization_dict: Dict[str, Dict[str, str]]
+):
+    """
+    Insert data into i18n_dict & generic_dict.
+    localization_dict = { item_id: { 'en': 'some text', 'chs': '中文', ...}, ... }
+    """
+    for item_id, translation in localization_dict.items():
+        i18n_entry = I18nDict(
+            game_id=game_id,
+            item_id=item_id,
+            chs_text=translation.get("chs", ""),
+            cht_text=translation.get("cht", ""),
+            de_text=translation.get("de", ""),
+            en_text=translation.get("en", ""),
+            es_text=translation.get("es", ""),
+            fr_text=translation.get("fr", ""),
+            id_text=translation.get("id", ""),
+            jp_text=translation.get("jp", ""),
+            kr_text=translation.get("kr", ""),
+            pt_text=translation.get("pt", ""),
+            ru_text=translation.get("ru", ""),
+            th_text=translation.get("th", ""),
+            vi_text=translation.get("vi", "")
+        )
+        db.add(i18n_entry)
+
+        # Insert into GenericDict
+        for lang_code, text_value in translation.items():
+            if text_value:
+                gdict = GenericDict(
+                    game_id=game_id,
+                    item_id=item_id,
+                    text=text_value,  # Properly set "text"
+                    lang=lang_code  # And "lang"
+                )
+                db.add(gdict)
+
+    db.commit()

db/crud.py

@@ -0,0 +1,48 @@
+from .mysql_db import Base
+from sqlalchemy import Column, Integer, String, Date
+
+# ------------------------------------------------------------------------
+# MODEL DECLARATIONS
+# ------------------------------------------------------------------------
+
+
+class I18nDict(Base):
+    """
+    Reflects the i18n_dict table with the following columns:
+    game_id, item_id, chs_text, cht_text, de_text, en_text, es_text,
+    fr_text, id_text, jp_text, kr_text, pt_text, ru_text, th_text, vi_text
+    """
+    __tablename__ = "i18n_dict"
+
+    # auto increment primary key if you want
+    id = Column(Integer, primary_key=True, index=True)
+    game_id = Column(Integer, index=True, nullable=False)
+    item_id = Column(String(255), index=True, nullable=False)
+
+    chs_text = Column(String(255), default="")
+    cht_text = Column(String(255), default="")
+    de_text = Column(String(255), default="")
+    en_text = Column(String(255), default="")
+    es_text = Column(String(255), default="")
+    fr_text = Column(String(255), default="")
+    id_text = Column(String(255), default="")
+    jp_text = Column(String(255), default="")
+    kr_text = Column(String(255), default="")
+    pt_text = Column(String(255), default="")
+    ru_text = Column(String(255), default="")
+    th_text = Column(String(255), default="")
+    vi_text = Column(String(255), default="")
+
+
+class GenericDict(Base):
+    """
+    Reflects the generic_dict table with columns:
+    game_id, item_id, text, lang
+    """
+    __tablename__ = "generic_dict"
+
+    id = Column(Integer, primary_key=True, index=True)
+    game_id = Column(Integer, index=True, nullable=False)
+    item_id = Column(String(255), index=True, nullable=False)
+    text = Column(String(255), nullable=False)
+    lang = Column(String(16), nullable=False)

db/models.py

@@ -0,0 +1,22 @@
+import os
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from base_logger import logging
+import socket
+
+if os.getenv("host.docker.internal", None):
+    MYSQL_HOST = os.getenv("host.docker.internal")
+else:
+    MYSQL_HOST = socket.gethostbyname('host.docker.internal')
+MYSQL_PORT = int(os.getenv("MYSQL_PORT", "3306"))
+MYSQL_USER = os.getenv("MYSQL_USER")
+MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD")
+MYSQL_DATABASE = os.getenv("MYSQL_DATABASE")
+
+SQLALCHEMY_DATABASE_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DATABASE}?charset=utf8mb4"
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()
+logging.info(f"MySQL connection established to {MYSQL_HOST}/{MYSQL_DATABASE}")

db/mysql_db.py

@@ -0,0 +1,18 @@
+from pydantic import BaseModel
+from typing import Optional, Any
+
+
+# ------------------------------------------------------------------------
+# Pydantic Schemas for Requests/Responses
+# ------------------------------------------------------------------------
+class TranslateRequest(BaseModel):
+    type: str
+    lang: str
+    game: str
+    item_name: Optional[str] = None
+    item_id: Optional[str] = None
+
+
+class TranslateResponse(BaseModel):
+    item_id: Optional[Any] = None
+    item_name: Optional[Any] = None