diff --git a/.env.example b/.env.example new file mode 100644 index 0000000..70be933 --- /dev/null +++ b/.env.example @@ -0,0 +1,8 @@ +APP_SECRET=s3cr3tk3yh3r3 +FLASK_DEBUG=1 + +MONGO_HOST=mongo +MONGO_USER=username +MONGO_PASSWORD=Pa$$w0rD + +ADUNATIO_PRIV_KEY=privkey.pem diff --git a/main.py b/main.py index 88c0145..428489d 100644 --- a/main.py +++ b/main.py @@ -4,6 +4,9 @@ from flask_jwt_extended import JWTManager, create_access_token from mongoengine import connect from werkzeug.security import check_password_hash +from os import environ, path +from dotenv import load_dotenv + from internal_lib.AuthMethots import AuthJWT from models.Group import Group from models.Union import Union @@ -11,13 +14,24 @@ from models.User import User from restapi import MongoApi from flask_admin import Admin +""" +Load .env file and get variables +""" + +base_path = path.abspath(path.dirname(__file__)) +load_dotenv(path.join(base_path, '.env')) + +MONGO_HOST = environ.get('MONGO_HOST') +MONGO_USER = environ.get('MONGO_USER') +MONGO_PASSWORD = environ.get('MONGO_PASSWORD') + """ Mongodb connection string """ -connect('adunatio', host='mongo', username="xcoder", password="4dun4710", authentication_source='admin') +connect('adunatio', host=MONGO_HOST, username=MONGO_USER, password=MONGO_PASSWORD, authentication_source='admin') app = Flask(__name__) -app.secret_key = "secret_key+secret_key" +app.secret_key = environ.get('APP_SECRET') app.config["JWT_TOKEN_LOCATION"] = "headers" app.config["JWT_HEADER_NAME"] = "Adunation_Session_Token" app.config["JWT_HEADER_TYPE"] = "Bearer" diff --git a/requirements.txt b/requirements.txt index 84324bb..86faddf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,5 @@ flask_jwt_extended flask_views flask_login pillow -flask-admin \ No newline at end of file +flask-admin +python-dotenv