csp-adapter-slack 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

csp_adapter_slack/__init__.py

@@ -1,4 +1,6 @@
-__version__ = "0.2.0"
+__version__ = "0.3.0"
 
 from .adapter import *
 from .adapter_config import *
+from .mention import *
+from .message import *

csp_adapter_slack/adapter.py

@@ -9,7 +9,6 @@ import csp
 from csp.impl.adaptermanager import AdapterManagerImpl
 from csp.impl.outputadapter import OutputAdapter
 from csp.impl.pushadapter import PushInputAdapter
-from csp.impl.struct import Struct
 from csp.impl.types.tstype import ts
 from csp.impl.wiring import py_output_adapter_def, py_push_adapter_def
 from slack_sdk.errors import SlackApiError
@@ -19,33 +18,13 @@ from slack_sdk.socket_mode.response import SocketModeResponse
 from slack_sdk.web import WebClient
 
 from .adapter_config import SlackAdapterConfig
+from .message import SlackMessage
 
 T = TypeVar("T")
 log = getLogger(__file__)
 
 
-__all__ = ("SlackMessage", "mention_user", "SlackAdapterManager", "SlackInputAdapterImpl", "SlackOutputAdapterImpl")
-
-
-class SlackMessage(Struct):
-    user: str
-    user_email: str  # email of the author
-    user_id: str  # user id of the author
-    tags: List[str]  # list of mentions
-
-    channel: str  # name of channel
-    channel_id: str  # id of channel
-    channel_type: str  # type of channel, in "message", "public" (app_mention), "private" (app_mention)
-
-    msg: str  # parsed text payload
-    reaction: str  # emoji reacts
-    thread: str  # thread id, if in thread
-    payload: dict  # raw message payload
-
-
-def mention_user(userid: str) -> str:
-    """Convenience method, more difficult to do in symphony but we want slack to be symmetric"""
-    return f"<@{userid}>"
+__all__ = ("SlackAdapterManager", "SlackInputAdapterImpl", "SlackOutputAdapterImpl")
 
 
 class SlackAdapterManager(AdapterManagerImpl):
@@ -69,14 +48,18 @@ class SlackAdapterManager(AdapterManagerImpl):
         self._thread: Thread = None
 
         # lookups for mentions and redirection
-        self._room_id_to_room_name: Dict[str, str] = {}
-        self._room_id_to_room_type: Dict[str, str] = {}
-        self._room_name_to_room_id: Dict[str, str] = {}
+        self._channel_id_to_channel_name: Dict[str, str] = {}
+        self._channel_id_to_channel_type: Dict[str, str] = {}
+        self._channel_name_to_channel_id: Dict[str, str] = {}
         self._user_id_to_user_name: Dict[str, str] = {}
         self._user_id_to_user_email: Dict[str, str] = {}
         self._user_name_to_user_id: Dict[str, str] = {}
         self._user_email_to_user_id: Dict[str, str] = {}
 
+        # if subscribed to mentions AND events, will get 2 copies,
+        # so we want to dedupe by id
+        self._seen_msg_ids = set()
+
     def subscribe(self):
         return _slack_input_adapter(self, push_mode=csp.PushMode.NON_COLLAPSING)
 
@@ -137,14 +120,33 @@ class SlackAdapterManager(AdapterManagerImpl):
             if ret.status_code == 200:
                 # TODO OAuth scopes required
                 for user in ret.data["members"]:
+                    # Grab name
                     name = user["profile"].get("real_name_normalized", user["name"])
-                    user_id = user["profile"]["id"]
-                    email = user["profile"]["email"]
+
+                    # Try to grab id
+                    if "id" in user:
+                        user_id = user["id"]
+                    elif "id" in user["profile"]:
+                        user_id = user["profile"]["id"]
+                    else:
+                        raise RuntimeError(f"No id found in user profile: {user}")
+
+                    # Try to grab email
+                    if "email" in user["profile"]:
+                        email = user["profile"]["email"]
+                    else:
+                        log.warning(f"No email found in user profile, using id: {user}")
+                        email = user_id
+
                     self._user_id_to_user_name[user_id] = name
                     self._user_name_to_user_id[name] = user_id  # TODO is this 1-1 in slack?
                     self._user_id_to_user_email[user_id] = email
                     self._user_email_to_user_id[email] = user_id
