admin_views.py

from db.db_session import create_session
from db.models import *
from db import logic
from flask_admin.contrib.sqla import ModelView
from loader_controller import StreamerController, ApiError
from flask import request, flash

from settings import platform_list

class UserView(ModelView):
    can_create = False
    def delete_model(self, model):
        if model.role == 'admin':
            return flash('нельзя удалить администратора')
        else:
            return super().delete_model(model)

    form_columns = ['login', 'password', 'role', 'is_approved']

class GameView(ModelView):
    def delete_model(self, model):
        logic.on_delete_game(model)
        return super().delete_model(model)

    form_columns = ['name']

class StreamerView(ModelView):

    def create_model(self, form):
        platform = request.form.get('platform')
        if platform not in platform_list:
            flash(f'Платформы с названием {platform} нет в списке поддерживаемых платформ')
        platform_id = request.form.get('platform_id')
        controller = StreamerController()
        try:
            if not controller.check_streamer_exist(platform, platform_id):
                flash(f'стримера на платформе {platform} с id {platform_id} не существует')
                return False
        except ApiError:
            flash('ошибка при проверке стримера. Повторите попытку')
        streamer = super().create_model(form)
        self.session.expunge(streamer)
        controller.add_streamer(streamer)
        return True
    
    def delete_model(self, model):
        self.on_model_delete(model)
        logic.on_delete_streamer(model)
        StreamerController().delete_streamer(model)
        self.session.delete(model)
        self.session.commit()
        self.after_model_delete(model)
    

    form_columns = ['game', 'name', 'platform', 'platform_id', 'threshold']


class TriggerView(ModelView):
    form_columns = ['name']

    def create_model(self, form):
        name = request.form.get('name')
        if not name:
            return False
        name = name.lower()
        self.session.add(Trigger(name=name))
        self.session.commit()
        return True