From 1fe1e0e9cfd47085062ee88dcc6532cd5a628f2f Mon Sep 17 00:00:00 2001
From: H4CK3R-01 <resiaknairolf@gmail.com>
Date: Sun, 27 Mar 2022 10:42:53 +0200
Subject: [PATCH] Create bot and admin user when creating tables

---
 api/app.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/api/app.py b/api/app.py
index 933b471..955cc18 100644
--- a/api/app.py
+++ b/api/app.py
@@ -1,8 +1,11 @@
+import os
+
 from apiflask import APIFlask
 
 from dotenv import load_dotenv
 from flask_cors import CORS
 
+from api.helper_functions import hash_password
 from models import *
 from api_blueprint_keyword import keyword_blueprint
 from api_blueprint_shares import shares_blueprint
@@ -35,6 +38,24 @@ def create_app():
     def init_database():
         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