-            return self._user_name_to_user_id.get(user_name, None)
+
+            user_id = self._user_name_to_user_id.get(user_name, None)
+            if user_id is None:
+                # no user found
+                raise ValueError(f"User {user_name} not found in Slack")
         return user_id
 
     def _channel_data_to_channel_kind(self, data) -> str:
@@ -156,8 +158,8 @@ class SlackAdapterManager(AdapterManagerImpl):
 
     def _get_channel_from_id(self, channel_id):
         # try to pull from cache
-        name = self._room_id_to_room_name.get(channel_id, None)
-        kind = self._room_id_to_room_type.get(channel_id, None)
+        name = self._channel_id_to_channel_name.get(channel_id, None)
+        kind = self._channel_id_to_channel_type.get(channel_id, None)
 
         # if none, refresh data via web client
         if name is None:
@@ -167,18 +169,29 @@ class SlackAdapterManager(AdapterManagerImpl):
                 kind = self._channel_data_to_channel_kind(ret.data["channel"])
                 if kind == "message":
                     # TODO use same behavior as symphony adapter
-                    name = "DM"
+                    name = "IM"
                 else:
                     name = ret.data["channel"]["name"]
 
-                self._room_id_to_room_name[channel_id] = name
-                self._room_name_to_room_id[name] = channel_id
-                self._room_id_to_room_type[channel_id] = kind
+                if name == "IM":
+                    # store by the name of the user
+                    user = ret.data["channel"]["user"]
+                    user_name = self._get_user_from_id(user)[0]
+                    self._channel_name_to_channel_id[user_name] = channel_id
+                else:
+                    self._channel_name_to_channel_id[name] = channel_id
+                self._channel_id_to_channel_name[channel_id] = name
+                self._channel_id_to_channel_type[channel_id] = kind
         return name, kind
 
     def _get_channel_from_name(self, channel_name):
-        # try to pull from cache
-        channel_id = self._room_name_to_room_id.get(channel_name, None)
+        # first, see if its a regular name or tagged name
+        if channel_name.startswith("<#") and channel_name.endswith("|>"):
+            # strip out the tag
+            channel_id = channel_name[2:-2]
+        else:
+            # try to pull from cache
+            channel_id = self._channel_name_to_channel_id.get(channel_name, None)
 
         # if none, refresh data via web client
         if channel_id is None:
@@ -191,10 +204,13 @@ class SlackAdapterManager(AdapterManagerImpl):
                     name = channel["name"]
                     channel_id = channel["id"]
                     kind = self._channel_data_to_channel_kind(channel)
-                    self._room_id_to_room_name[channel_id] = name
-                    self._room_name_to_room_id[name] = channel_id
-                    self._room_id_to_room_type[channel_id] = kind
-            return self._room_name_to_room_id.get(channel_name, None)
+                    self._channel_id_to_channel_name[channel_id] = name
+                    self._channel_name_to_channel_id[name] = channel_id
+                    self._channel_id_to_channel_type[channel_id] = kind
+            channel_id = self._channel_name_to_channel_id.get(channel_name, None)
+            if channel_id is None:
+                # no channel found
+                raise ValueError(f"Channel {channel_name} not found in Slack")
         return channel_id
 
     def _get_tags_from_message(self, blocks) -> List[str]:
@@ -222,9 +238,18 @@ class SlackAdapterManager(AdapterManagerImpl):
             response = SocketModeResponse(envelope_id=req.envelope_id)
             client.send_socket_mode_response(response)
 
+            if req.payload["event"]["ts"] in self._seen_msg_ids:
+                # already seen, pop it and move on
+                self._seen_msg_ids.remove(req.payload["event"]["ts"])
+                return
+
+            # else add it so we don't process it again
+            self._seen_msg_ids.add(req.payload["event"]["ts"])
+
             if req.payload["event"]["type"] in ("message", "app_mention") and req.payload["event"].get("subtype") is None:
                 user, user_email = self._get_user_from_id(req.payload["event"]["user"])
                 channel, channel_type = self._get_channel_from_id(req.payload["event"]["channel"])
