py.lib

     1.1 --- a/db/migrator.py	Sun Jun 06 14:04:19 2021 +0300
     1.2 +++ b/db/migrator.py	Sun Jun 06 14:13:24 2021 +0300
     1.3 @@ -76,8 +76,8 @@
     1.4      def init_db(self, db):
     1.5          cursor = db.cursor()
     1.6          for c in self.get_commands(self.schema):
     1.7 -           cursor.execute(c)
     1.8 -           db.commit()
     1.9 +            cursor.execute(c)
    1.10 +            db.commit()
    1.11  
    1.12          db.commit()
    1.13  
    1.14 @@ -108,4 +108,17 @@
    1.15          """ % (self.control_table, new_ver))
    1.16          db.commit()
    1.17  
    1.18 +    @staticmethod
    1.19 +    def get_conn_from_my_obj(obj: object):
    1.20 +        """\
    1.21 +        Получиение объекта соединения из обёрток, которые я сам себе пишу для работы с DB-API
    1.22  
    1.23 +        :param obj:
    1.24 +        :return:
    1.25 +        """
    1.26 +        if hasattr(obj, '_conn'):
    1.27 +            return obj._conn
    1.28 +        elif hasattr(obj, '_con'):
    1.29 +            return obj._con
    1.30 +        else:
    1.31 +            raise TypeError('No known connection object in given database object found')