import json import os import time from flask import Flask, render_template, redirect, request, url_for, send_from_directory, jsonify from werkzeug.utils import secure_filename from src.service.ModelStore import ModelStore from src.model.entity.Slide import Slide from src.model.enum.ContentType import ContentType from src.interface.ObController import ObController from src.util.utils import str_to_enum, get_optional_string from src.util.UtilFile import randomize_filename class SlideController(ObController): def register(self): self._app.add_url_rule('/manage', 'manage', self.manage, methods=['GET']) self._app.add_url_rule('/slideshow', 'slideshow_slide_list', self._auth(self.slideshow), methods=['GET']) self._app.add_url_rule('/slideshow/playlist/set/', 'slideshow_slide_list_playlist_use', self._auth(self.slideshow), methods=['GET']) self._app.add_url_rule('/slideshow/slide/add', 'slideshow_slide_add', self._auth(self.slideshow_slide_add), methods=['POST']) self._app.add_url_rule('/slideshow/slide/edit', 'slideshow_slide_edit', self._auth(self.slideshow_slide_edit), methods=['POST']) self._app.add_url_rule('/slideshow/slide/toggle', 'slideshow_slide_toggle', self._auth(self.slideshow_slide_toggle), methods=['POST']) self._app.add_url_rule('/slideshow/slide/delete', 'slideshow_slide_delete', self._auth(self.slideshow_slide_delete), methods=['DELETE']) self._app.add_url_rule('/slideshow/slide/position', 'slideshow_slide_position', self._auth(self.slideshow_slide_position), methods=['POST']) self._app.add_url_rule('/slideshow/player-refresh', 'slideshow_player_refresh', self._auth(self.slideshow_player_refresh), methods=['GET']) def manage(self): return redirect(url_for('slideshow_slide_list')) def slideshow(self, playlist_id: int = 0): current_playlist = self._model_store.playlist().get(playlist_id) playlist_id = current_playlist.id if current_playlist else None return render_template( 'slideshow/slides/list.jinja.html', current_playlist=current_playlist, playlists=self._model_store.playlist().get_enabled_playlists(), enabled_slides=self._model_store.slide().get_slides(playlist_id=playlist_id, enabled=True), disabled_slides=self._model_store.slide().get_slides(playlist_id=playlist_id, enabled=False), var_last_restart=self._model_store.variable().get_one_by_name('last_restart'), contents={content.id: content.name for content in self._model_store.content().get_contents()}, enum_content_type=ContentType ) def slideshow_slide_add(self): content = None if 'type' in request.form: content = self._model_store.content().add_form_raw( name=request.form['name'], type=str_to_enum(request.form['type'], ContentType), request_files=request.files, upload_dir=self._app.config['UPLOAD_FOLDER'], location=request.form['object'] if 'object' in request.form else None ) slide = Slide( content_id=content.id if content else request.form['content_id'], duration=request.form['duration'], is_notification=True if 'is_notification' in request.form else False, playlist_id=request.form['playlist_id'] if 'playlist_id' in request.form and request.form['playlist_id'] else None, cron_schedule=get_optional_string(request.form['cron_schedule']), cron_schedule_end=get_optional_string(request.form['cron_schedule_end']), ) self._model_store.slide().add_form(slide) self._post_update() if slide.playlist_id: return redirect(url_for('slideshow_slide_list_playlist_use', playlist_id=slide.playlist_id)) return redirect(url_for('slideshow_slide_list')) def slideshow_slide_edit(self): slide = self._model_store.slide().update_form( id=request.form['id'], content_id=request.form['content_id'], duration=request.form['duration'], is_notification=True if 'is_notification' in request.form else False, cron_schedule=request.form['cron_schedule'], cron_schedule_end=request.form['cron_schedule_end'], ) self._post_update() if slide.playlist_id: return redirect(url_for('slideshow_slide_list_playlist_use', playlist_id=slide.playlist_id)) return redirect(url_for('slideshow_slide_list')) def slideshow_slide_toggle(self): data = request.get_json() self._model_store.slide().update_enabled(data.get('id'), data.get('enabled')) self._post_update() return jsonify({'status': 'ok'}) def slideshow_slide_delete(self): data = request.get_json() self._model_store.slide().delete(data.get('id')) self._post_update() return jsonify({'status': 'ok'}) def slideshow_slide_position(self): data = request.get_json() self._model_store.slide().update_positions(data) self._post_update() return jsonify({'status': 'ok'}) def slideshow_player_refresh(self): self._model_store.variable().update_by_name("refresh_player_request", time.time()) return redirect( url_for( 'slideshow_slide_list', refresh_player=self._model_store.variable().get_one_by_name('polling_interval').as_int() ) ) def _post_update(self): self._model_store.variable().update_by_name("last_slide_update", time.time())