databaseUtils.py

import pymongo
from datetime import *
import time

#Remote declarations.
client = pymongo.MongoClient("mongodb://localhost:27017/")
print(client.list_database_names())
waifudb = client['waifu']
userdb = waifudb['user_data']
waifu_datadb = waifudb['waifu_data_db']
roll_statsdb = waifudb['roll_stats']
rate_limitdb = waifudb['rate_limits']
waifu_tierdb = waifudb['waifu_tier_db']

time_limit = 150

def waifu_roll_in_progress(user_id, is_in_progress):
    insertion = {
        "user_id" : user_id,
        "is_in_progress" : is_in_progress
    }
    
    #checking if the entry for the user already exists on DB
    if (len(list(roll_statsdb.find({"user_id" : user_id}))) == 0):
        roll_statsdb.insert_one(insertion)
    
    #otherwise, just finding and replacing
    else:
        roll_statsdb.find_one_and_replace({"user_id" : user_id}, insertion)
    
def is_waifu_roll_in_progress(user_id):
    query = {
        "user_id" : user_id
    }
    try:
        return roll_statsdb.find_one(query)["is_in_progress"]
    except:
        return False

def list_hentai():
    hentaidb = waifudb['hentaidigits']
    for x in hentaidb.find():
        print(x)

def insert_hentai(n):
    insertion = {"id" : n}
    hentaidb = waifudb['hentaidigits']
    hentaidb.insert_one(insertion)

def claim_waifu(user_id, anilist_id):
    insertion = {
        "user_id" : user_id,
        "anilist_id" : anilist_id
    }
    userdb.insert_one(insertion)

def remove_waifu(user_id, anilist_id):
    removal = {
        "user_id" : user_id,
        "anilist_id" : anilist_id
    }
    userdb.find_one_and_delete(removal)

def get_waifu_list_for_user(user_id):
    query = {
        'user_id' : user_id
    }
    return list(userdb.find(query))

def insert_waifu_data_into_db(anilist_id, name, image_url, anime_name):
    insertion = {
        'anilist_id' : anilist_id,
        'name' : name,
        'anime_name' : anime_name,
        'image_url' : image_url
    }
    if (len(list(waifu_datadb.find(insertion))) == 0):
        waifu_datadb.insert_one(insertion)

def get_waifu_data_from_db(anilist_id):
    query = {
        'anilist_id' : anilist_id
    }
    return waifu_datadb.find_one(query)

def set_roll_rate_limit(user_id):
    query = {
        "user_id" : user_id
    }
    insertion = {
        "user_id" : user_id,
        "time_rolled_last" : datetime.now()
    }
    
    if (len(list(rate_limitdb.find(query))) == 0):
        rate_limitdb.insert_one(insertion)
    
    else:
        rate_limitdb.find_one_and_replace(query, insertion)

def is_rate_limited(user_id):
    query = {
        "user_id" : user_id
    }
    try:
        rate_time = rate_limitdb.find_one(query)["time_rolled_last"]
    except:
        return False
    
    time_diff = int((datetime.now() - rate_time).total_seconds())
    if(time_diff< time_limit):
        return True
    
    else:
        return False

def get_time_left(user_id):
    query = {
        "user_id" : user_id
    }
    rate_time = rate_limitdb.find_one(query)["time_rolled_last"]
    time_diff = int((datetime.now() - rate_time).total_seconds())
    return str(time_limit-time_diff)

def reset_user_roll_stats():
    rate_limitdb.drop()
    roll_statsdb.drop()

def user_stats_from_db(user_id):
    query = {
        "user_id" : user_id
    }
    #TBD

def insert_waifu_pos_data(anilist_id, pos):
    insertion = {
        "anilist_id" : anilist_id,
        "pos" : pos
    }
    waifu_tierdb.insert_one(insertion)

def get_user_stats(user_id):
    query = {
        "user_id" : user_id
    }
    user_data = {"number_of_waifus" : len(list(userdb.find(query)))}