kubic_user.py

from flask import session
from pymongo import MongoClient
from bson.objectid import ObjectId
from secrets import token_urlsafe 
from passlib.hash import pbkdf2_sha512
from datetime import datetime
from dateutil.relativedelta import relativedelta
#import logging

client = MongoClient('localhost',27017)
db = client.user
trafficLimit = 3000

# def getEmail():
    # email_logined = "21800409@handong.edu"
    
    # email_logined = session['id']
    # return email_logined
    # return session['id']

# email_logined = getEmail()

def countAPI():
    count = db.apiUser.count({"user_email": session['id']})
    return count

def generateCode(givenCode):
    if givenCode==None :
        key = token_urlsafe(16)
    else: 
        key = givenCode
    
    hashKey = pbkdf2_sha512.hash(key)
    return key, hashKey

def registerAPI(app_type, app_name, app_purpose):
    today = datetime.today()
    key, hashKey = generateCode()

    post = {
        "app_type": app_type,
        "app_name" : app_name,
        "app_purpose" : app_purpose,
        "user_email" : session['id'],
        "veri_code" : hashKey,
        "reporting_date" : today,
        "expiration_date" : (today+relativedelta(years=1)),
        "traffic":0
    }

    db.apiUser.insert_one(post)
    return key

def preRegisterAPI(email, app_name, app_purpose):
    today = datetime.today()
    # key, hashKey = generateCode()
    key = token_urlsafe(16)
    post = {
        # "app_type": app_type,
        "app_name" : app_name,
        "app_purpose" : app_purpose,
        "email" : email,
        "key_id" : key,
        "reporting_date" : today,
        # "expiration_date" : (today+relativedelta(years=1)),
        # "traffic":0
    }

    db.preApiUser.insert_one(post)
    return key  

def getPreuserInfoByKey(key):
    if key == None: return None, None, None

    doc = db.preApiUser.find_one({"key_id": key})
    if doc != None:
        email = doc['email']
        app_name = doc['app_name']
        app_purpose = doc['app_purpose']
        return email, app_name, app_purpose
    return None, None, None

def getDocListPreUser():
    docList = db.preApiUser.find({"accept": {'$ne': 1}})
    return docList

def updatePreuserInfoByKey(key, accept, reason):
    post = {
        'accept': accept,
        'reason': reason
        }
    db.preApiUser.update({"key_id": key}, {'$set': post})

def reissueAPI(_id):
    today = datetime.today()
    key, hashKey = generateCode()

    post = {
        # "app_name" : "testtesttest",
        "veri_code" : hashKey,
        "reporting_date" : today,
        "expiration_date" : (today+relativedelta(years=1)),
        }

    db.apiUser.update({"_id": ObjectId(_id)}, {'$set': post})
    # print("reissue> app_name", post['app_name'],"key", key)
    return key

def deleteAPI(_id):
    try: db.apiUser.remove({"_id": ObjectId(_id)})
    except: return False
    return True

def getDocByEmail():
    docList = db.apiUser.find({"user_email": session['id']})
    return docList

def getDocById(_id):
    doc = db.apiUser.find_one({"_id": ObjectId(_id)})
    return doc

# def findHash():
#     doc = getDocByEmail()
#     hashKeyList = [item['veri_code'] for item in doc]
#     return hashKeyList

def verification(serviceKey):
    docList = db.apiUser.find({},{"veri_code":1})
    # print(hashKeyList)
    # hashKey=pbkdf2_sha512.hash(serviceKey)
    for doc in docList:
        if(pbkdf2_sha512.verify(serviceKey, doc['veri_code'])):
            # doc = db.apiUser.find_one({"veri_code": hashKey})
            return doc['_id']
    # hashKeyList = findHash()
    # for hashKey in hashKeyList:
    #     if(pbkdf2_sha512.verify(serviceKey, hashKey)):
    #         doc = db.apiUser.find_one({"veri_code": hashKey})
    #         return doc['_id']
    return False

def limitTraffic(_id):
    # if(ObjectId(_id)==ObjectId('606a98e2d7f9d800f11f17bc')): return True
    doc = getDocById(_id)
    if doc['traffic'] > trafficLimit:
        return False
    return True

def limitDate(_id):
    doc = getDocById(_id)
    if doc['expiration_date'] < datetime.today():
        return False
    return True

def raiseTraffic(_id, numOfCnt):
    doc = getDocById(_id)
    post = {"traffic" : doc['traffic']+numOfCnt}
    db.apiUser.update({"_id": ObjectId(_id)}, {'$set':post})
    doc = getDocById(_id)
    print(doc)

def getAppType(_id):
    doc = getDocById(_id)
    return doc['app_type']

def getMyDocByEmail():
    doc = db.mydocs.find_one({"userEmail": session['id']})
    print(doc) 
    print("myDoc:", doc['savedDocIds'])
    return doc['savedDocIds']