68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
from flask import current_app
|
|
from apiflask import APIFlask
|
|
|
|
from dotenv import load_dotenv
|
|
from flask_cors import CORS
|
|
|
|
from app.blueprints.keyword import keyword_blueprint
|
|
from app.blueprints.portfolio import portfolio_blueprint
|
|
from app.blueprints.shares import shares_blueprint
|
|
from app.blueprints.transactions import transaction_blueprint
|
|
from app.blueprints.telegram import telegram_blueprint
|
|
from app.blueprints.user import users_blueprint
|
|
from app.helper_functions import hash_password
|
|
from app.models import *
|
|
|
|
|
|
def create_app(config_filename=None):
|
|
load_dotenv()
|
|
|
|
# Create Flask app load app.config
|
|
application = APIFlask(__name__, openapi_blueprint_url_prefix='/api')
|
|
application.config.from_pyfile(config_filename)
|
|
|
|
CORS(application, resources={r"*": {"origins": "*"}})
|
|
|
|
application.app_context().push()
|
|
|
|
db.init_app(application)
|
|
|
|
# api blueprints
|
|
application.register_blueprint(keyword_blueprint)
|
|
application.register_blueprint(shares_blueprint)
|
|
application.register_blueprint(transaction_blueprint)
|
|
application.register_blueprint(portfolio_blueprint)
|
|
application.register_blueprint(users_blueprint)
|
|
application.register_blueprint(telegram_blueprint)
|
|
|
|
@application.before_first_request
|
|
def init_database():
|
|
db.create_all()
|
|
|
|
if current_app.config['BOT_EMAIL'] is not None and current_app.config['BOT_USERNAME'] is not None and current_app.config['BOT_PASSWORD'] is not None:
|
|
if db.session.query(User).filter_by(email=current_app.config['BOT_EMAIL']).first() is None: # Check if user already exist
|
|
bot = User(
|
|
email=current_app.config['BOT_EMAIL'],
|
|
username=current_app.config['BOT_USERNAME'],
|
|
password=hash_password(current_app.config['BOT_PASSWORD']),
|
|
admin=False
|
|
)
|
|
db.session.add(bot)
|
|
db.session.commit()
|
|
|
|
if current_app.config['ADMIN_EMAIL'] is not None and current_app.config['ADMIN_USERNAME'] is not None and current_app.config['ADMIN_PASSWORD'] is not None:
|
|
if db.session.query(User).filter_by(email=current_app.config['ADMIN_EMAIL']).first() is None: # Check if user already exist
|
|
admin = User(
|
|
email=current_app.config['ADMIN_EMAIL'],
|
|
username=current_app.config['ADMIN_USERNAME'],
|
|
password=hash_password(current_app.config['ADMIN_PASSWORD']),
|
|
admin=True
|
|
)
|
|
db.session.add(admin)
|
|
db.session.commit()
|
|
|
|
return application
|
|
|
|
|
|
app = create_app("config/flask.cfg")
|