From ce4caeabd3f02262374e5bff7121741602e526ff Mon Sep 17 00:00:00 2001
From: Petr Kalis <petr.kalis@gmail.com>
Date: Thu, 24 Mar 2022 13:47:07 +0100
Subject: [PATCH] Added configurable maximum file size of review upload to
 Slack

---
 .../plugins/publish/collect_slack_family.py   | 29 +++++++++++--------
 .../plugins/publish/integrate_slack_api.py    | 18 ++++++++++--
 .../projects_schema/schema_project_slack.json |  9 ++++++
 3 files changed, 42 insertions(+), 14 deletions(-)

diff --git a/openpype/modules/slack/plugins/publish/collect_slack_family.py b/openpype/modules/slack/plugins/publish/collect_slack_family.py
index 6c965b04cd..7475bdc89e 100644
--- a/openpype/modules/slack/plugins/publish/collect_slack_family.py
+++ b/openpype/modules/slack/plugins/publish/collect_slack_family.py
@@ -35,20 +35,25 @@ class CollectSlackFamilies(pyblish.api.InstancePlugin):
             return
 
         # make slack publishable
-        if profile:
-            self.log.info("Found profile: {}".format(profile))
-            if instance.data.get('families'):
-                instance.data['families'].append('slack')
-            else:
-                instance.data['families'] = ['slack']
+        if not profile:
+            return
 
-            instance.data["slack_channel_message_profiles"] = \
-                profile["channel_messages"]
+        self.log.info("Found profile: {}".format(profile))
+        if instance.data.get('families'):
+            instance.data['families'].append('slack')
+        else:
+            instance.data['families'] = ['slack']
 
-            slack_token = (instance.context.data["project_settings"]
-                                                ["slack"]
-                                                ["token"])
-            instance.data["slack_token"] = slack_token
+        selected_profiles = profile["channel_messages"]
+        for prof in selected_profiles:
+            prof["review_upload_limit"] = profile.get("review_upload_limit",
+                                                      50)
+        instance.data["slack_channel_message_profiles"] = selected_profiles
+
+        slack_token = (instance.context.data["project_settings"]
+                                            ["slack"]
+                                            ["token"])
+        instance.data["slack_token"] = slack_token
 
     def main_family_from_instance(self, instance):  # TODO yank from integrate
         """Returns main family of entered instance."""
diff --git a/openpype/modules/slack/plugins/publish/integrate_slack_api.py b/openpype/modules/slack/plugins/publish/integrate_slack_api.py
index 018a7594bb..8fde25b42d 100644
--- a/openpype/modules/slack/plugins/publish/integrate_slack_api.py
+++ b/openpype/modules/slack/plugins/publish/integrate_slack_api.py
@@ -35,7 +35,7 @@ class IntegrateSlackAPI(pyblish.api.InstancePlugin):
             message = self._get_filled_message(message_profile["message"],
                                                instance,
                                                review_path)
-            self.log.info("message:: {}".format(message))
+            self.log.debug("message:: {}".format(message))
             if not message:
                 return
 
@@ -43,7 +43,8 @@ class IntegrateSlackAPI(pyblish.api.InstancePlugin):
                 publish_files.add(thumbnail_path)
 
             if message_profile["upload_review"] and review_path:
-                publish_files.add(review_path)
+                message, publish_files = self._handle_review_upload(
+                    message, message_profile, publish_files, review_path)
 
             project = instance.context.data["anatomyData"]["project"]["code"]
             for channel in message_profile["channels"]:
@@ -75,6 +76,19 @@ class IntegrateSlackAPI(pyblish.api.InstancePlugin):
                 dbcon = mongo_client[database_name]["notification_messages"]
                 dbcon.insert_one(msg)
 
+    def _handle_review_upload(self, message, message_profile, publish_files,
+                              review_path):
+        """Check if uploaded file is not too large"""
+        review_file_size_MB = os.path.getsize(review_path) / 1024 / 1024
+        file_limit = message_profile.get("review_upload_limit", 50)
+        if review_file_size_MB > file_limit:
+            if review_path not in message:
+                message += "\n Review upload omitted because of " + \
+                           "file size, file located at: {}".format(review_path)
+        else:
+            publish_files.add(review_path)
+        return message, publish_files
+
     def _get_filled_message(self, message_templ, instance, review_path=None):
         """Use message_templ and data from instance to get message content.
 
diff --git a/openpype/settings/entities/schemas/projects_schema/schema_project_slack.json b/openpype/settings/entities/schemas/projects_schema/schema_project_slack.json
index 14814d8b01..9ca7c35e10 100644
--- a/openpype/settings/entities/schemas/projects_schema/schema_project_slack.json
+++ b/openpype/settings/entities/schemas/projects_schema/schema_project_slack.json
@@ -75,6 +75,15 @@
                                         "type": "list",
                                         "object_type": "text"
                                     },
+                                    {
+                                        "type": "number",
+                                        "key": "review_upload_limit",
+                                        "label": "Upload review of maximally size (MB)",
+                                        "decimal": 2,
+                                        "default": 50,
+                                        "minimum": 0,
+                                        "maximum": 1000000
+                                    },
                                     {
                                         "type": "separator"
                                     },