<reponame>penguins-dstu/ttk-control-server<filename>init_db.py<gh_stars>0
from app import app, db
from app.models import User, Role, Status, WhiteList
from app.db_sup import multiadd

'''
Данный модуль предназначен для заполнения БД начальными записями и настройками.
'''

def add_roles():
    '''Добавление ролей пользователя в базу данных.'''
    multiadd(*[Role(role_name) for role_name in app.config['ROLE_NAMES']])

def add_status():
    '''Добавление видов состояния активности пользователя в базу данных.'''
    multiadd(*[Status(status_name) for status_name in app.config['STATUS_NAMES']])

def add_admin():
    '''Установка пароля администратора.'''
    msg = 'Установить пароль администратора: '
    admin = User(username=app.config['ADMIN_USERNAME'])
    admin.set_role(1)
    admin.set_password(input(msg))
    multiadd(admin)

def add_whitelist():
    '''Добавление "белого списка" приложений сотрудников.'''
    multiadd(*[WhiteList(wl_item) for wl_item in app.config['WHITELIST']])

def init():
    '''Общий метод инициализации.'''
    add_roles()
    add_status()
    add_admin()
    add_whitelist()

if __name__ == '__main__':
    init()

from app import app, db
from app.models import User, Role, Status, WhiteList
from app.db_sup import multiadd

'''
Данный модуль предназначен для заполнения БД начальными записями и настройками.
'''

def add_roles():
    '''Добавление ролей пользователя в базу данных.'''
    multiadd(*[Role(role_name) for role_name in app.config['ROLE_NAMES']])

def add_status():
    '''Добавление видов состояния активности пользователя в базу данных.'''
    multiadd(*[Status(status_name) for status_name in app.config['STATUS_NAMES']])

def add_admin():
    '''Установка пароля администратора.'''
    msg = 'Установить пароль администратора: '
    admin = User(username=app.config['ADMIN_USERNAME'])
    admin.set_role(1)
    admin.set_password(input(msg))
    multiadd(admin)

def add_whitelist():
    '''Добавление "белого списка" приложений сотрудников.'''
    multiadd(*[WhiteList(wl_item) for wl_item in app.config['WHITELIST']])

def init():
    '''Общий метод инициализации.'''
    add_roles()
    add_status()
    add_admin()
    add_whitelist()

if __name__ == '__main__':
    init()
Данный модуль предназначен для заполнения БД начальными записями и настройками. Добавление ролей пользователя в базу данных. Добавление видов состояния активности пользователя в базу данных. Установка пароля администратора. Добавление "белого списка" приложений сотрудников. Общий метод инициализации.