wip

Author: rozen03

bot_logic.py

@@ -20,6 +20,7 @@
 )
 from typing import Dict, Final
 
+import models
 # Local imports
 from models import (Session, Command, Grupo, GrupoOptativa, ECI, GrupoOtros,
                     Obligatoria, Optativa, Otro, File, Listable, Noticia)
@@ -42,7 +43,9 @@
 @contextmanager
 def get_session():
     """Provide a transactional scope around a series of operations."""
-    session = Session()
+    if models.Session is None:
+        models.init_db()
+    session = models.Session()
     try:
         yield session
         session.commit()

main.py

@@ -27,8 +27,7 @@ def main():
         level=logging.INFO,
         format='[%(asctime)s] - [%(name)s] - [%(levelname)s] - %(message)s',
     )
-    
-    init_db()
+
 
     application = Application.builder().token(os.environ["TELEGRAM_BOT_TOKEN"]).build()
 

models.py

@@ -59,37 +59,37 @@ class Listable(Base):
 
 class Obligatoria(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'obligatoria',
+        'polymorphic_identity': 'Obligatoria',
     }
 
 class Optativa(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'optativa',
+        'polymorphic_identity': 'Optativa',
     }
 
 class ECI(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'eci',
+        'polymorphic_identity': 'ECI',
     }
 
 class Otro(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'otro',
+        'polymorphic_identity': 'Otro',
     }
 
 class Grupo(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'grupo',
+        'polymorphic_identity': 'Grupo',
     }
 
 class GrupoOptativa(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'grupo_optativa',
+        'polymorphic_identity': 'GrupoOptativa',
     }
 
 class GrupoOtros(Listable):
     __mapper_args__ = {
-        'polymorphic_identity': 'grupo_otros',
+        'polymorphic_identity': 'GrupoOtros',
     }
 
 class Noticia(Base):