Merge pull request #85 from skiedude/master

Migrate files_upload.py to use slack_sdk

Author: nzlosh

CHANGES.md

@@ -1,5 +1,9 @@
 # Change Log
 
+# 2.3.0
+
+* migrate files_upload.py to use the slack_sdk helper method files_upload_v2. API endpoint files.upload is deprecated and will stop functioning on March 11, 2025. 
+
 # 2.2.1
 
 * fix "__all__" reference for `post_attachment`

actions/files_upload.py

@@ -1,11 +1,13 @@
-from run import SlackAction
+from st2common.runners.base_action import Action
+from slack_sdk import WebClient
+from slack_sdk.errors import SlackApiError
 
 
-class FilesUploadAction(SlackAction):
+class FilesUploadAction(Action):
 
     def run(self, **kwargs):
-        # https://requests.readthedocs.io/en/master/user/quickstart/#post-a-multipart-encoded-file
-        files = None
+        token = kwargs.get('token') if kwargs.get('token') else self.config['action_token']
+        client = WebClient(token=token)
         if 'file_path' in kwargs and kwargs['file_path']:
             if 'file' in kwargs and kwargs['file']:
                 raise RuntimeError('Passing in "file" and "file_path" at the same time is'
@@ -14,10 +16,20 @@ def run(self, **kwargs):
                                    ' then use the "file_path" parameter. Otherwise, if you'
                                    ' would like to read the file yourself and pass in the data'
                                    ' then use the "file" parameter set to the content.')
-            # the name 'file' is hard coded because that's the name of the parameter
-            # that the Slack API is expecting
-            files = {'file': open(kwargs['file_path'], 'rb')}
-            kwargs.pop('file', None)
-            kwargs.pop('file_path')
-
-        super(FilesUploadAction, self).run(files=files, **kwargs)
+        try:
+            file = kwargs['file'] if kwargs['file'] else kwargs['file_path']
+            response = client.files_upload_v2(
+                filename=kwargs['filename'],
+                file=file,
+                content=kwargs['content'],
+                title=kwargs['title'],
+                channel=kwargs['channels'],
+                initial_comment=kwargs['initial_comment'],
+                thread_ts=kwargs['thread_ts'],
+            )
+            assert response['file']
+            return response.get('file', {}).get('name')
+        except SlackApiError as e:
+            assert e.response["ok"] is False
+            assert e.response["error"]
+            return False, e.response['error']

pack.yaml

@@ -7,7 +7,7 @@ keywords:
   - chat
   - messaging
   - instant messaging
-version: 2.2.1
+version: 2.3.0
 python_versions:
   - "3"
 author : StackStorm, Inc.

requirements.txt

@@ -4,3 +4,4 @@ beautifulsoup4==4.6.0
 lxml==4.6.5
 jinja2>=2.10.1
 slackclient==1.3.1
+slack_sdk==3.34.0