def main():
logger.info("Test")
# f = open("quotes.json", "r")
# quotes = f.read()
# quotes = json.loads(quotes)
# for quote in quotes.values():
# logger.info(quote)
# dao.insert_quote(quote)
# f = open("bakchods.json", "r")
# quotes = f.read()
# quotes = json.loads(quotes)
# for quote in quotes:
# logger.info(quote)
# bakchod = Bakchod(quote["id"], None, None)
# bakchod.username = quote.get("username")
# bakchod.first_name = quote.get("first_name")
# bakchod.lastseen = quote.get("lastseen")
# bakchod.rokda = quote.get("rokda")
# bakchod.censored = quote.get("censored")
# bakchod.history = quote.get("history")
# dao.insert_bakchod(bakchod)
f = open("groups.json", "r")
quotes = f.read()
quotes = json.loads(quotes)
for quote in quotes:
logger.info(quote)
group = Group(None, None)
group.id = quote.get("id")
group.title = quote.get("title")
group.members = quote.get("members")
dao.insert_group(group)
def get_group_by_id(group_id):
group = None
try:
c.execute("""SELECT * FROM groups WHERE id=:id""", {"id": group_id})
query_result = c.fetchone()
if query_result is not None:
group = Group(query_result[0], query_result[1])
# Santize Group Members
members_str = []
members = json.loads(query_result[2])
for id in members:
if str(id) not in members_str:
members_str.append(str(id))
group.members = members_str
except Exception as e:
logger.error(
"Caught Error in dao.get_group_by_id - {} \n {}",
e,
traceback.format_exc(),
)
return group
def get_all_groups():
groups = []
try:
c.execute("""SELECT * FROM groups""")
query_results = c.fetchall()
if query_results is not None:
for query_result in query_results:
group = Group(query_result[0], query_result[1])
# Santize Group Members
members_str = []
members = json.loads(query_result[2])
for id in members:
if str(id) not in members_str:
members_str.append(str(id))
group.members = members_str
groups.append(group)
except Exception as e:
logger.error(
"Caught Error in dao.get_all_groups - {} \n {}",
e,
traceback.format_exc(),
)
return groups
