WIP -User management - users resource

.gitignore

@@ -1,5 +1,7 @@
-
-#Build result
-
 *.log
-_pycache_/
+/.vagrant/*
+*/.idea/*
+*.box
+*.pyc
+_pycache_/
+

Vagrantfile

@@ -1,11 +1,7 @@
-FLASK_PORT=5000
-POSTGRESQL_PORT=5432
-
 Vagrant.configure("2") do |config|
   config.vm.box = "ubuntu/bionic64"
   config.vm.provision :shell, path: "setup.sh"
-  config.vm.network :forwarded_port, guest: FLASK_PORT, host: FLASK_PORT
- config.vm.network :forwarded_port, guest: POSTGRESQL_PORT, host: POSTGRESQL_PORT
+  config.vm.network "private_network", type: "dhcp"
   config.vm.provider "virtualbox" do |v|
     v.gui = false
     v.name = "Edison_test"

edison/__init__.py

@@ -0,0 +1,12 @@
+import os
+
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from edison.config import get_config_object
+
+
+# Put app and db here so the entire app could import them.
+app = Flask(__name__)
+app.config.from_object(get_config_object(app.config["ENV"]))
+basedir = os.path.abspath(os.path.dirname(__file__))
+db = SQLAlchemy(app)

edison/app.py

@@ -0,0 +1,31 @@
+import edison
+from flask import render_template
+from flask_restful import Api
+from flask_jwt_extended import JWTManager
+from edison.resources import *
+import edison.models as models
+
+
+# API description in swagger - https://app.swaggerhub.com/apis/DoRTaL94/UserManagment/1.0.0
+
+app = edison.app
+# Creates all tables in app. Table's name defined with __tablename__ variable.
+edison.db.create_all()
+api = Api(app)
+# Creation of Json-Web-Token manager.
+# In order to reach secured endpoints client should add an authorization header with the value Bearer <token>.
+jwt = JWTManager(app)
+
+
+@jwt.token_in_blacklist_loader
+def check_if_token_in_blacklist(decrypted_token):
+    jti = decrypted_token['jti']
+    filters = {'jti': jti}
+    return models.Token.query.filter_by(**filters).first() is not None
+
+
+@app.route("/")
+def index():
+	return render_template('index.html')
+
+api.add_resource(Users, '/api/users')

edison/config.py

@@ -0,0 +1,39 @@
+import secrets
+import importlib, inspect
+import sys, inspect
+
+config_dict = {}
+
+# If config_dict is empty this function builds it dynamically 
+# and returns the appropriate config object path.
+def get_config_object(env_keyword: str):
+    if(len(config_dict) == 0):
+        # Iterating through all config.py members
+        for name, obj in inspect.getmembers(sys.modules[__name__]):
+            # We're interested only with the derived classes of the Config class
+            if inspect.isclass(obj) and name != "Config":
+                config_dict[obj.ENV_KEYWORD] = ".".join([obj.__module__, name])
+
+    return config_dict[env_keyword]
+
+class Config:
+    ENV_KEYWORD = ""
+    DEBUG = False
+    # Turn off the Flask-SQLAlchemy event system
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
+    # Enables response message for unauthenticated requests
+    PROPAGATE_EXCEPTIONS = True
+    # This tells the JWTManager to use jwt.token_in_blacklist_loader callback
+    JWT_BLACKLIST_ENABLED = True
+    # JWTManager uses this secret key for creating tokens
+    JWT_SECRET_KEY = secrets.token_hex(24)
+    # We're going to check if both access_token and refresh_token are black listed
+    JWT_BLACKLIST_TOKEN_CHECKS = ['access', 'refresh']
+    SQLALCHEMY_DATABASE_URI = 'postgresql://postgres:[email protected]/edison'
+
+class ProductionConfig(Config):
+    ENV_KEYWORD = "production"
+
+class DevelopmentConfig(Config):
+    ENV_KEYWORD = "development"
+    DEBUG = True

edison/models/__init__.py

@@ -0,0 +1,2 @@
+from .token import Token
+from .user import User

edison/models/token.py

@@ -0,0 +1,12 @@
+from edison import db
+
+
+class Token(db.Model):
+    __tablename__ = 'token_blacklist'
+
+    id = db.Column(db.Integer, primary_key=True)
+    jti = db.Column(db.String(150), nullable=False, unique=True)
+    creation_timestamp = db.Column(db.TIMESTAMP(timezone=False), nullable=False)
+
+    def __repr__(self):
+        return f"<Token: jti: {self.jti}, creation time: {self.creation_timestamp}>"

edison/models/user.py

@@ -0,0 +1,25 @@
+from edison import db
+
+
+class User(db.Model):
+    __tablename__ = 'users'
+    __table_args__ = {'extend_existing': True}
+
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(50), nullable=False, unique=True)
+    password = db.Column(db.String(150), nullable=False)
+    first_name = db.Column(db.String(50), nullable=False)
+    last_name = db.Column(db.String(50), nullable=False)
+    email = db.Column(db.String(150), nullable=False)
+
+    def to_json(self):
+        return {
+            "username": self.username,
+            "first_name": self.first_name,
+            "last_name": self.last_name,
+            "email": self.email
+        }
+
+    def __repr__(self):
+        return f"<User: id = {self.id}, first_name = {self.first_name}, " \
+               f"last_name = {self.last_name}, username = {self.username}, email = {self.email}>"

edison/resources/__init__.py

@@ -0,0 +1,4 @@
+from .user import User
+from .users import Users
+from .login import Login
+from .signup import SignUp

edison/resources/users.py

@@ -0,0 +1,17 @@
+from flask_restful import Resource
+from flask_jwt_extended import jwt_required
+import edison.models as models
+
+
+class Users(Resource):
+
+    @jwt_required
+    def get(self):
+        status = 200
+        response = list(
+            map(
+                lambda user: user.to_json(), models.User.query.all()
+            )
+        )
+
+        return response, status

requirements.txt

@@ -1 +1,6 @@
 Flask==1.1.1
+flask-restful==0.3.8
+flask-jwt-extended==3.24.1
+passlib==1.7.2
+flask-sqlalchemy==2.4.1
+psycopg2-binary==2.8.5

setup.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+FLASK_PORT=5000
+
 echo "updating apt before installation"
 sudo apt-get update
 
@@ -15,6 +17,15 @@ sudo apt-get install -y postgresql postgresql-contrib
 echo "install requirements"
 pip3 install -r /vagrant/requirements.txt
 
-echo "running flask_init.py"
-export FLASK_APP=/vagrant/flask_init.py
-python3 -m flask run --host=0.0.0.0 >> /vagrant/log.log 2>&1 &
+echo "configuring database"
+sudo su postgres <<POSTGRESQL_COMMANDS
+psql
+CREATE DATABASE edison;
+ALTER ROLE postgres WITH PASSWORD 'edison';
+POSTGRESQL_COMMANDS
+
+export FLASK_ENV=development
+
+echo "running app.py"
+export FLASK_APP=/vagrant/edison/app.py
+flask run -h 0.0.0.0 -p $FLASK_PORT >> /vagrant/edison/app.log 2>&1 &

tests/init_test.py

@@ -0,0 +1,11 @@
+import unittest
+import requests
+
+class Tester(unittest.TestCase):
+	def test_home(self):
+		response = requests.get("http://127.0.0.1:5000")
+		self.assertEqual(response.status_code, 200)
+
+
+if __name__ == '__main__':
+	unittest.main()