[#H24] Flask app

Shops/sport_atribute_shop/flask_app/config.py

@@ -0,0 +1,8 @@
+from flask import Flask
+from storage import Storage
+
+storage = Storage("storage.json")
+app = Flask(__name__)
+
+
+secret_key = ''

Shops/sport_atribute_shop/flask_app/run.py

@@ -0,0 +1,6 @@
+from views import app
+
+
+
+if __name__ == "__main__":
+    app.run(debug=True, host='0.0.0.0', port=5000)

Shops/sport_atribute_shop/flask_app/service.py

@@ -0,0 +1,37 @@
+import hashlib
+from config import storage, secret_key
+import jwt
+
+
+def hash_md5(data):
+    return hashlib.md5(data.encode('utf-8')).hexdigest()
+
+
+def registrate(account):
+    result = {
+        "error": True,
+        "message": "User already exists."
+    }
+    account['password'] = hash_md5(account.get('password'))
+    email = account.get("email")
+    storage_account = storage.get_by_email(email)
+    if storage_account is None:
+        storage.add(account)
+        result = {
+            'error': False,
+            'message': "OK!"
+        }
+    return result
+
+
+def login(account):
+    email = account.get("email")
+    data = storage.get_by_email(email)
+    password = hash_md5(account.get("password"))
+    saved_password = data.get("password")
+    if password == saved_password:
+        return make_jwt(data)
+
+
+def make_jwt(data):
+    return jwt.encode(data, secret_key, algorithm='HS256')

Shops/sport_atribute_shop/flask_app/storage.json

@@ -0,0 +1 @@
+[]

Shops/sport_atribute_shop/flask_app/storage.py

@@ -0,0 +1,27 @@
+import json 
+
+
+class Storage:
+
+    def __init__(self, path):
+        self.path = path
+
+    def get(self, *args, **kwargs):
+        with open (self.path, "r") as file:
+            data = json.load(file)
+        return data
+
+    def get_by_email(self, email):
+        data = self.get()
+        result = list(filter(lambda acc: acc.get("email") == email, data))
+        if not result:
+            result = None
+        else:
+            result = result[0]
+        return result
+
+    def add(self, data):
+        storage_data = self.get()
+        storage_data.append(data)
+        with open(self.path, "w") as file:
+            json.dump(storage_data, file)

Shops/sport_atribute_shop/flask_app/views.py

@@ -0,0 +1,29 @@
+from config import app
+from flask import request, jsonify, make_response
+from service import registrate, login as log
+
+
+@app.route('/login', methods=["POST"])
+def login():
+    data = request.get_json()
+    try:
+        token = log(data)
+        result = ({"token": token,
+                  "status": "ok"}, 200)
+    except Exception as e:
+        result = ({
+            "status": "error",
+            "detail": "Passwords or email didn't match.",
+        }, 400)
+    return make_response(jsonify(result))
+
+
+@app.route('/reg', methods=["POST"])
+def reg():
+    data = request.get_json()
+    code = 200
+    result = registrate(data)
+    if result.get('error'):
+        code = 422
+    return make_response(jsonify({"message": result.get("message")})), code
+