+
                 tags = self._get_tags_from_message(req.payload["event"]["blocks"])
                 slack_msg = SlackMessage(
                     user=user or "",
@@ -254,8 +279,8 @@ class SlackAdapterManager(AdapterManagerImpl):
                 # grab channel or DM
                 if hasattr(slack_msg, "channel_id") and slack_msg.channel_id:
                     channel_id = slack_msg.channel_id
+
                 elif hasattr(slack_msg, "channel") and slack_msg.channel:
-                    # TODO DM
                     channel_id = self._get_channel_from_name(slack_msg.channel)
 
                 # pull text or reaction
@@ -266,6 +291,7 @@ class SlackAdapterManager(AdapterManagerImpl):
                         name=slack_msg.reaction,
                         timestamp=slack_msg.thread,
                     )
+
                 elif hasattr(slack_msg, "msg") and slack_msg.msg:
                     try:
                         # send text to channel
@@ -274,11 +300,10 @@ class SlackAdapterManager(AdapterManagerImpl):
                             text=getattr(slack_msg, "msg", ""),
                         )
                     except SlackApiError:
-                        # TODO
-                        ...
+                        log.exception("Failed to send message to Slack")
                 else:
                     # cannot send empty message, log an error
-                    log.error(f"Received malformed SlackMessage instance: {slack_msg}")
+                    log.exception(f"Received malformed SlackMessage instance: {slack_msg}")
 
             if not self._inqueue.empty():
                 # pull all SlackMessages from queue

csp_adapter_slack/examples/hello.py

@@ -0,0 +1,44 @@
+import csp
+from csp import ts
+
+from csp_adapter_slack import SlackAdapterConfig, SlackAdapterManager, SlackMessage
+
+config = SlackAdapterConfig(
+    app_token=".app_token",
+    bot_token=".bot_token",
+)
+
+
+@csp.node
+def add_reaction_when_mentioned(msg: ts[SlackMessage]) -> ts[SlackMessage]:
+    """Add a reaction to every message that starts with hello."""
+    if msg.msg.lower().startswith("hello"):
+        return SlackMessage(
+            channel=msg.channel,
+            thread=msg.thread,
+            reaction="wave",
+        )
+
+
+def graph():
+    # Create a DiscordAdapter object
+    adapter = SlackAdapterManager(config)
+
+    # Subscribe and unroll the messages
+    msgs = csp.unroll(adapter.subscribe())
+
+    # Print it out locally for debugging
+    csp.print("msgs", msgs)
+
+    # Add the reaction node
+    reactions = add_reaction_when_mentioned(msgs)
+
+    # Print it out locally for debugging
+    csp.print("reactions", reactions)
+
+    # Publish the reactions
+    adapter.publish(reactions)
+
+
+if __name__ == "__main__":
+    csp.run(graph, realtime=True)

csp_adapter_slack/mention.py

@@ -0,0 +1,10 @@
+__all__ = ("mention_user",)
+
+
+def mention_user(userid: str) -> str:
+    """Convenience method, more difficult to do in symphony but we want slack to be symmetric"""
+    if userid.startswith("<@") and userid.endswith(">"):
+        return userid
+    if userid.startswith("@"):
+        return f"<{userid}>"
+    return f"<@{userid}>"

csp_adapter_slack/message.py

@@ -0,0 +1,40 @@
+from typing import List
+
+from csp.impl.struct import Struct
+
+__all__ = ("SlackMessage",)
+
+
+class SlackMessage(Struct):
+    user: str
+    """name of the author of the message"""
+
+    user_email: str
+    """email of the author of the message, if available"""
+
+    user_id: str
+    """platform-specific id of the author of the message, if available"""
+
+    tags: List[str]
+    """list of users tagged in the `msg` of the message"""
+
+    channel: str
+    """name of the channel for the slack message, if available"""
+
+    channel_id: str
+    """id of the channel for the slack message, if available"""
+
+    channel_type: str
+    """type of the channel. either "message", "public", or "private" """
+
+    msg: str
+    """parsed text of the message"""
+
+    reaction: str
+    """emoji reaction to put on a thread. Exclusive with `msg`, requires `thread`"""
+
+    thread: str
+    """thread to post msg under, or msg id on which to apply reaction"""
+
+    payload: dict
+    """raw slack message payload"""

