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()
