def verify_password(user_name, password):
if not user_exist(user_name):
return False
hashed_pw = users.find({
"user_name": user_name
})[0]["password"]
if bcrypt.hashpw(password.encode('utf8'), hashed_pw) == hashed_pw:
return True
else:
return False
def post(self):
posted_data = request.get_json()
user_name = posted_data["user_name"]
password = posted_data["password"]
ur = posted_data["url"]
return_json, error = verify_credentials(user_name, password)
print(return_json)
if error:
return return_json
tokens = users.find({
"user_name": user_name
})[0]["token"]
print("\n")
print("\n")
print(tokens)
print("\n")
if tokens <= 0:
return generate_return_dictionary(303, "Not Enough Tokens ! ")
r = requests.get(ur)
return_json = {}
with open("temp.jpg", "wb") as f:
f.write(r.content)
proc = subprocess.Popen(
"python classify_image.py --model_dir=. --image_file=temp.jpg", shell=True)
proc.communicate()[0]
proc.wait()
with open("text.txt") as g:
return_json = json.load(g)
users.update({
"user_name": user_name
}, {
"$set": {
"token": tokens-1
}
})
return return_json
def user_exist(user_name):
if users.find({"user_name": user_name}).count() == 0:
return False
else:
return True
def count_tokens(user_name):
tokens = users.find({
"user_name": user_name
})[0]["token"]
return tokens