csp_adapter_slack/tests/test_adapter.py

@@ -134,15 +134,15 @@ class TestSlack:
             reqmock.type = "events_api"
             reqmock.payload = payload
 
-            # mock out the user/room lookup responses
+            # mock out the user/channel lookup responses
             mock_user_response = MagicMock(name="users_info_mock")
             mock_user_response.status_code = 200
             mock_user_response.data = {"user": {"profile": {"real_name_normalized": "johndoe", "email": "johndoe@some.email"}, "name": "blerg"}}
             clientmock.return_value.web_client.users_info.return_value = mock_user_response
-            mock_room_response = MagicMock(name="conversations_info_mock")
-            mock_room_response.status_code = 200
-            mock_room_response.data = {"channel": {"is_im": False, "is_private": True, "name": "a private channel"}}
-            clientmock.return_value.web_client.conversations_info.return_value = mock_room_response
+            mock_channel_response = MagicMock(name="conversations_info_mock")
+            mock_channel_response.status_code = 200
+            mock_channel_response.data = {"channel": {"is_im": False, "is_private": True, "name": "a private channel"}}
+            clientmock.return_value.web_client.conversations_info.return_value = mock_channel_response
             mock_list_response = MagicMock(name="conversations_list_mock")
             mock_list_response.status_code = 200
             mock_list_response.data = {
@@ -193,7 +193,7 @@ class TestSlack:
 
             # check all inbound mocks got called
             if payload == PUBLIC_CHANNEL_MENTION_PAYLOAD:
-                assert clientmock.return_value.web_client.users_info.call_count == 2
+                assert clientmock.return_value.web_client.users_info.call_count == 3
             else:
                 assert clientmock.return_value.web_client.users_info.call_count == 1
             assert clientmock.return_value.web_client.conversations_info.call_count == 1

{csp_adapter_slack-0.2.0.dist-info → csp_adapter_slack-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: csp_adapter_slack
-Version: 0.2.0
+Version: 0.3.0
 Summary: A csp adapter for slack
 Project-URL: Repository, https://github.com/point72/csp-adapter-slack
 Project-URL: Homepage, https://github.com/point72/csp-adapter-slack
@@ -206,32 +206,32 @@ License:                                  Apache License
            WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
            See the License for the specific language governing permissions and
            limitations under the License.
+License-File: LICENSE
 Keywords: chat,chatbot,csp,slack,stream-processing
 Classifier: Development Status :: 4 - Beta
 Classifier: Framework :: Jupyter
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Requires-Dist: csp
 Requires-Dist: pydantic>=2
 Requires-Dist: slack-sdk>=3
 Provides-Extra: develop
 Requires-Dist: bump-my-version; extra == 'develop'
 Requires-Dist: check-manifest; extra == 'develop'
-Requires-Dist: codespell<2.4,>=2.2.6; extra == 'develop'
+Requires-Dist: codespell<2.5,>=2.2.6; extra == 'develop'
 Requires-Dist: hatchling; extra == 'develop'
 Requires-Dist: mdformat-tables<1.1,>=1; extra == 'develop'
 Requires-Dist: mdformat<0.8,>=0.7.17; extra == 'develop'
 Requires-Dist: pytest; extra == 'develop'
 Requires-Dist: pytest-cov; extra == 'develop'
-Requires-Dist: ruff<0.9,>=0.5; extra == 'develop'
+Requires-Dist: ruff<0.12,>=0.5; extra == 'develop'
 Requires-Dist: twine<7,>=5; extra == 'develop'
 Provides-Extra: test
 Requires-Dist: pytest; extra == 'test'
@@ -249,8 +249,14 @@ A [csp](https://github.com/point72/csp) adapter for [slack](https://slack.com)
 
 ## Features
 
+The Slack adapter allows for reading and writing of messages from the [Slack](https://slack.com/) message platform using the [`slack-sdk`](https://tools.slack.dev/python-slack-sdk/).
+
 [More information is available in our wiki](https://github.com/Point72/csp-adapter-slack/wiki)
 
+## Chat Framework
+
+[`csp-bot`](https://github.com/Point72/csp-bot) is a framework for writing cross-platform, command oriented chat bots.
+
 ## Installation
 
 Install with `pip`:
@@ -267,4 +273,4 @@ conda install csp csp-adapter-slack -c conda-forge
 
 ## License
 
-This software is licensed under the Apache 2.0 license. See the [LICENSE](LICENSE) file for details.
+This software is licensed under the Apache 2.0 license. See the [LICENSE](https://github.com/Point72/csp-adapter-slack/blob/main/LICENSE) file for details.

csp_adapter_slack-0.3.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+csp_adapter_slack/__init__.py,sha256=Nt8hM5Q1Kb7-X7xKfVei0AQCAV6UWBdU41lTXp0qylw,122
+csp_adapter_slack/adapter.py,sha256=bAs6cTE-ifH-uzheYG9BqkcKbWmfWlt1Txaih60hehM,14890
+csp_adapter_slack/adapter_config.py,sha256=ltLdb9ESSz3O95xCfxKHMNoxDAHYjo07r4DwgGs4SYc,1221
+csp_adapter_slack/mention.py,sha256=q8xTDY6ETP-yvhNheUSqEsrzXXLUH-WL6WBESaohvDM,330
+csp_adapter_slack/message.py,sha256=UUa1ZPj-d-_HyXHt-Sv7S96LsdFc1Zp7QtVT-yUOcg0,994
+csp_adapter_slack/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+csp_adapter_slack/examples/hello.py,sha256=1lvSqtI59CkFnW2PwxxkHo242ivMcDBU8YV8w0BDocQ,1063
+csp_adapter_slack/tests/test_adapter.py,sha256=a-Sjr_jtUUWlI4eNB9Z8x7nF0wco3-7Yt8mLTW2XB0Y,8539
+csp_adapter_slack-0.3.0.dist-info/METADATA,sha256=qdCdNeMXW6Lm3k3yN2g05Jjdu_DTSnaMOpFPCW62lJs,16009
+csp_adapter_slack-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+csp_adapter_slack-0.3.0.dist-info/licenses/LICENSE,sha256=Dz904Ba4BFTsnLKU2nrCbcc8nqE1hZvaVDl_oOPueDE,11344
+csp_adapter_slack-0.3.0.dist-info/RECORD,,

{csp_adapter_slack-0.2.0.dist-info → csp_adapter_slack-0.3.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.26.3
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

csp_adapter_slack-0.2.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-csp_adapter_slack/__init__.py,sha256=ra0C5usflKAD5tO-ck7bd1BdChQuex6Y7UsXC9sXSks,76
-csp_adapter_slack/adapter.py,sha256=21KnGE7eP68sGB63Fy3atG1BLRxIjxDRgQg6cW0dUhc,13650
-csp_adapter_slack/adapter_config.py,sha256=ltLdb9ESSz3O95xCfxKHMNoxDAHYjo07r4DwgGs4SYc,1221
-csp_adapter_slack/tests/test_adapter.py,sha256=OczVSohe9U4kp0OS9rEjUb9wvOgmcoswY4gGBSuSqpo,8524
-csp_adapter_slack-0.2.0.dist-info/METADATA,sha256=X63AjcpUqwxqsUccB3zkqlRBsAkvtupksZs9itCI6oY,15658
-csp_adapter_slack-0.2.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-csp_adapter_slack-0.2.0.dist-info/licenses/LICENSE,sha256=Dz904Ba4BFTsnLKU2nrCbcc8nqE1hZvaVDl_oOPueDE,11344
-csp_adapter_slack-0.2.0.dist-info/RECORD,,