webex-bot 0.5.0__py2.py3-none-any.whl → 0.5.1__py2.py3-none-any.whl

webex_bot/__init__.py

@@ -1,4 +1,4 @@
 """Top-level package for Webex Bot."""
 
 __author__ = """Finbarr Brady"""
-__version__ = '0.5.0'
+__version__ = '0.5.1'

webex_bot/webex_bot.py

@@ -31,7 +31,8 @@ class WebexBot(WebexWebsocketClient):
                  bot_help_subtitle="Here are my available commands. Click one to begin.",
                  threads=True,
                  help_command=None,
-                 log_level="INFO"):
+                 log_level="INFO",
+                 proxies=None):
         """
         Initialise WebexBot.
 
@@ -46,7 +47,7 @@ class WebexBot(WebexWebsocketClient):
         @param threads: If True, respond to msg by creating a thread.
         @param help_command: If None, use internal HelpCommand, otherwise override.
         @param log_level: Set loggin level.
-
+        @param proxies: Dictionary of proxies for connections.
         """
 
         coloredlogs.install(level=os.getenv("LOG_LEVEL", log_level),
@@ -58,7 +59,8 @@ class WebexBot(WebexWebsocketClient):
                                       teams_bot_token,
                                       on_message=self.process_incoming_message,
                                       on_card_action=self.process_incoming_card_action,
-                                      device_url=device_url)
+                                      device_url=device_url,
+                                      proxies=proxies)
 
         if help_command is None:
             self.help_command = HelpCommand(

webex_bot/websockets/webex_websocket_client.py

@@ -11,6 +11,12 @@ import requests
 import websockets
 from webexteamssdk import WebexTeamsAPI
 
+try:
+    from websockets_proxy import Proxy, proxy_connect
+except ImportError:
+    Proxy = None
+    proxy_connect = None
+
 logger = logging.getLogger(__name__)
 
 DEFAULT_DEVICE_URL = "https://wdm-a.wbx2.com/wdm/api/v1"
@@ -36,16 +42,23 @@ class WebexWebsocketClient(object):
                  access_token,
                  device_url=DEFAULT_DEVICE_URL,
                  on_message=None,
-                 on_card_action=None):
+                 on_card_action=None,
+                 proxies=None):
         self.access_token = access_token
-        self.teams = WebexTeamsAPI(access_token=access_token)
+        self.teams = WebexTeamsAPI(access_token=access_token, proxies=proxies)
         self.device_url = device_url
         self.device_info = None
         self.on_message = on_message
         self.on_card_action = on_card_action
+        self.proxies = proxies
         self.websocket = None
         self.share_id = None
 
