def login():
req_data = request.get_json()
msg = ""
if not req_data['username'] or not req_data['password']:
msg = {"status": {"type": "failure", "message": "somethings is wrong"}}
return jsonify(msg), 200
user = User(req_data['username'], req_data['password'], None, None, None)
if user.check_username():
if user.check_password(req_data['password']):
token_data = {'username': user.username, 'admin': user.admin}
token = jwt.encode(token_data, app.config['SECRET_KEY'])
return jsonify({
'token': token.decode('UTF-8'),
"username": user.username,
"status": {
"type": "success"
}
}), 200
else:
msg = {"status": {"type": "failure", "message": "wrong password"}}
return jsonify(msg), 200
else:
msg = {"status": {"type": "failure", "message": "user not found"}}
return jsonify(msg), 200
def register():
username = request.form.get('username')
password = request.form.get('password')
firstname = request.form.get('firstname')
lastname = request.form.get('lastname')
ext = request.form.get('ext')
admin = False
msg = ""
if not username or not password:
msg = {"status": {"type": "failure", "message": "somethings is wrong"}}
return jsonify(msg), 201
if 'file' not in request.files:
msg = {"status": {"type": "failure", "message": "no file part"}}
return jsonify(msg), 201
user = User(username=username,
password=password,
admin=admin,
firstname=firstname,
lastname=lastname)
file = request.files['file']
if file.filename == '':
return jsonify({"msg": "no file selected"}), 201
found = user.check_username()
if not found:
os.chdir(app.config['UPLOAD_FOLDER'])
msg = {"status": {"type": "success", "message": "username available"}}
user.set_password(password)
user.insert()
try:
os.mkdir("{foldername}".format(foldername=username))
except OSError:
print("fail")
if file and allowed_file(file.filename):
_dir = os.path.join(app.config['UPLOAD_FOLDER'],
"{foldername}".format(foldername=username))
file_count = len(os.listdir(_dir)) + 1
filename = "{0}_{1}.{2}".format(username, file_count, ext)
file.save(os.path.join(_dir, filename))
return jsonify(msg), 200
else:
msg = {"status": {"type": "failure", "message": "user already exist"}}
return jsonify(msg), 404
def register():
req_data = request.get_json()
username = req_data['username']
password = req_data['password']
admin = False
msg = ""
if not username or not password:
msg = {"status": {"type": "failure", "message": "somethings is wrong"}}
return jsonify(msg), 200
user = User(username=username, password=password, admin=admin)
found = user.check_username()
if not found:
msg = {"status": {"type": "success", "message": "username available"}}
user.set_password(password)
user.insert()
return jsonify(msg), 200
else:
msg = {"status": {"type": "failure", "message": "user already exist"}}
return jsonify(msg), 401
