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)
