def clear():
connection = DBConnect()
try:
cursor = connection.cursor()
queries = Query.truncate_query(Tables.USER_TABLE_NAME, Tables.USER_FOLLOW_TABLE_NAME, Tables.THREAD_TABLE_NAME,
Tables.THREAD_VOTES_TABLE_NAME, Tables.THREAD_SUBSCRIBES_TABLE_NAME,
Tables.POST_TABLE_NAME, Tables.POST_VOTES_TABLE_NAME, Tables.FORUM_TABLE_NAME)
for query in queries:
cursor.execute(query)
except Exception as e:
return {'code': Codes.UNKNOWN_ERROR, 'response': str(e)}
finally:
connection.close()
return {'code': Codes.OK, 'response': 'OK'}
def status():
connection = DBConnect()
try:
cursor = connection.cursor()
result = {}
tables = [Tables.USER_TABLE_NAME, Tables.THREAD_TABLE_NAME, Tables.FORUM_TABLE_NAME, Tables.POST_TABLE_NAME]
for table in tables:
query = Query.select.format(columns='count(*)', table=table)
cursor.execute(query)
result[table] = cursor.fetchone()[0]
except Exception as e:
return {'code': Codes.UNKNOWN_ERROR, 'response': str(e)}
finally:
connection.close()
return {'code': Codes.OK, 'response': result}
