From d7bf281f61a7d36577ab8d3e00d2cdc2754361ed Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Sun, 31 Mar 2019 19:15:22 +0200
Subject: [PATCH] Create an async contextmanager for Alchemy

---
 royalnet/database/alchemy.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/royalnet/database/alchemy.py b/royalnet/database/alchemy.py
index f1187e45..b1b7b07d 100644
--- a/royalnet/database/alchemy.py
+++ b/royalnet/database/alchemy.py
@@ -1,19 +1,40 @@
 import typing
+import asyncio
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
+from contextlib import asynccontextmanager
 from ..utils import classdictjanitor
 
+loop = asyncio.get_event_loop()
+
 
 class Alchemy:
     def __init__(self, database_uri: str = "sqlite://", tables: typing.Optional[typing.List] = None):
         self.engine = create_engine(database_uri)
         self.Base = declarative_base(bind=self.engine)
-        self.Session = sessionmaker(bind=self.engine)
+        self._Session = sessionmaker(bind=self.engine)
         self._create_tables(tables)
 
     def _create_tables(self, tables: typing.Optional[typing.List]):
         for table in tables:
             name = table.__name__
-            self.__setattr__(name, type(name, (self.Base,), classdictjanitor(table)))
+            try:
+                self.__getattribute__(name)
+            except AttributeError:
+                # Actually the intended result
+                self.__setattr__(name, type(name, (self.Base,), classdictjanitor(table)))
+            else:
+                raise NameError(f"{name} is a reserved name and can't be used as a table name")
         self.Base.metadata.create_all()
+
+    @asynccontextmanager
+    async def Session(self):
+        session = await loop.run_in_executor(None, self._Session)
+        try:
+            yield session
+        except Exception:
+            session.rollback()
+            raise
+        finally:
+            session.close()