collab button (changes genre to 2)

WiiLink24 · Jun 18, 2024 · f93ce00 · f93ce00
1 parent e17f4bb
commit f93ce00
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/templates/movie_add.html b/templates/movie_add.html
@@ -20,6 +20,9 @@
             {{ form.thumbnail.label(class_="label") }} {{ form.thumbnail(size=32) }}
             <br>
             <br>
+            {{ form.is_collab.label(class_="label") }} {{ form.is_collab }}
+            <br>
+            <br>
             {{ form.category.label(class_="label") }} {{ form.category }}
         </p>
         <br>

diff --git a/theunderground/forms.py b/theunderground/forms.py
@@ -75,6 +75,7 @@ class MovieUploadForm(FlaskForm):
     ds_movie = FileField("DSi Movie")
     title = StringField("Movie title", validators=[DataRequired(), Length(max=48)])
     thumbnail = FileField("Movie thumbnail", validators=[FileRequired()])
+    is_collab = BooleanField("Collaboration?")
     # Choices for the select field are only evaluated once, so we must set it when necessary.
     category = SelectField("Movie category", validators=[DataRequired()])
     upload = SubmitField("Add Movie")

diff --git a/theunderground/movies.py b/theunderground/movies.py
@@ -67,14 +67,19 @@ def add_movie():
                 # Get the Mobiclip's length from header.
                 length = get_mobiclip_length(movie_data)
 
+                if form.is_collab.data:
+                    genre = 2
+                else:
+                    genre = 0
+
                 # Insert this movie to the database.
                 # For right now, we will assume defaults.
                 db_movie = Movies(
                     title=form.title.data,
                     category_id=form.category.data,
                     length=length,
                     aspect=True,
-                    genre=0,
+                    genre=genre,
                     sp_page_id=0,
                     staff=False,
                 )
@@ -91,7 +96,8 @@ def add_movie():
 
                     if validation_ds:
                         db_movie.ds_mov_id = db_movie.movie_id
-                        db_movie.genre = 1
+                        if genre == 0:
+                            db_movie.genre = 1
 
                         # Re-commit the DSi stuff
                         db.session.commit()