From 492d1ca91cc95c5f1fd86fbc88df5286bb95db71 Mon Sep 17 00:00:00 2001 From: Jan-Pieter Baert Date: Tue, 19 Apr 2022 20:08:29 +0200 Subject: [PATCH] Update script to read admins from configuration file --- app/add_admins.py | 23 ++++++----------------- app/config.example.py | 1 + 2 files changed, 7 insertions(+), 17 deletions(-) diff --git a/app/add_admins.py b/app/add_admins.py index 3e71c60..b3f8112 100644 --- a/app/add_admins.py +++ b/app/add_admins.py @@ -2,24 +2,13 @@ from models import User from app import db +from models import User +from config import Configuration def add() -> None: """Add users as admin.""" - feli = User() - feli.configure("feliciaan", True, 0) - db.session.add(feli) - - destro = User() - destro.configure("destro", True, 0) - db.session.add(destro) - - iepoev = User() - iepoev.configure("iepoev", True, 1) - db.session.add(iepoev) - - flynn = User() - flynn.configure("flynn", True, 0) - db.session.add(flynn) - - # To future developers, add yourself here + for username in Configuration.HALDIS_ADMINS: + user = User() + user.configure(username, True, 0) + db.session.add(user) diff --git a/app/config.example.py b/app/config.example.py index 191ce24..daac1ab 100644 --- a/app/config.example.py +++ b/app/config.example.py @@ -8,6 +8,7 @@ class Configuration: SQLALCHEMY_DATABASE_URI = "sqlite:///haldis.db" SQLALCHEMY_TRACK_MODIFICATIONS = False DEBUG = True + HALDIS_ADMIN_USERS = [] SECRET_KEY = "" SLACK_WEBHOOK = None LOGFILE = "haldis.log"