Exemple #1
0
 def cursor(self):
     global engine
     if self.connection is None:
         con = engine.connect()
         log.info('open connection <%s>...' % hex(id(connection)))
         self.connection = con
     return self.connection.cursor()
Exemple #2
0
 def cursor(self):
     global engine
     if self.connection is None:
         con = engine.connect()
         log.info('open connection <%s>...' % hex(id(connection)))
         self.connection = con
     return self.connection.cursor()
Exemple #3
0
 def __enter__(self):
     global _db_ctx
     self.should_close_conn = False
     if not _db_ctx.is_init():
         # needs open a connection first:
         _db_ctx.init()
         self.should_close_conn = True
     _db_ctx.transactions += 1
     log.info('begin transaction...' if _db_ctx.transactions == 1 else 'join current transaction...')
     return self
Exemple #4
0
 def commit(self):
     global _db_ctx
     log.info('commit transaction...')
     try:
         _db_ctx.connection.commit()
         log.info('commit ok.')
     except:
         log.warning('commit failed. try rollback...')
         _db_ctx.connection.rollback()
         log.warning('rollback ok.')
         raise
Exemple #5
0
 def __enter__(self):
     global _db_ctx
     self.should_close_conn = False
     if not _db_ctx.is_init():
         # needs open a connection first:
         _db_ctx.init()
         self.should_close_conn = True
     _db_ctx.transactions += 1
     log.info('begin transaction...' if _db_ctx.transactions ==
              1 else 'join current transaction...')
     return self
Exemple #6
0
 def commit(self):
     global _db_ctx
     log.info('commit transaction...')
     try:
         _db_ctx.connection.commit()
         log.info('commit ok.')
     except:
         log.warning('commit failed. try rollback...')
         _db_ctx.connection.rollback()
         log.warning('rollback ok.')
         raise
Exemple #7
0
    def __new__(cls, name, bases, attrs):
        if name == 'Model':
            return type.__new__(cls, name, bases, attrs)
        if not hasattr(cls, 'subclasses'):
            cls.subclasses = {}
        if name not in cls.subclasses:
            cls.subclasses[name] = name
        else:
            log.warning('Redefine class: %s' % name)

        log.info('Scan ORMapping %s...' % name)
        mappings = dict()
        primary_key = None
        for k, v in attrs.iteritems():
            if isinstance(v, Field):
                if not v.name:
                    v.name = k
                log.info('Found mapping: %s => %s' % (k, v))

                if v.primary_key:
                    if primary_key:
                        raise TypeError(
                            'Cannot define more than 1 primary key in class: %s'
                            % name)
                    if v.writable:
                        log.warning(
                            'NOTE: change primary key to non-writable.')
                        v.writable = False
                    if v.nullable:
                        log.warning(
                            'NOTE: change primary key to non-nullable.')
                        v.nullable = False
                    primary_key = v
                mappings[k] = v

        # check exist of primary key:
        if not primary_key:
            raise TypeError('Primary key not defined in class: %s' % name)
        for k in mappings.iterkeys():
            attrs.pop(k)
        if '__table__' not in attrs:
            attrs['__table__'] = name.lower()
        attrs['__mappings__'] = mappings
        attrs['__primary_key__'] = primary_key
        attrs['__sql__'] = lambda self: _gen_sql(attrs['__table__'], mappings)
        for trigger in _triggers:
            if trigger not in attrs:
                attrs[trigger] = None
        return type.__new__(cls, name, bases, attrs)
Exemple #8
0
def _update(sql, *args):
    global _db_ctx
    cursor = None
    sql = sql.replace('?', '%s')
    log.info('SQL: %s, ARGS: %s' % (sql, args))
    try:
        cursor = _db_ctx.connection.cursor()
        cursor.execute(sql, args)
        r = cursor.rowcount
        if _db_ctx.transactions == 0:
            # no transaction enviroment:
            log.info('auto commit')
            _db_ctx.connection.commit()
        return r
    finally:
        if cursor:
            cursor.close()
Exemple #9
0
def _update(sql, *args):
    global _db_ctx
    cursor = None
    sql = sql.replace('?', '%s')
    log.info('SQL: %s, ARGS: %s' % (sql, args))
    try:
        cursor = _db_ctx.connection.cursor()
        cursor.execute(sql, args)
        r = cursor.rowcount
        if _db_ctx.transactions == 0:
            # no transaction enviroment:
            log.info('auto commit')
            _db_ctx.connection.commit()
        return r
    finally:
        if cursor:
            cursor.close()
