feat: Allow for editing of Posters

WiiLink24 · Jun 30, 2024 · c5239fc · c5239fc
1 parent 153ac12
commit c5239fc
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 4 deletions.
diff --git a/templates/poster_add.html → templates/poster_action.html b/templates/poster_add.html → templates/poster_action.html
@@ -1,7 +1,7 @@
 {% extends "generic/base.html" %}
 
 {% block title %}
-    Upload a Movie
+    {{ action }} a Movie
 {% endblock %}
 
 {% block content %}
@@ -17,7 +17,11 @@
             {{ form.movie_id.label(class_="label") }} {{ form.movie_id(class_="input") }}
             <br>
             <br>
-            {{ form.poster.label(class_="label") }} {{ form.poster(size=32) }}
+            {{ form.poster.label(class_="label") }}
+            {% if poster_id %}
+            {{ existing_image('poster', url_for('get_poster', poster=poster_id)) }}
+            {% endif %}
+            {{ form.poster(size=32) }}
         </p>
         <br>
         <p>{{ form.upload(class_="button is-success") }}</p>

diff --git a/templates/poster_list.html b/templates/poster_list.html
@@ -30,6 +30,15 @@
                 <td>{{ poster.movie_id }}</td>
                 <td>
                     <div class="field has-addons">
+                         <p class="control">
+                            <a href="{{ url_for('edit_poster', poster_id=poster.poster_id) }}" class="button">
+                            <span class="icon is-small">
+                                <i class="fas fa-edit"></i>
+                            </span>
+                                <span>Edit Poster</span>
+                            </a>
+                        </p>
+
                         <p class="control">
                             <a href="{{ url_for('remove_poster', poster=poster.poster_id) }}" class="button is-danger">
                             <span class="icon is-small">

diff --git a/theunderground/forms.py b/theunderground/forms.py
@@ -193,7 +193,7 @@ class PosterForm(FlaskForm):
     title = StringField("Title", validators=[DataRequired(), Length(max=47)])
     msg = StringField("Message", validators=[DataRequired(), Length(max=15)])
     movie_id = IntegerField("Movie ID", validators=[DataRequired()])
-    poster = FileField("Poster", validators=[FileRequired()])
+    poster = FileField("Poster")
     upload = SubmitField("Create Poster!")
 
 

diff --git a/theunderground/posters.py b/theunderground/posters.py
@@ -1,6 +1,7 @@
 from io import BytesIO
 
 from flask import render_template, flash, url_for, redirect
+from werkzeug import exceptions
 
 from theunderground.forms import PosterForm
 from theunderground.operations import manage_delete_item
@@ -9,6 +10,7 @@
 from models import Posters, db
 from room import app, s3
 from url1.event_today import event_today
+from flask_wtf.file import FileRequired
 import config
 
 
@@ -30,6 +32,7 @@ def list_posters():
 @oidc.require_login
 def add_poster():
     form = PosterForm()
+    form.poster.validators = [FileRequired()]
 
     if form.validate_on_submit():
         db_poster = Posters(
@@ -52,7 +55,44 @@ def add_poster():
 
         return redirect(url_for("list_posters"))
 
-    return render_template("poster_add.html", form=form)
+    return render_template("poster_action.html", form=form, action="Upload")
+
+
+@app.route("/theunderground/movies/poster/<poster_id>/edit", methods=["GET", "POST"])
+@oidc.require_login
+def edit_poster(poster_id):
+    form = PosterForm()
+
+    poster = Posters.query.filter_by(poster_id=poster_id).first()
+    if not poster:
+        return exceptions.NotFound()
+
+    if form.validate_on_submit():
+        poster.title = form.title.data
+        poster.msg = form.msg.data
+        poster.movie_id = form.movie_id.data
+
+        if form.poster.data:
+            # Now upload poster
+            PosterAsset(poster_id, False).encode(form.poster)
+
+        # Commit before uploading to s3
+        db.session.commit()
+        if s3:
+            event_xml = event_today()
+            s3.upload_fileobj(
+                BytesIO(event_xml), config.r2_bucket_name, "event/today.xml"
+            )
+
+        return redirect(url_for("list_posters"))
+    else:
+        form.title.data = poster.title
+        form.msg.data = poster.msg
+        form.movie_id.data = poster.movie_id
+
+    return render_template(
+        "poster_action.html", form=form, poster_id=poster_id, action="Edit"
+    )
 
 
 @app.route("/theunderground/posters/<poster>/remove", methods=["GET", "POST"])