def initdb(create=False, drop=False, force=False, audit=True, verbose=False):
"""Initialize the database."""
if drop and force:
models.drop_tables()
if verbose:
logger = logging.getLogger("peewee")
if logger:
logger.setLevel(logging.DEBUG)
logger.addHandler(logging.StreamHandler())
try:
models.create_tables()
except Exception:
app.logger.exception("Failed to create tables...")
if audit:
app.logger.info("Creating audit tables...")
models.create_audit_tables()
super_user, created = models.User.get_or_create(
email="*****@*****.**", roles=models.Role.SUPERUSER)
if not created:
return
super_user.name = "The University of Auckland"
super_user.confirmed = True
super_user.save()
org, _ = models.Organisation.get_or_create(
name="The University of Auckland", tuakiri_name="University of Auckland", confirmed=True)
def tearDown(self):
#pass
self.session.commit()
self.session.close()
self.engine.execute(text('drop table crime cascade;'))
self.engine.execute(text('drop table crime_type cascade;'))
self.engine.execute(text('drop table week cascade;'))
self.engine.execute(text('drop table zip cascade;'))
drop_tables(self.engine)
def test_db(self):
"""tests weather data can be stored in an empty database"""
print("testing database operations")
models.drop_tables()
models.create_tables()
lottery = models.Lottery.create(name="test")
self.assertIsNotNone(lottery)
owner = models.Participant.create(name="test")
self.assertIsNotNone(owner)
ticket = models.Ticket.create(lottery=lottery, owner=owner)
self.assertIsNotNone(ticket)
def init_test_data(self, create=True, drop=True):
from models import create_tables, drop_tables
if drop:
logging.info("drop table ...")
drop_tables()
if create:
logging.info("create table ...")
create_tables()
logging.info("init test data ...")
from tests import init_data
init_data.init_all()
def validate_args():
if args.clear_db:
drop_tables(db)
create_tables(db)
log.info("Drop and create complete.")
exit(0)
if args.list:
print "--- Authorization keys: ---"
query = Authorizations.select(Authorizations.token,
Authorizations.name)
for token in query:
print ("Token: " + token.token + "\tAssigned: " +
token.name)
exit(0)
if args.generate:
query = Authorizations.select(Authorizations.token).where(
Authorizations.name ==
args.generate)
if len(query):
print "Token already exists for " + args.generate
exit(0)
print "Generating authorization token for " + args.generate
lst = [random.choice(string.ascii_letters + string.digits)
for n in xrange(32)]
token = "".join(lst)
authorization = {'token': token, 'name': args.generate}
bulk_upsert(Authorizations, {0: authorization}, db)
print ("Authorization token for " + args.generate + " is: " +
token)
exit(0)
if args.revoke:
query = Authorizations.delete().where(
Authorizations.token ==
args.revoke)
if query.execute():
print "Token revoked."
else:
print "No token found."
exit(0)
if args.no_gyms:
args.no_gymdetail = True
args.ignore_pokemon = Set([int(i) for i in args.ignore_pokemon])
def tearDown(self):
self.db.close()
models.drop_tables(self.app.engine)
def fin(): # 销毁函数, 测试函数退出时执行
drop_tables()
location=url,
title=soup.title.text if soup.title is not None else ''
)
code_nodes = soup.select(selector)
for node in code_nodes:
save_snippet(node, page)
page_index += 1
pbar.update(page_index)
pbar.finish()
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Fetch snippets and save code and text")
args = parser.parse_args()
if Page.table_exists():
print "Data exists! Truncate tables to continue? (y/n): ",
decision = raw_input()
if decision == 'y':
print "Truncating tables."
models.drop_tables()
else:
raise SystemExit("Leaving tables alone. Now exiting.")
models.create_tables()
main()
def drop_tables(self):
engine = db_engine_connect()
drop_tables(engine)
def teardown():
drop_tables()
def teardown():
drop_tables()
def dropdb():
drop_tables()
click.echo('Database dropped')
def dropdb():
drop_tables()
click.echo("Database dropped !")
import models models.drop_tables()
