From e1c86371a6aeb30f7cfe392e841a29313096dfa6 Mon Sep 17 00:00:00 2001
From: rimu <3310831+rimu@users.noreply.github.com>
Date: Fri, 16 Feb 2024 06:24:28 +1300
Subject: [PATCH] nsfw and nsfl icons on posts - editing

---
 app/post/routes.py                | 13 ++++++++-----
 app/templates/post/post_edit.html |  2 +-
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/app/post/routes.py b/app/post/routes.py
index 99827bcc..dffbb163 100644
--- a/app/post/routes.py
+++ b/app/post/routes.py
@@ -559,11 +559,14 @@ def post_edit(post_id: int):
     post = Post.query.get_or_404(post_id)
     form = CreatePostForm()
     if post.user_id == current_user.id or post.community.is_moderator():
-        if get_setting('allow_nsfw', False) is False:
-            form.nsfw.render_kw = {'disabled': True}
-        if get_setting('allow_nsfl', False) is False:
+        if g.site.enable_nsfl is False:
             form.nsfl.render_kw = {'disabled': True}
-        images_disabled = 'disabled' if not get_setting('allow_local_image_posts', True) else ''
+        if post.community.nsfw:
+            form.nsfw.data = True
+            form.nsfw.render_kw = {'disabled': True}
+        if post.community.nsfl:
+            form.nsfl.data = True
+            form.nsfw.render_kw = {'disabled': True}
 
         form.communities.choices = [(c.id, c.display_name()) for c in current_user.communities()]
 
@@ -669,7 +672,7 @@ def post_edit(post_id: int):
                 form.image_alt_text.data = post.image.alt_text
             form.notify_author.data = post.notify_author
             return render_template('post/post_edit.html', title=_('Edit post'), form=form, post=post,
-                                   images_disabled=images_disabled, markdown_editor=True,
+                                   markdown_editor=True,
                                    moderating_communities=moderating_communities(current_user.get_id()),
                                    joined_communities=joined_communities(current_user.get_id()),
                                    inoculation=inoculation[randint(0, len(inoculation) - 1)]
diff --git a/app/templates/post/post_edit.html b/app/templates/post/post_edit.html
index adfdac1b..99757bf5 100644
--- a/app/templates/post/post_edit.html
+++ b/app/templates/post/post_edit.html
@@ -51,7 +51,7 @@
                 <button class="nav-link" id="link-tab" data-bs-toggle="tab" data-bs-target="#link-tab-pane"
                         type="button" role="tab" aria-controls="link-tab-pane" aria-selected="false">Link</button>
                 <button class="nav-link" id="image-tab" data-bs-toggle="tab" data-bs-target="#image-tab-pane"
-                        type="button" role="tab" aria-controls="image-tab-pane" aria-selected="false" {{ images_disabled }}>Image</button>
+                        type="button" role="tab" aria-controls="image-tab-pane" aria-selected="false">Image</button>
                 <button class="nav-link" id="poll-tab" data-bs-toggle="tab" data-bs-target="#poll-tab-pane"
                             type="button" role="tab" aria-controls="poll-tab-pane" aria-selected="false" disabled>Poll</button>
               </div>