Create bot and admin user when creating tables

This commit is contained in:
Administrator 2022-03-27 10:42:53 +02:00
parent b3f76ec8bf
commit 1fe1e0e9cf

View File

@ -1,8 +1,11 @@
import os
from apiflask import APIFlask from apiflask import APIFlask
from dotenv import load_dotenv from dotenv import load_dotenv
from flask_cors import CORS from flask_cors import CORS
from api.helper_functions import hash_password
from models import * from models import *
from api_blueprint_keyword import keyword_blueprint from api_blueprint_keyword import keyword_blueprint
from api_blueprint_shares import shares_blueprint from api_blueprint_shares import shares_blueprint
@ -35,6 +38,24 @@ def create_app():
def init_database(): def init_database():
db.create_all() db.create_all()
if os.getenv("BOT_USER") is not None and os.getenv("BOT_PASSWORD") is not None:
bot = User(
username=os.getenv("BOT_USER"),
password=hash_password(os.getenv("BOT_PASSWORD")),
admin=False
)
db.session.add(bot)
db.session.commit()
if os.getenv("ADMIN_USER") is not None and os.getenv("ADMIN_PASSWORD") is not None:
admin = User(
username=os.getenv("ADMIN_USER"),
password=hash_password(os.getenv("ADMIN_PASSWORD")),
admin=True
)
db.session.add(admin)
db.session.commit()
return application return application