def setUp(self): self.con = connect_db(config.TEST_DATABASE_URI) self.assertIsInstance(self.con, sqlite3.Connection) self.cur = self.con.cursor() self.assertIsInstance(self.cur, sqlite3.Cursor) r = init_stat(self.cur) self.assertTrue(r)
for row in rows: set_stat(cur, row[0], row[1]) if not check_passwd(row[2]): logs[row[1]] = row[0] save_log(logs) if __name__ == '__main__': start = timer() try: parser = argparse.ArgumentParser(description='Check user data.') parser.add_argument('--stats', action='store_true', help='show stats') args = parser.parse_args() con = connect_db() cur = con.cursor() if args.stats: stats = get_stats(cur) for stat in stats: print(stat[0], stat[1]) con.close() else: init_stat(cur) main(cur) con.commit() except KeyboardInterrupt: print("Import data interrupted", file=sys.stderr) end = timer() print ("Execution time:", end - start)