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