Merge pull request #1 from pandermatt/feature/add-wordcloud

Author: pandermatt

.gitignore

@@ -127,3 +127,7 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+application.yml
+data/result
+application.yml

api/auth.py

@@ -0,0 +1,19 @@
+from flask_httpauth import HTTPTokenAuth
+
+from api.error_handler import unauthorized_response
+from config import config
+
+token_auth = HTTPTokenAuth()
+
+
+@token_auth.verify_token
+def verify_token(token):
+    try:
+        return token == config.get_env('API_TOKEN')
+    except KeyError:
+        token_auth_error()
+
+
+@token_auth.error_handler
+def token_auth_error():
+    return unauthorized_response()

api/error_handler.py

@@ -0,0 +1,8 @@
+from werkzeug.exceptions import BadRequest, Unauthorized, NotFound
+
+from util.logger import log
+
+
+def unauthorized_response():
+    log.error(f'Unauthorized')
+    raise Unauthorized()

app.py

@@ -4,6 +4,8 @@
 from flask_cors import CORS
 from flask_restx import Api, Resource
 
+from api.auth import token_auth
+from config import config
 from util.logger import log
 
 app = Flask(__name__)
@@ -22,6 +24,8 @@
 }
 
 swagger_ui_enabled = '/'
+if config.get_env('PRODUCTION') == 'Y':
+    swagger_ui_enabled = False
 
 api = Api(app, version='0.0.1', title='HateyBot API',
           description='API for HateyBot',
@@ -35,6 +39,7 @@
 
 @auth.route('/')
 class AuthHandler(Resource):
+    @token_auth.login_required
     def get(self):
         return 'successful'
 

application.example.yml

@@ -0,0 +1,2 @@
+API_TOKEN: "1234567890"
+PRODUCTION: Y

config.py

@@ -0,0 +1,51 @@
+import os
+from os.path import dirname, abspath, join, exists
+
+import yaml
+
+from util.logger import log
+
+
+def get_or_create(dir_path):
+    if not exists(dir_path):
+        os.makedirs(dir_path)
+        log.info("Creating: " + dir_path)
+    return dir_path
+
+
+class Config:
+    SAVE_TO_FILE = True
+
+    __root_dir = dirname(abspath(__file__))
+    __application_config_path = join(__root_dir, 'application.yml')
+    __config_file = None
+
+    def data_dir(self, folder):
+        directory = get_or_create(abspath(join(self.__root_dir, 'data')))
+        return get_or_create(join(directory, folder))
+
+    def result_file(self, file_name):
+        return join(self.data_dir('result'), file_name)
+
+    def input_file(self, file_name):
+        return join(self.data_dir('input'), file_name)
+
+    def get_env(self, var):
+        return self.__get_var(var)
+
+    def __get_var(self, var):
+        if not self.__config_file:
+            try:
+                with open(self.__application_config_path, 'r') as stream:
+                    self.__config_file = yaml.safe_load(stream)
+                    log.info("Config Loaded")
+            except FileNotFoundError:
+                log.info("Config not found, using ENV Var")
+                return os.environ.get(var)
+        try:
+            return os.environ.get(var) or self.__config_file[var]
+        except KeyError:
+            log.error('Can not find ENV var: %s' % var)
+
+
+config = Config()