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 dropdb(): drop_tables() click.echo('Database dropped')
def dropdb(): drop_tables() click.echo("Database dropped !")
import models models.drop_tables()