+        if self.proxies:
+            # Connecting through a proxy
+            if proxy_connect is None:
+                raise ImportError("Failed to load libraries for proxy, maybe forgot [proxy] option during installation.")
+
     def _process_incoming_websocket_message(self, msg):
         """
         Handle websocket data.
@@ -124,7 +137,8 @@ class WebexWebsocketClient(object):
                                                             f"{verb}/{activity_id}")
         headers = {"Authorization": f"Bearer {self.access_token}"}
         conversation_message = requests.get(conversation_message_url,
-                                            headers=headers).json()
+                                            headers=headers,
+                                            proxies=self.proxies).json()
         logger.debug(f"conversation_message={conversation_message}")
         return conversation_message['id']
 
@@ -197,7 +211,20 @@ class WebexWebsocketClient(object):
         async def _connect_and_listen():
             ws_url = self.device_info['webSocketUrl']
             logger.info(f"Opening websocket connection to {ws_url}")
-            async with websockets.connect(ws_url, ssl=ssl_context) as _websocket:
+
+            if self.proxies and "wss" in self.proxies:
+                logger.info(f"Using proxy for websocket connection: {self.proxies['wss']}")
+                proxy = Proxy.from_url(self.proxies["wss"])
+                connect = proxy_connect(ws_url, ssl=ssl_context, proxy=proxy)
+            elif self.proxies and "https" in self.proxies:
+                logger.info(f"Using proxy for websocket connection: {self.proxies['https']}")
+                proxy = Proxy.from_url(self.proxies["https"])
+                connect = proxy_connect(ws_url, ssl=ssl_context, proxy=proxy)
+            else:
+                logger.debug(f"Not using proxy for websocket connection.")
+                connect = websockets.connect(ws_url, ssl=ssl_context)
+
+            async with connect as _websocket:
                 self.websocket = _websocket
                 logger.info("WebSocket Opened.")
                 msg = {'id': str(uuid.uuid4()),
@@ -216,4 +243,4 @@ class WebexWebsocketClient(object):
                 logger.error('could not create device info')
                 raise Exception("No WDM device info")
             # trigger re-connect
-            asyncio.get_event_loop().run_until_complete(_connect_and_listen())
+            asyncio.get_event_loop().run_until_complete(_connect_and_listen())

{webex_bot-0.5.0.dist-info → webex_bot-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: webex-bot
-Version: 0.5.0
+Version: 0.5.1
 Summary: Python package for a Webex Bot based on websockets.
 Home-page: https://github.com/fbradyirl/webex_bot
 Author: Finbarr Brady
@@ -21,6 +21,8 @@ Requires-Dist: webexteamssdk ==1.6.1
 Requires-Dist: coloredlogs
 Requires-Dist: websockets ==11.0.3
 Requires-Dist: backoff
+Provides-Extra: proxy
+Requires-Dist: websockets-proxy >=0.1.1 ; extra == 'proxy'
 
 # Introduction
 
@@ -57,6 +59,10 @@ You can find a sample project, using OpenAI/ChatGPT with this library here: http
 
 `pip install webex_bot`
 
+If you need optional proxy support, use this command instead:
+
+`pip install webex_bot[proxy]`
+
 2. On the Webex Developer portal, create a new [bot token][3] and expose it as an environment variable.
 
 ```sh
@@ -75,11 +81,19 @@ import os
 from webex_bot.commands.echo import EchoCommand
 from webex_bot.webex_bot import WebexBot
 
+# (Optional) Proxy configuration
+# Supports https or wss proxy, wss prioritized.
+proxies = {
+    'https': 'http://proxy.esl.example.com:80',
+    'wss': 'socks5://proxy.esl.example.com:1080'
+}
+
 # Create a Bot Object
 bot = WebexBot(teams_bot_token=os.getenv("WEBEX_TEAMS_ACCESS_TOKEN"),
                approved_rooms=['06586d8d-6aad-4201-9a69-0bf9eeb5766e'],
                bot_name="My Teams Ops Bot",
-               include_demo_commands=True)
+               include_demo_commands=True,
+               proxies=proxies)
 
 # Add new commands for the bot to listen out for.
 bot.add_command(EchoCommand())
@@ -357,13 +371,16 @@ and off you go!
 
 ### 0.5.0 (2024-Apr-25)
 
-* Add max backoff time (#55)
-* Attached files. Help, threading and log level overrides. (#54)
-* add stop() call to gracefully exit the bot (#42)
-* feat(20231212): add help image size parameter (#46)
-* update websockets to 11.0.3 (#43)
+* Add max backoff time ([#55][pr55])
+* Attached files. Help, threading and log level overrides. ([#54][pr54])
+* add stop() call to gracefully exit the bot ([#42][pr42])
+* feat(20231212): add help image size parameter ([#46][pr46])
+* update websockets to 11.0.3 ([#43][pr43])
 * Fix for help_command syntax issue
 
+### 0.5.1 (2024-Apr-25)
+
+* Add Proxy Support. ([#56][pr56])
 
 [1]: https://github.com/aaugustin/websockets
 
@@ -379,6 +396,18 @@ and off you go!
 
 [7]: https://eurl.io/#TeBLqZjLs
 
+[pr43]: https://github.com/fbradyirl/webex_bot/pull/43
+
+[pr46]: https://github.com/fbradyirl/webex_bot/pull/46
+
+[pr42]: https://github.com/fbradyirl/webex_bot/pull/42
+
+[pr54]: https://github.com/fbradyirl/webex_bot/pull/54
+
+[pr55]: https://github.com/fbradyirl/webex_bot/pull/55
+
+[pr56]: https://github.com/fbradyirl/webex_bot/pull/56
+
 [i1]: https://github.com/fbradyirl/webex_bot/issues/1
 
 [i2]: https://github.com/fbradyirl/webex_bot/issues/2

{webex_bot-0.5.0.dist-info → webex_bot-0.5.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
-webex_bot/__init__.py,sha256=o9hxpp3-rpiFtnd86lMQGKVy9tWQAb1JazWjTWECDtA,95
+webex_bot/__init__.py,sha256=fIrggpbtbmnosDI6vwtRg8leUTo8BeJZJZqojDOzZGM,95
 webex_bot/exceptions.py,sha256=qs9yVitfJtvxwBMC8uCvTDOxUQ_oZjWFf1dU8Oaue14,740
 webex_bot/formatting.py,sha256=jvPKym-z8CIJygpPVTVbt6vFXQo9_HQHpRDJB-nh-SI,382
-webex_bot/webex_bot.py,sha256=Fvga8PUgBtFkdkYirDpNU35uRkfhdIRPUrFj6z6amYo,20325
+webex_bot/webex_bot.py,sha256=f4qJeNfl1x-dvFyAxcBrAWTT0HB-pAq0Kl3gfjrXYe0,20473
 webex_bot/cards/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 webex_bot/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 webex_bot/commands/echo.py,sha256=YmeRVh56pzsj0_lRHDvgyrv6YqQQdRiusG65dHTh9qU,3397
@@ -10,9 +10,9 @@ webex_bot/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 webex_bot/models/command.py,sha256=Rok661eZM2Mg465CZ4GgctWi1RD-BD5Bruq0FsJ8GDQ,5023
 webex_bot/models/response.py,sha256=1U0EQv8O5R3-demaumxMExn278DG4AeQHhXP_g4xndk,2544
 webex_bot/websockets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-webex_bot/websockets/webex_websocket_client.py,sha256=uBpvpRLmfhgtfDr0Yomg6FNei9CIsCEigNsAEu5c5iY,9427
-webex_bot-0.5.0.dist-info/LICENSE,sha256=93eGb10xmgkBP2Fh_n0E9YDXe0c0oz-FsnAimXG0S4Y,1072
-webex_bot-0.5.0.dist-info/METADATA,sha256=RqiYq2AdWfRXMD9x0h4DoQIXEh764hWry7rBb7zbc-U,11154
-webex_bot-0.5.0.dist-info/WHEEL,sha256=DZajD4pwLWue70CAfc7YaxT1wLUciNBvN_TTcvXpltE,110
-webex_bot-0.5.0.dist-info/top_level.txt,sha256=q1Y0RtYYinR7oXSwL93cK59c2KN_CbMVca8MLWeF63M,10
-webex_bot-0.5.0.dist-info/RECORD,,
+webex_bot/websockets/webex_websocket_client.py,sha256=QUTQUWgnBYg3bSZ857m_vM49Ns9GWHWT69TYP4U6R9I,10627
+webex_bot-0.5.1.dist-info/LICENSE,sha256=93eGb10xmgkBP2Fh_n0E9YDXe0c0oz-FsnAimXG0S4Y,1072
+webex_bot-0.5.1.dist-info/METADATA,sha256=aSiwp8MqPv55j1njJOpHRCF_FnDiB077O-f_hzKqBhg,11991
+webex_bot-0.5.1.dist-info/WHEEL,sha256=DZajD4pwLWue70CAfc7YaxT1wLUciNBvN_TTcvXpltE,110
+webex_bot-0.5.1.dist-info/top_level.txt,sha256=q1Y0RtYYinR7oXSwL93cK59c2KN_CbMVca8MLWeF63M,10
+webex_bot-0.5.1.dist-info/RECORD,,