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