obscreen/src/controller/FleetNodePlayerGroupController.py

54 lines
2.7 KiB
Python

import json
from flask import Flask, render_template, redirect, request, url_for, jsonify
from src.service.ModelStore import ModelStore
from src.model.entity.NodePlayerGroup import NodePlayerGroup
from src.interface.ObController import ObController
class FleetNodePlayerGroupController(ObController):
def guard_fleet(self, f):
def decorated_function(*args, **kwargs):
if not self._model_store.variable().map().get('fleet_player_enabled').as_bool():
return redirect(url_for('manage'))
return f(*args, **kwargs)
return decorated_function
def register(self):
self._app.add_url_rule('/fleet/node-player-group/list', 'fleet_node_player_group_list', self.guard_fleet(self._auth(self.fleet_node_player_group_list)), methods=['GET'])
self._app.add_url_rule('/fleet/node-player-group/add', 'fleet_node_player_group_add', self.guard_fleet(self._auth(self.fleet_node_player_group_add)), methods=['POST'])
self._app.add_url_rule('/fleet/node-player-group/edit', 'fleet_node_player_group_edit', self.guard_fleet(self._auth(self.fleet_node_player_group_edit)), methods=['POST'])
self._app.add_url_rule('/fleet/node-player-group/delete', 'fleet_node_player_group_delete', self.guard_fleet(self._auth(self.fleet_node_player_group_delete)), methods=['DELETE'])
def fleet_node_player_group_list(self):
return render_template(
'fleet/player-group/list.jinja.html',
node_player_groups=self._model_store.node_player_group().get_groups(with_default=True),
playlists=self._model_store.playlist().get_all_labels_indexed()
)
def fleet_node_player_group_add(self):
playlist_id = request.form['playlist_id'] if 'playlist_id' in request.form and request.form['playlist_id'] else None
self._model_store.node_player_group().add_form(NodePlayerGroup(
name=request.form['name'],
playlist_id=playlist_id,
))
return redirect(url_for('fleet_node_player_group_list'))
def fleet_node_player_group_edit(self):
playlist_id = request.form['playlist_id'] if 'playlist_id' in request.form and request.form['playlist_id'] else None
self._model_store.node_player_group().update_form(request.form['id'], request.form['name'], playlist_id)
return redirect(url_for('fleet_node_player_group_list'))
def fleet_node_player_group_delete(self):
data = request.get_json()
id = data.get('id')
if self._model_store.node_player().count_node_players_for_group(id) > 0:
return jsonify({'status': 'error', 'message': self.t('node_player_group_delete_has_node_player')}), 400
self._model_store.node_player_group().delete(id)
return jsonify({'status': 'ok'})