Exemple #10
0
    def __new__(cls, name, bases, attrs):
        if name == 'Model':
            return type.__new__(cls, name, bases, attrs)
        if not hasattr(cls, 'subclasses'):
            cls.subclasses = {}
        if name not in cls.subclasses:
            cls.subclasses[name] = name
        else:
            log.warning('Redefine class: %s' % name)

        log.info('Scan ORMapping %s...' % name)
        mappings = dict()
        primary_key = None
        for k, v in attrs.iteritems():
            if isinstance(v, Field):
                if not v.name:
                    v.name = k
                log.info('Found mapping: %s => %s' % (k, v))

                if v.primary_key:
                    if primary_key:
                        raise TypeError('Cannot define more than 1 primary key in class: %s' % name)
                    if v.writable:
                        log.warning('NOTE: change primary key to non-writable.')
                        v.writable = False
                    if v.nullable:
                        log.warning('NOTE: change primary key to non-nullable.')
                        v.nullable = False
                    primary_key = v
                mappings[k] = v

        # check exist of primary key:
        if not primary_key:
            raise TypeError('Primary key not defined in class: %s' % name)
        for k in mappings.iterkeys():
            attrs.pop(k)
        if '__table__' not in attrs:
            attrs['__table__'] = name.lower()
        attrs['__mappings__'] = mappings
        attrs['__primary_key__'] = primary_key
        attrs['__sql__'] = lambda self: _gen_sql(attrs['__table__'], mappings)
        for trigger in _triggers:
            if trigger not in attrs:
                attrs[trigger] = None
        return type.__new__(cls, name, bases, attrs)
Exemple #11
0
def _select(sql, first, *args):
    global _db_ctx
    cursor = None
    names = []
    sql = sql.replace('?', '%s')
    log.info('SQL: %s, ARGS: %s', sql, args)
    try:
        cursor = _db_ctx.connection.cursor()
        cursor.execute(sql, args)
        if cursor.description:
            names = [x[0] for x in cursor.description]
        if first:
            values = cursor.fetchone()
            if not values:
                return None
            return Dict(names, values)
        return [Dict(names, x) for x in cursor.fetchall()]
    finally:
        if cursor:
            cursor.close()
Exemple #12
0
def _select(sql, first, *args):
    global _db_ctx
    cursor = None
    names = []
    sql = sql.replace('?', '%s')
    log.info('SQL: %s, ARGS: %s', sql, args)
    try:
        cursor = _db_ctx.connection.cursor()
        cursor.execute(sql, args)
        if cursor.description:
            names = [x[0] for x in cursor.description]
        if first:
            values = cursor.fetchone()
            if not values:
                return None
            return Dict(names, values)
        return [Dict(names, x) for x in cursor.fetchall()]
    finally:
        if cursor:
            cursor.close()
Exemple #13
0
 def init(self):
     log.info('open lazy connection...')
     self.connection = _LasyConnection()
     self.transactions = 0
Exemple #14
0
 def cleanup(self):
     if self.connection:
         con = self.connection
         self.connection = None
         log.info('close connection <%s>...' % hex(id(con)))
         con.close()
Exemple #15
0
 def rollback(self):
     global _db_ctx
     log.warning('rollback transaction...')
     _db_ctx.connection.rollback()
     log.info('rollback ok.')
Exemple #16
0
def profiling(start, sql=''):
    t = time.time() - start
    if t > 0.1:
        log.warning('[PROFILING] [DB] %s: %s', t, sql)
    else:
        log.info('[PROFILING] [DB] %s: %s', t, sql)
Exemple #17
0
 def rollback(self):
     global _db_ctx
     log.warning('rollback transaction...')
     _db_ctx.connection.rollback()
     log.info('rollback ok.')
Exemple #18
0
 def init(self):
     log.info('open lazy connection...')
     self.connection = _LasyConnection()
     self.transactions = 0
Exemple #19
0
 def cleanup(self):
     if self.connection:
         con = self.connection
         self.connection = None
         log.info('close connection <%s>...' % hex(id(con)))
         con.close()
Exemple #20
0
def profiling(start, sql=''):
    t = time.time() - start
    if t > 0.1:
        log.warning('[PROFILING] [DB] %s: %s', t, sql)
    else:
        log.info('[PROFILING] [DB] %s: %s', t, sql)