User Management backend - Base

.gitignore

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

Vagrantfile

@@ -1,11 +1,11 @@
-FLASK_PORT=5000
-POSTGRESQL_PORT=5432
+CLIENT_PORT=5000
+BE_PORT=3000
 
 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 :forwarded_port, guest: CLIENT_PORT, host: CLIENT_PORT
+  config.vm.network :forwarded_port, guest: BE_PORT, host: BE_PORT
   config.vm.provider "virtualbox" do |v|
     v.gui = false
     v.name = "Edison_test"

backend/__init__.py

@@ -0,0 +1,16 @@
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+import os
+
+
+db_role = 'postgres'
+db_password = 'edison'
+db_url = '0.0.0.0'
+db_name = 'edison'
+
+# Put app and db here so the entire app could import them.
+app = Flask(__name__)
+basedir = os.path.abspath(os.path.dirname(__file__))
+app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://{}:{}@{}/{}'.format(db_role, db_password, db_url, db_name)
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+db = SQLAlchemy(app)

backend/app.py

@@ -0,0 +1,22 @@
+from flask_restful import Api
+from flask_jwt_extended import JWTManager
+import secrets
+import backend
+
+# API description in swagger - https://app.swaggerhub.com/apis/DoRTaL94/UserManagment/1.0.0
+
+app = backend.app
+# Creates all tables in app. Table's name defined with __tablename__ variable.
+backend.db.create_all()
+# Enables response message for unauthenticated requests
+app.config['PROPAGATE_EXCEPTIONS'] = True
+# JWTManager uses this secret key for creating tokens
+app.config['JWT_SECRET_KEY'] = secrets.token_hex(24)
+# This tells the JWTManager to call 'check_if_token_in_blacklist' function below on every request
+app.config['JWT_BLACKLIST_ENABLED'] = True
+# We're going to check if both access_token and refresh_token are black listed
+app.config['JWT_BLACKLIST_TOKEN_CHECKS'] = ['access', 'refresh']
+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)

backend/models/__init__.py

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

backend/models/token.py

@@ -0,0 +1,12 @@
+from backend 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}."

backend/models/user.py

@@ -0,0 +1,25 @@
+from backend 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}>"

backend/services/base_storage_handler.py

@@ -0,0 +1,29 @@
+from abc import ABC, abstractmethod
+from backend import db
+from typing import Dict
+
+class BaseStorageHandler(ABC):
+
+    @abstractmethod
+    def get_by_filters(self, model: db.Model, filters: Dict[str, str]):
+        pass
+
+    @abstractmethod
+    def get_by_id(self, model: db.Model, _id: int):
+        pass
+
+    @abstractmethod
+    def get_all(self, model: db.Model):
+        pass
+
+    @abstractmethod
+    def add(self, model: db.Model):
+        pass
+
+    @abstractmethod
+    def delete(self, model: db.Model):
+        pass
+
+    @abstractmethod
+    def update(self):
+        pass

backend/services/db_handler.py

@@ -0,0 +1,27 @@
+from backend import db
+from datetime import datetime
+from backend.services.base_storage_handler import BaseStorageHandler
+from typing import Dict
+
+
+class DBHandler(BaseStorageHandler):
+
+    def get_by_filters(self, model: db.Model, filters: Dict[str, str]):
+        return model.query.filter_by(**filters).first()
+
+    def get_by_id(self, model: db.Model, _id: int):
+        return model.query.get(_id)
+
+    def get_all(self, model: db.Model):
+        return model.query.all()
+
+    def add(self, model: db.Model):
+        db.session.add(model)
+        db.session.commit()
+
+    def delete(self, model: db.Model):
+        db.session.delete(model)
+        db.session.commit()
+
+    def update(self):
+        db.session.commit()

requirements.txt

@@ -1 +1,6 @@
 Flask==1.1.1
+flask-restful
+flask-jwt-extended
+passlib
+flask-sqlalchemy
+psycopg2-binary

setup.sh

@@ -15,6 +15,20 @@ sudo apt-get install -y postgresql postgresql-contrib
 echo "install requirements"
 pip3 install -r /vagrant/requirements.txt
 
+echo "installing postgresql"
+sudo apt-get install -y postgresql postgresql-contrib
+
+echo "configuring database"
+sudo su postgres <<POSTGRESQL_COMMANDS
+psql
+CREATE DATABASE edison;
+ALTER ROLE postgres WITH PASSWORD 'edison';
+POSTGRESQL_COMMANDS
+
 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 &
+flask run -h 0.0.0.0 -p 5000 >> /vagrant/flask_init.log 2>&1 &
+
+echo "running app.py"
+export FLASK_APP=/vagrant/backend/app.py
+flask run -h 0.0.0.0 -p 3000 >> /vagrant/app.log 2>&1 &