feat: Allow for editing of Pay Posters

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

Hide file tree

Showing 4 changed files with 52 additions and 6 deletions.
diff --git a/templates/pay_poster_add.html → templates/pay_poster_action.html b/templates/pay_poster_add.html → templates/pay_poster_action.html
@@ -1,7 +1,7 @@
 {% extends "generic/base.html" %}
 
 {% block title %}
-    Upload a Pay Poster
+    {{ action }} a Pay Poster
 {% endblock %}
 
 {% block content %}
@@ -17,7 +17,11 @@
             {{ form.movie.label(class_="label") }} {{ form.movie(size=32) }}
             <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_pay_poster', poster=poster_id)) }}
+            {% endif %}
+            {{ form.poster(size=32) }}
         </p>
         <br>
         <p>{{ form.upload(class_="button is-success") }}</p>

diff --git a/templates/pay_poster_list.html b/templates/pay_poster_list.html
@@ -29,7 +29,7 @@
                 <td>
                     <div class="field has-addons">
                         <p class="control">
-                            <a href="" class="button">
+                            <a href="{{ url_for('edit_pay_poster', poster_id=poster.poster_id) }}" class="button">
                             <span class="icon is-small">
                                 <i class="fas fa-edit"></i>
                             </span>

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

diff --git a/theunderground/pay_posters.py b/theunderground/pay_posters.py
@@ -1,13 +1,15 @@
 from Crypto.Cipher import AES
 from Crypto.Util.Padding import pad
 from flask import render_template, flash, url_for, redirect
+from werkzeug import exceptions
 
 from theunderground.forms import PayPosterForm
 from theunderground.operations import manage_delete_item
 from theunderground.admin import oidc
 from asset_data import PosterAsset, PayMovieAsset
 from models import PayPosters, db
 from room import app
+from flask_wtf.file import FileRequired
 
 import os
 
@@ -34,6 +36,10 @@ def list_pay_posters():
 def add_pay_poster():
     form = PayPosterForm()
 
+    # Add the file validators
+    form.poster.validators = [FileRequired()]
+    form.movie.validators = [FileRequired()]
+
     if form.validate_on_submit():
         db_poster = PayPosters(
             msg=form.msg.data, title=form.title.data, type=1, aspect=False
@@ -62,7 +68,43 @@ def add_pay_poster():
 
         return redirect(url_for("list_pay_posters"))
 
-    return render_template("pay_poster_add.html", form=form)
+    return render_template("pay_poster_action.html", form=form, action="Upload")
+
+
+@app.route("/theunderground/payposters/<poster_id>/edit", methods=["GET", "POST"])
+@oidc.require_login
+def edit_pay_poster(poster_id):
+    form = PayPosterForm()
+
+    poster = PayPosters.query.filter_by(poster_id=poster_id).first()
+    if not poster:
+        return exceptions.NotFound()
+
+    if form.validate_on_submit():
+        poster.msg = form.msg.data
+        poster.title = form.title.data
+
+        # Encrypt movie
+        if form.movie.data:
+            cipher = AES.new(PAY_POSTER_KEY, AES.MODE_CBC, iv=PAY_POSTER_IV)
+            encrypted_movie = cipher.encrypt(
+                pad(form.movie.data.read(), AES.block_size)
+            )
+            PayMovieAsset(poster.poster_id).upload_movie(encrypted_movie)
+
+        if form.poster.data:
+            # Now upload poster
+            PosterAsset(poster.poster_id, True).encode(form.poster)
+
+        db.session.commit()
+        return redirect(url_for("list_pay_posters"))
+    else:
+        form.msg.data = poster.msg
+        form.title.data = poster.title
+
+    return render_template(
+        "pay_poster_action.html", form=form, action="Edit", poster_id=poster_id
+    )
 
 
 @app.route("/theunderground/payposters/<poster>/remove", methods=["GET", "POST"])