Fetching translation strings and constants from tournament DB

2 files changed, 37 insertions, 1 deletions

diff --git a/ausbutler/model.py b/ausbutler/model.py
index 6536b6a..fc088fa 100644
--- a/ausbutler/model.py
+++ b/ausbutler/model.py
@@ -1,7 +1,7 @@
 from cached_property import cached_property
 from sqlalchemy import Column, MetaData, Table, func
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.types import Float, Integer
+from sqlalchemy.types import Float, Integer, String
 
 from .db import get_session
 
@@ -72,3 +72,13 @@ class Segment(Base):
     @cached_property
     def butler_count(self):
         return Segment.count_cache[(self.rnd, self.segment, self.tabl)]
+
+class Translation(Base):
+    __table__ = Table('logoh', MetaData(bind=session.bind),
+                      Column('id', Integer, primary_key=True),
+                      autoload=True)
+
+class Admin(Base):
+    __table__ = Table('admin', MetaData(bind=session.bind),
+                      Column('shortname', String, primary_key=True),
+                      autoload=True)
diff --git a/ausbutler/tour_config.py b/ausbutler/tour_config.py
new file mode 100644
index 0000000..43a8dc7
--- /dev/null
+++ b/ausbutler/tour_config.py
@@ -0,0 +1,26 @@
+from .config import load_config
+from .db import get_session
+from .model import Translation, Admin
+
+session = get_session()
+
+class Translations(object):
+
+    translation_cache = { t.id: t.dane for t
+                          in session.query(Translation).all() }
+    translation_mapping = load_config('logoh')
+    custom_translations = load_config('translations')
+
+    @staticmethod
+    def get_translation(key):
+        if key in Translations.translation_mapping:
+            return Translations.translation_cache[Translations.translation_mapping[key]]
+        if key in Translations.custom_translations:
+            return Translations.custom_translations[key][Translations.detect_language()]
+        return '{{%s}}' % (key)
+
+    @staticmethod
+    def detect_language():
+        return 'en' if Translations.get_translation('ROUND').lower().strip() == 'round' else 'pl'
+
+Constants = session.query(Admin).one()