def initDB(): pydo.initAlias('littlewiki', 'sqlite3', 'littlewiki.db') if not os.path.exists('littlewiki.db'): conn=pydo.getConnection('littlewiki') cursor=conn.cursor() cursor.execute(CREATE_SQL) conn.commit()
def initDB(): pydo.initAlias('fumanchu', 'sqlite3', dict(database='hit.db'), pool=False) if not os.path.exists('hit.db'): create_sql="CREATE TABLE hits (id INTEGER NOT NULL PRIMARY KEY, hit_at TIMESTAMP, remote_ip VARCHAR(15))" dbi=pydo.getConnection('fumanchu') c=dbi.cursor() c.execute(create_sql) c.close() dbi.commit()
def test_guess_columns1(): create="""CREATE TABLE test_guess_columns1 ( id %s, x INTEGER NOT NULL UNIQUE, y INTEGER NOT NULL, z INTEGER UNIQUE, r1 INTEGER NOT NULL, r2 INTEGER NOT NULL, UNIQUE (r1, r2) )""" % get_sequence_sql() db=P.getConnection('pydotest') c=db.cursor() c.execute(create) try: class testclass(P.PyDO): guess_columns=True table='test_guess_columns1' connectionAlias='pydotest' cols=testclass.getColumns() uniq=testclass.getUniquenessConstraints() seq=testclass.getSequences() assert set(cols)==set(('id', 'x', 'y', 'z', 'r1', 'r2')) assert seq.keys()==['id'] assert uniq==frozenset((frozenset(('r1', 'r2')), frozenset(('x',)), frozenset(('id',)))) finally: if db.autocommit: c.execute('drop table test_guess_columns1') else: db.rollback() c.close()
def rollbackConnection(*args, **kwargs): db=pydo.getConnection('littlewiki', False) if db: log.debug("rolling back littlewiki connection") db.rollback()
def rollbackConnection(*args, **kwargs): db=pydo.getConnection('hitcounter', False) if db: log.debug("rolling back hitcounter connection") db.rollback()