def get(self):
token = request.args.get('token')
if token == None:
return "ERROR 401: Failed to locate token", 401
if not verify_auth_token(token):
return "ERROR 401: Failed to authenticate, check validity or token has expired", 401
fileI = open('list.csv', 'w')
fileO = csv.writer(fileI)
temp = request.args.get('top')
if temp != None:
top = int(temp)
_items = db.tododb.find().limit(top)
else:
_items = db.tododb.find()
items = [item for item in _items]
fileO.writerow(['km\t' + 'Close Time'])
for i in items:
fileO.writerow([i['kming'] + '\t' + i['closing']])
fileR = open('list.csv', 'r')
return Response(fileR, mimetype='text/csv')
def get(self):
openT = []
closeT = []
#check if token was given, then verify
token = request.args.get('token')
if token == None:
return "ERROR 401: Failed to locate token", 401
if not verify_auth_token(token):
return "ERROR 401: Failed to authenticate, check validity or token is expired", 401
temp = request.args.get('top')
if temp != None:
top = int(temp)
_items = db.tododb.find().limit(top)
else:
_items = db.tododb.find()
items = [item for item in _items]
for i in items:
openT.append(i['opening'])
closeT.append(i['closing'])
results = {'opening': openT, 'closing': closeT}
return results
def DNRW(ip):
token_cred = False
ip_cred = False
if session.get('token'):
if testToken.verify_auth_token(session['token']):
token_cred = True
_items = db.user_db.find()
for user in _items:
if ip == user['ip']:
if user['token']:
if testToken.verify_auth_token(user['token']):
ip_cred = True
else:
app.logger.debug("Too many users on same IP")
break
return token_cred, ip_cred
def get(self, token):
if (testToken.verify_auth_token(token)):
_items = db.tododb.find()
items = [item for item in _items]
all_list4 = []
for time in items:
all_list4.append(time['open'])
all_list4.append(time['close'])
all_list4.sort()
all_string = ''
for i in all_list4:
all_string += str(i) + ','
all_string = all_string[0:-1]
return all_string
else:
return "Wrong", 401
def get(self):
is_auth = None
if request.authorization:
cur_user = request.authorization['username']
cur_pass = request.authorization['password']
is_auth = check_auth(cur_user, cur_pass)
elif session.get('token') and testToken.verify_auth_token(
session['token']):
t = testToken.generate_auth_token(600)
session['token'] = t
return t.decode("utf-8")
else:
return "This page is for authorized users only.", 401
if is_auth:
if is_auth['auth']:
if session.get('token'):
session.pop('token')
t = testToken.generate_auth_token(600)
session['token'] = t
ret_token = {'duration': '600', 'token': t.decode("utf-8")}
# Store that token for user
_users = db.user_db.find()
for user in _users:
if cur_user == user['username']:
db.user_db.update_one({'username': cur_user},
{"$set": {
'token': t
}})
break
return ret_token
else:
return is_auth['message'], 401
else:
if is_auth:
return is_auth['message'], 401
else:
return "This page is for authorized users only.", 401
def setup():
if session.get('token'):
if testToken.verify_auth_token(session['token']):
return render_template('setup.html')
else:
return redirect(url_for('nope'))
def index():
if session.get('token'):
if testToken.verify_auth_token(session['token']):
return render_template('index.html')
else:
return render_template('nope.html')
def has_valid_token(self):
dbuser = Userdb.todouserdb.find_one({"id": self.id})
if verify_auth_token(dbuser['token']) == 'Success':
return True
else:
return False
