From fc756f2283f8802e5277cfa7d685a248e4f14cd8 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Mon, 24 Jul 2023 13:05:05 +0200 Subject: [PATCH 01/24] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20pytest-loguru=20=3D?= =?UTF-8?q?=20"^0.2.0"?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index 0535c2b09..748cfeb31 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -85,6 +85,7 @@ pytest = "^7.0" pytest-cov = "^4.1" pytest-asyncio = "^0.21.0" pytest-mock = "^3.11.1" +pytest-loguru = "^0.2.0" [tool.poetry.group.docs] optional = true From 44767ee4414460f4e0beb2a795099b746b000216 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Mon, 24 Jul 2023 13:06:26 +0200 Subject: [PATCH 02/24] =?UTF-8?q?=E2=9C=85=20pytest-loguru?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/conftest.py | 16 ---------------- 1 file changed, 16 deletions(-) delete mode 100644 tests/conftest.py diff --git a/tests/conftest.py b/tests/conftest.py deleted file mode 100644 index dde2b62b7..000000000 --- a/tests/conftest.py +++ /dev/null @@ -1,16 +0,0 @@ -import pytest -from _pytest.logging import LogCaptureFixture -from loguru import logger - - -@pytest.fixture -def caplog(caplog: LogCaptureFixture): - handler_id = logger.add( - caplog.handler, - format="{message}", - level=0, - filter=lambda record: record["level"].no >= caplog.handler.level, - enqueue=False, # Set to 'True' if your test is spawning child processes. - ) - yield caplog - logger.remove(handler_id) \ No newline at end of file From ca9635ead7869978086732bb87c328a46a712c59 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Mon, 24 Jul 2023 18:33:30 +0200 Subject: [PATCH 03/24] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20dep?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 748cfeb31..72904bb86 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,7 +50,7 @@ ping3 = "^4.0.4" ccxt = "^4.0.0" dxsp = "^4.1.0" findmyorder = "^1.6.1" -iamlistening = "^2.0.0" +iamlistening = "^2.1.0" talkytrend = "^1.10.0" From 71bd071b20746b6191fb56a6df2627f40dacd949 Mon Sep 17 00:00:00 2001 From: mraniki Date: Mon, 24 Jul 2023 21:29:50 +0200 Subject: [PATCH 04/24] :loud_sound: loguru sink change --- tt/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tt/config.py b/tt/config.py index fb9bbc0ec..d713f215c 100644 --- a/tt/config.py +++ b/tt/config.py @@ -53,7 +53,7 @@ def loguru_setup(): log.remove() # log.configure(**config) log.add( - sink=sys.stderr, + sink=sys.stdout, level=settings.loglevel, ) From c75836ee9e0dc19da2962cd42e2503bb5af7497c Mon Sep 17 00:00:00 2001 From: mraniki Date: Mon, 24 Jul 2023 22:03:46 +0200 Subject: [PATCH 05/24] :lipstick: Add TalkyTrendPlugin cron task for fetching key events. --- tt/plugins/default_plugins/talkytrend_plugin.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tt/plugins/default_plugins/talkytrend_plugin.py b/tt/plugins/default_plugins/talkytrend_plugin.py index 6106c7e1d..d662460dc 100644 --- a/tt/plugins/default_plugins/talkytrend_plugin.py +++ b/tt/plugins/default_plugins/talkytrend_plugin.py @@ -24,6 +24,9 @@ async def start(self): await self.plugin_notify_cron_task( user_name="talky_signal", function=self.trend.check_signal) + await self.plugin_notify_cron_task( + user_name="talky_events", + function=self.trend.fetch_key_events) async def stop(self): """Stops the TalkyTrend plugin""" From 1319796dc4039e7a1958b342cb18ee6137069f22 Mon Sep 17 00:00:00 2001 From: mraniki Date: Mon, 24 Jul 2023 20:13:19 +0000 Subject: [PATCH 06/24] Update Requirements --- .requirements/requirements.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.requirements/requirements.txt b/.requirements/requirements.txt index eafd27403..2b220eb04 100644 --- a/.requirements/requirements.txt +++ b/.requirements/requirements.txt @@ -21,7 +21,7 @@ click==8.1.6 ; python_version >= "3.10" and python_version < "4.0" colorama==0.4.6 ; python_version >= "3.10" and python_version < "4.0" and (sys_platform == "win32" or platform_system == "Windows") cryptography==41.0.2 ; python_version >= "3.10" and python_version < "4.0" cytoolz==0.12.2 ; python_version >= "3.10" and python_version < "4" and implementation_name == "cpython" -dxsp==4.1.2 ; python_version >= "3.10" and python_version < "4.0" +dxsp==4.2.1 ; python_version >= "3.10" and python_version < "4.0" dynaconf==3.2.0 ; python_version >= "3.10" and python_version < "4.0" emoji==2.6.0 ; python_version >= "3.10" and python_version < "4.0" eth-abi==4.1.0 ; python_version >= "3.10" and python_version < "4" @@ -35,7 +35,7 @@ eth-typing==3.4.0 ; python_version >= "3.10" and python_version < "4" eth-utils==2.2.0 ; python_version >= "3.10" and python_version < "4" exceptiongroup==1.1.2 ; python_version >= "3.10" and python_version < "3.11" fastapi==0.100.0 ; python_version >= "3.10" and python_version < "4.0" -findmyorder==1.6.1 ; python_version >= "3.10" and python_version < "4.0" +findmyorder==1.7.0 ; python_version >= "3.10" and python_version < "4.0" frozendict==2.3.8 ; python_version >= "3.10" and python_version < "4.0" frozenlist==1.4.0 ; python_version >= "3.10" and python_version < "4.0" future==0.18.3 ; python_version >= "3.10" and python_version < "4.0" @@ -102,7 +102,7 @@ sniffio==1.3.0 ; python_version >= "3.10" and python_version < "4.0" soupsieve==2.4.1 ; python_version >= "3.10" and python_version < "4.0" starlette==0.27.0 ; python_version >= "3.10" and python_version < "4.0" talkytrend==1.11.0 ; python_version >= "3.10" and python_version < "4.0" -telethon==1.29.1 ; python_version >= "3.10" and python_version < "4.0" +telethon==1.29.2 ; python_version >= "3.10" and python_version < "4.0" toml==0.10.2 ; python_version >= "3.10" and python_version < "4.0" toolz==0.12.0 ; python_version >= "3.10" and python_version < "4" and (implementation_name == "pypy" or implementation_name == "cpython") tradingview-ta==3.3.0 ; python_version >= "3.10" and python_version < "4.0" From 97a25b57bf4b06c5a1ba49c61d19b8d1fcfa62cd Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 08:51:20 +0200 Subject: [PATCH 07/24] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20Update=20for=20talky?= =?UTF-8?q?trend=20changes?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/index.md | 294 ------------------ tests/test_talkytrend_plugin.py | 8 +- .../default_plugins/dex_exchange_plugin.py | 1 - .../default_plugins/talkytrend_plugin.py | 15 +- tt/plugins/plugin_manager.py | 2 +- 5 files changed, 9 insertions(+), 311 deletions(-) diff --git a/docs/index.md b/docs/index.md index 3bed098a5..6b1100df6 100644 --- a/docs/index.md +++ b/docs/index.md @@ -22,300 +22,6 @@ Easily deploy via Docker on self-hosted platform or Paas.
screenshot -
-Get started - -
    -
  1. Create your channel/room and your platform bot -
    2. -
  2. Get your -
    3. -
  3. Create your config file settings.toml or use env variables
    4. - -
    -settings example - - - - - -
    - -
  4. Deploy via: -
      -
    • docker - docker pull mraniki/tt:latest or docker pull ghcr.io/mraniki/tt:latest
      • -
    • locally - git clone https://github.com/mraniki/tt:main && pip install -r requirements.txt
      • -
    5. -
  5. Start your container or if deployed locally use python3 bot.py to start
    6. -
  6. Try it now
    7. - -
- -
- -
- -
-Config - -
-env vars list -

- - - - -
- -
- -
- -
-Plugins - -

Talky

-
- - - - - - - -
- -
-
- -
-
- - 		- Connect CEX and DEX exchanges
- across multi messaging platforms
- with plugin support. -
-
- -

FindMyOrder

- - - - - - - -
-
-
-
-
-
-
- 		- Find My order,
- a parsing package to find trading order -
- -
How to use it
-
-
-      from findmyorder import FindMyOrder
-         fmo = FindMyOrder()
-         msg_order = "buy EURUSD sl=1000 tp=1000 q=1 comment=FOMC"
-         order = await fmo.get_order(msg_order)
-         #{'action': 'BUY', 'instrument': 'EURUSD', 'stop_loss': '1000', 'take_profit': '1000', 'quantity': '2', 'order_type': None, 'leverage_type': None, 'comment': None, 'timestamp': datetime.datetime(2023, 5, 3, 12, 10, 28, 731282, tzinfo=datetime.timezone.utc)}
-
-
- -
Example
- -https://github.com/mraniki/findmyorder/blob/6211e686abf1321ab3d5a8ac33068f854b0e9336/examples/example.py#L1-L90 - -

DXSP

-
- - - - - - - - -
-
-
-
-
-
-
- 		-Swap made easy
-Trade on any blockchains with uniswap based router or 0x protocol. -
- -
How to use it
-
-
-   from dxsp import DexSwap
-    dex = DexSwap()
-    #BUY 10 USDT to SWAP with BITCOIN
-    demo_tx = await dex.get_swap('USDT','wBTC',10)
-    print("demo_tx ", demo_tx)
-
-
- -
Example
- -https://github.com/mraniki/dxsp/blob/f76fd035eddadc4de2a8509a7c26250c187b0658/examples/example.py#L1-L68 - - -

IamListening

-
- - - - - - - -
-
-
-
-
-
-
- 		- A python package to listen to messaging platforms,
- such as discord, telegram and matrix -
- -
How to use it
-
-
-      from iamlistening import Listener
-        listener = Listener()
-        task = asyncio.create_task(listener.run_forever())
-        while True:
-          try:
-              msg = await listener.get_latest_message()
-              if msg:
-                  print(f"Frasier👂: {msg}"
-          except Exception as error:
-              print(error)
-        await task
-
-
- -
Example
- -https://github.com/mraniki/iamlistening/blob/52db0f4c194fe9edd12f48199737b4e1c73f7194/examples/example.py#L1-L54 - -

TalkyTrend

-
- - - - - - - -
-
-
-
-
-
-
- 		-Retrieve asset trend and economic data.
-Trading view connectivity with signal scanner
-News connectivity
- FOMC reminder
-
- -
How to use it
-
-
-   from talkytrend import TalkyTrend
-   talky = TalkyTrend()
-    result = await talky.check_signal()
-    #  BUY
-    result = await talky.fetch_key_events()
-    print(result)
-    #  Title:  FDA advisers say new Alzheimer’s drug lecanemab slows cognitive decline
-    # Description:  Panel’s opinion could pave way for full regulatory approval next month for treatment of disease that affects 6.5m Americans
-    monitor = await talky.scanner() #ongoing monitoring
-    # New signal for BTCUSD (4h): STRONG_SELL
-    # Key event: {'title': 'OPEC-JMMC Meetings', 'country': 'ALL', 'date': '2023-06-04T06:15:00-04:00', 'impact': 'High', 'forecast': '', 'previous': ''}\
-    # Key news: FDA advisers say new Alzheimer’s drug lecanemab slows cognitive decline
-
-
- -
Example
-https://github.com/mraniki/talkytrend/blob/af472db335afec4e6a643077f7483d030e8511ac/examples/example.py#L1-L50 - -



diff --git a/tests/test_talkytrend_plugin.py b/tests/test_talkytrend_plugin.py index 975752e46..cdc579b22 100644 --- a/tests/test_talkytrend_plugin.py +++ b/tests/test_talkytrend_plugin.py @@ -52,14 +52,14 @@ async def test_plugin_tv(plugin): @pytest.mark.asyncio async def test_plugin_trend(plugin): """Test notification """ - plugin.check_signal = AsyncMock() + plugin.fetch_signal = AsyncMock() await plugin.handle_message(f"{settings.bot_prefix}{settings.bot_command_trend}") - plugin.check_signal.assert_awaited_once + plugin.fetch_signal.assert_awaited_once @pytest.mark.asyncio async def test_plugin_news(plugin): """Test notification """ - plugin.fetch_key_feed = AsyncMock() + plugin.fetch_feed = AsyncMock() await plugin.handle_message(f"{settings.bot_prefix}{settings.bot_command_news}") - plugin.fetch_key_feed.assert_awaited_once \ No newline at end of file + plugin.fetch_feed.assert_awaited_once \ No newline at end of file diff --git a/tt/plugins/default_plugins/dex_exchange_plugin.py b/tt/plugins/default_plugins/dex_exchange_plugin.py index f5abcb13e..9a1ce92b1 100644 --- a/tt/plugins/default_plugins/dex_exchange_plugin.py +++ b/tt/plugins/default_plugins/dex_exchange_plugin.py @@ -65,4 +65,3 @@ async def handle_message(self, msg): if command in command_mapping: function = command_mapping[command] await self.send_notification(f"{await function()}") - diff --git a/tt/plugins/default_plugins/talkytrend_plugin.py b/tt/plugins/default_plugins/talkytrend_plugin.py index d662460dc..ff3826a60 100644 --- a/tt/plugins/default_plugins/talkytrend_plugin.py +++ b/tt/plugins/default_plugins/talkytrend_plugin.py @@ -18,15 +18,9 @@ def __init__(self): async def start(self): """Starts the TalkyTrend plugin""" if self.enabled: - await self.plugin_notify_schedule_task( - user_name="talky_feed", - function=self.trend.fetch_key_feed) await self.plugin_notify_cron_task( - user_name="talky_signal", - function=self.trend.check_signal) - await self.plugin_notify_cron_task( - user_name="talky_events", - function=self.trend.fetch_key_events) + user_name="talky_monitor", + function=self.trend.monitor) async def stop(self): """Stops the TalkyTrend plugin""" @@ -49,11 +43,10 @@ async def handle_message(self, msg): settings.bot_command_help: self.trend.get_talkytrend_help, settings.bot_command_info: self.trend.get_talkytrend_info, settings.bot_command_tv: self.trend.get_tv, - settings.bot_command_trend: self.trend.check_signal, - settings.bot_command_news: self.trend.fetch_key_feed, + settings.bot_command_trend: self.trend.fetch_signal, + settings.bot_command_news: self.trend.fetch_feed, } if command in command_mapping: function = command_mapping[command] await self.send_notification(f"{await function()}") - diff --git a/tt/plugins/plugin_manager.py b/tt/plugins/plugin_manager.py index 1f4f34e00..36e1d7048 100644 --- a/tt/plugins/plugin_manager.py +++ b/tt/plugins/plugin_manager.py @@ -105,7 +105,7 @@ async def plugin_notify_cron_task( user_timezone="UTC", function=None): """Handles task cron scheduling for notification - monday to Friday at 8AM, 12PM and 4PM with time being UTC based""" + monday to Friday at 8AM, 12PM and 4PM UTC based""" if function: self.scheduler.add_task( name=user_name, From 4f75cc75fe63df88685474eec2b024b671608738 Mon Sep 17 00:00:00 2001 From: mraniki Date: Tue, 25 Jul 2023 06:53:35 +0000 Subject: [PATCH 08/24] Update Requirements --- .requirements/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.requirements/requirements.txt b/.requirements/requirements.txt index 2b220eb04..98add2848 100644 --- a/.requirements/requirements.txt +++ b/.requirements/requirements.txt @@ -13,7 +13,7 @@ asyncz==0.4.0 ; python_version >= "3.10" and python_version < "4.0" attrs==23.1.0 ; python_version >= "3.10" and python_version < "4.0" beautifulsoup4==4.12.2 ; python_version >= "3.10" and python_version < "4.0" bitarray==2.8.0 ; python_version >= "3.10" and python_version < "4" -ccxt==4.0.36 ; python_version >= "3.10" and python_version < "4.0" +ccxt==4.0.38 ; python_version >= "3.10" and python_version < "4.0" certifi==2023.7.22 ; python_version >= "3.10" and python_version < "4.0" cffi==1.15.1 ; python_version >= "3.10" and python_version < "4.0" charset-normalizer==3.2.0 ; python_version >= "3.10" and python_version < "4.0" @@ -96,7 +96,7 @@ rocketchat-api==1.30.0 ; python_version >= "3.10" and python_version < "4.0" rpds-py==0.9.2 ; python_version >= "3.10" and python_version < "4.0" rsa==4.9 ; python_version >= "3.10" and python_version < "4" setuptools==68.0.0 ; python_version >= "3.10" and python_version < "4.0" -simplematrixbotlib==2.9.0 ; python_version >= "3.10" and python_version < "4.0" +simplematrixbotlib==2.9.1 ; python_version >= "3.10" and python_version < "4.0" six==1.16.0 ; python_version >= "3.10" and python_version < "4.0" sniffio==1.3.0 ; python_version >= "3.10" and python_version < "4.0" soupsieve==2.4.1 ; python_version >= "3.10" and python_version < "4.0" From 46fbf97233009b4bb169e3f631da765d73baf798 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 08:57:55 +0200 Subject: [PATCH 09/24] =?UTF-8?q?=F0=9F=94=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tt/talky_settings.toml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tt/talky_settings.toml b/tt/talky_settings.toml index 8c54a9e3f..ef43093c0 100644 --- a/tt/talky_settings.toml +++ b/tt/talky_settings.toml @@ -192,7 +192,6 @@ talkytrend_commands = "📺 /live\n📰 /news\n📊 /trend" bot_command_tv = "live" bot_command_trend = "trend" bot_command_news = "news" -scanner_frequency = 86400 enable_signals = true assets = [ { id ="EURUSD", exchange='FX_IDC',screener="forex"}, @@ -329,7 +328,7 @@ talkytrend_commands = "📺 /live\n📰 /news\n📊 /trend" bot_command_tv = "live" bot_command_trend = "trend" bot_command_news = "news" -scanner_frequency = 86400 + enable_signals = true assets = [ { id ="EURUSD", exchange='FX_IDC',screener="forex"}, From 9428b7a0cd1e010f6b1e23a6449a3d4ef98e1366 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 09:49:24 +0200 Subject: [PATCH 10/24] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20dep?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 72904bb86..66f5ac3f8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,8 +50,8 @@ ping3 = "^4.0.4" ccxt = "^4.0.0" dxsp = "^4.1.0" findmyorder = "^1.6.1" -iamlistening = "^2.1.0" -talkytrend = "^1.10.0" +iamlistening = "^3.0.0" +talkytrend = "^1.12.0" [tool.poetry.group.dev.dependencies] From 56c5522a82848db7023df408a5e7dd9eda6f4557 Mon Sep 17 00:00:00 2001 From: mraniki Date: Tue, 25 Jul 2023 08:18:24 +0000 Subject: [PATCH 11/24] Update Requirements --- .requirements/requirements.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.requirements/requirements.txt b/.requirements/requirements.txt index 98add2848..14e318654 100644 --- a/.requirements/requirements.txt +++ b/.requirements/requirements.txt @@ -23,7 +23,7 @@ cryptography==41.0.2 ; python_version >= "3.10" and python_version < "4.0" cytoolz==0.12.2 ; python_version >= "3.10" and python_version < "4" and implementation_name == "cpython" dxsp==4.2.1 ; python_version >= "3.10" and python_version < "4.0" dynaconf==3.2.0 ; python_version >= "3.10" and python_version < "4.0" -emoji==2.6.0 ; python_version >= "3.10" and python_version < "4.0" +emoji==2.7.0 ; python_version >= "3.10" and python_version < "4.0" eth-abi==4.1.0 ; python_version >= "3.10" and python_version < "4" eth-account==0.9.0 ; python_version >= "3.10" and python_version < "4" eth-hash==0.5.2 ; python_version >= "3.10" and python_version < "4" @@ -47,7 +47,7 @@ html5lib==1.1 ; python_version >= "3.10" and python_version < "4.0" httpcore==0.17.3 ; python_version >= "3.10" and python_version < "4.0" httpx==0.24.1 ; python_version >= "3.10" and python_version < "4.0" hyperframe==6.0.1 ; python_version >= "3.10" and python_version < "4.0" -iamlistening==2.1.0 ; python_version >= "3.10" and python_version < "4.0" +iamlistening==3.0.1 ; python_version >= "3.10" and python_version < "4.0" idna==3.4 ; python_version >= "3.10" and python_version < "4.0" jsonschema-specifications==2023.7.1 ; python_version >= "3.10" and python_version < "4.0" jsonschema==4.18.4 ; python_version >= "3.10" and python_version < "4.0" @@ -101,7 +101,7 @@ six==1.16.0 ; python_version >= "3.10" and python_version < "4.0" sniffio==1.3.0 ; python_version >= "3.10" and python_version < "4.0" soupsieve==2.4.1 ; python_version >= "3.10" and python_version < "4.0" starlette==0.27.0 ; python_version >= "3.10" and python_version < "4.0" -talkytrend==1.11.0 ; python_version >= "3.10" and python_version < "4.0" +talkytrend==1.12.1 ; python_version >= "3.10" and python_version < "4.0" telethon==1.29.2 ; python_version >= "3.10" and python_version < "4.0" toml==0.10.2 ; python_version >= "3.10" and python_version < "4.0" toolz==0.12.0 ; python_version >= "3.10" and python_version < "4" and (implementation_name == "pypy" or implementation_name == "cpython") From d8284e10128188cdf9bf8b771d6d2e39fcdcf454 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 11:26:45 +0200 Subject: [PATCH 12/24] =?UTF-8?q?=E2=9C=85=20Updating=20unit=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pyproject.toml | 4 ++-- tests/test_unit.py | 26 +++++++++++++------- tt/config.py | 59 +++++++++++++++++++++++++++++++++------------- tt/utils.py | 20 ++++++---------- 4 files changed, 68 insertions(+), 41 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 66f5ac3f8..68b2f733b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,8 +50,8 @@ ping3 = "^4.0.4" ccxt = "^4.0.0" dxsp = "^4.1.0" findmyorder = "^1.6.1" -iamlistening = "^3.0.0" -talkytrend = "^1.12.0" +iamlistening = "^3.0.1" +talkytrend = "^1.12.1" [tool.poetry.group.dev.dependencies] diff --git a/tests/test_unit.py b/tests/test_unit.py index d760b1940..323a0cf1a 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -11,9 +11,8 @@ from tt.bot import app from tt.config import settings from tt.plugins.plugin_manager import BasePlugin, PluginManager -from tt.utils import send_notification, start_listener, start_plugins +from tt.utils import run_bot, send_notification, start_bot, start_plugins -#start_bot, run_bot @pytest.fixture(scope="session", autouse=True) def set_test_settings(): @@ -24,6 +23,10 @@ def set_test_settings(): def listener_test(): return Listener() +@pytest.fixture(name="plugin_manager_obj") +def pluginmngr_test(): + return PluginManager() + @pytest.fixture def message(): return "Test message" @@ -65,13 +68,6 @@ async def test_send_notification(caplog): assert "json://localhost/" in caplog.text -@pytest.mark.asyncio -async def test_start_listener(): - listener, task = await start_listener(max_iterations=1) - assert isinstance(listener, Listener) - assert isinstance(task, asyncio.Task) - - @pytest.mark.asyncio async def test_start_plugins(): plugin_manager = AsyncMock(spec=PluginManager) @@ -79,6 +75,18 @@ async def test_start_plugins(): plugin_manager.load_plugins.assert_called_once() +@pytest.mark.asyncio +async def test_start_bot(listener_obj,plugin_manager_obj): + bot = await start_bot(listener_obj,plugin_manager_obj) + assert isinstance(bot, asyncio.Task) + + +@pytest.mark.asyncio +async def test_run_bot(): + bot = await run_bot() + assert isinstance(bot, asyncio.Task) + + @pytest.mark.asyncio async def test_baseplugins(): plugin = BasePlugin diff --git a/tt/config.py b/tt/config.py index d713f215c..487775318 100644 --- a/tt/config.py +++ b/tt/config.py @@ -12,7 +12,7 @@ from asyncz.schedulers.asyncio import AsyncIOScheduler from dynaconf import Dynaconf -from loguru import logger as log +from loguru import logger as loguru_logger ######################################## ### ⚙️ Settings ### @@ -36,37 +36,62 @@ default_env="default", ) - ######################################## ### 🧐 Logging ### ######################################## -# logging.basicConfig( -# format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", -# level=settings.loglevel -# ) -# logger = logging.getLogger("TalkyTrader") +class InterceptHandler(logging.Handler): + def emit(self, record): + # Get corresponding Loguru level if it exists. + try: + level = logger.level(record.levelname).name + except ValueError: + level = record.levelno + + # Find caller from where originated the logged message. + frame, depth = sys._getframe(6), 6 + while frame and frame.f_code.co_filename == logging.__file__: + frame = frame.f_back + depth += 1 + logger.opt( + depth=depth, exception=record.exc_info).log( + level, record.getMessage()) def loguru_setup(): - log.remove() + loguru_logger.remove() # log.configure(**config) - log.add( - sink=sys.stdout, + log_filters = { + "discord": "INFO", + "telethon": "INFO", + "web3": "INFO", + "apprise": "INFO", + "urllib3": "INFO", + } + logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True) + loguru_logger.add( + sink=sys.stderr, level=settings.loglevel, + filter=log_filters, ) - return log + return loguru_logger logger = loguru_setup() -if settings.loglevel == "DEBUG": - logging.getLogger("discord").setLevel(logging.ERROR) - logging.getLogger("telethon").setLevel(logging.ERROR) - logging.getLogger("urllib3").setLevel(logging.ERROR) - logging.getLogger("apprise").setLevel(logging.ERROR) - logging.getLogger("web3").setLevel(logging.ERROR) + +# logging.basicConfig( +# format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", +# level=settings.loglevel +# ) +# logger = logging.getLogger("TalkyTrader") +# if settings.loglevel == "DEBUG": +# logging.getLogger("discord").setLevel(logging.ERROR) +# logging.getLogger("telethon").setLevel(logging.ERROR) +# logging.getLogger("urllib3").setLevel(logging.ERROR) +# logging.getLogger("apprise").setLevel(logging.ERROR) +# logging.getLogger("web3").setLevel(logging.ERROR) ######################################## ### ⏱️ Scheduling ### diff --git a/tt/utils.py b/tt/utils.py index f1de1b145..f598fe956 100644 --- a/tt/utils.py +++ b/tt/utils.py @@ -30,15 +30,6 @@ async def send_notification(msg): body_format=msg_format) -async def start_listener(max_iterations=None): - """ - Start the chat listener. - """ - bot_listener = Listener() - task = asyncio.create_task(bot_listener.run_forever(max_iterations)) - return bot_listener, task - - async def start_plugins(plugin_manager): """ Start all plugins. @@ -48,15 +39,19 @@ async def start_plugins(plugin_manager): loop = asyncio.get_running_loop() loop.create_task(plugin_manager.start_all_plugins()) + async def start_bot(listener, plugin_manager): """ Listen to the message in the bot channel and dispatch to plugins """ + await listener.start() + await start_plugins(plugin_manager) while True: try: - msg = await listener.get_latest_message() + msg = await listener.handler.get_latest_message() if msg and settings.plugin_enabled: + logger.debug("👂 listener: {}", msg) await plugin_manager.process_message(msg) except Exception as error: logger.error("👂 listener: {}", error) @@ -68,7 +63,6 @@ async def run_bot(max_iterations=None): """ Run the chat bot & the plugins. """ - listener, task = await start_listener(max_iterations) + listener = Listener() plugin_manager = PluginManager() - await start_plugins(plugin_manager) - await asyncio.gather(start_bot(listener, plugin_manager), task) \ No newline at end of file + await asyncio.gather(start_bot(listener, plugin_manager)) \ No newline at end of file From 2590b47fcac1a73ad88ccab56b00ae3ae98fb89f Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 11:51:37 +0200 Subject: [PATCH 13/24] =?UTF-8?q?=E2=9C=85=20runbot=20unit=20test=20case?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 323a0cf1a..374b3c98d 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -76,15 +76,17 @@ async def test_start_plugins(): @pytest.mark.asyncio -async def test_start_bot(listener_obj,plugin_manager_obj): - bot = await start_bot(listener_obj,plugin_manager_obj) - assert isinstance(bot, asyncio.Task) - - -@pytest.mark.asyncio -async def test_run_bot(): - bot = await run_bot() - assert isinstance(bot, asyncio.Task) +async def test_run_bot(listener_obj, plugin_manager_obj): + assert isinstance(listener_obj, Listener) + assert isinstance(plugin_manager_obj, PluginManager) + task = asyncio.create_task(start_bot(listener_obj, plugin_manager_obj)) + print(start_bot) + print(task) + assert task is not None + assert start_bot is not None + task.cancel() + with pytest.raises(asyncio.CancelledError): + await task @pytest.mark.asyncio From 9361851a7296b5b1e8671a15f881c009b41c869c Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 12:04:03 +0200 Subject: [PATCH 14/24] =?UTF-8?q?=E2=9C=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 374b3c98d..0bca4ba02 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -11,7 +11,7 @@ from tt.bot import app from tt.config import settings from tt.plugins.plugin_manager import BasePlugin, PluginManager -from tt.utils import run_bot, send_notification, start_bot, start_plugins +from tt.utils import send_notification, start_bot, start_plugins @pytest.fixture(scope="session", autouse=True) From 2207b4cb3ea0db494f5d78b9ef0075ecbec83679 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 12:40:33 +0200 Subject: [PATCH 15/24] =?UTF-8?q?=E2=9C=85=20unit=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 0bca4ba02..e68cf09a6 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -76,7 +76,9 @@ async def test_start_plugins(): @pytest.mark.asyncio -async def test_run_bot(listener_obj, plugin_manager_obj): +async def test_run_bot(listener_obj, plugin_manager_obj, caplog): + plugin_manager_obj.start_plugins = AsyncMock() + plugin_manager_obj.start_all_plugins = AsyncMock() assert isinstance(listener_obj, Listener) assert isinstance(plugin_manager_obj, PluginManager) task = asyncio.create_task(start_bot(listener_obj, plugin_manager_obj)) @@ -84,6 +86,8 @@ async def test_run_bot(listener_obj, plugin_manager_obj): print(task) assert task is not None assert start_bot is not None + # plugin_manager_obj.start_plugins.assert_awaited_once() + # assert any("Plugin loaded" in record.message for record in caplog.records) task.cancel() with pytest.raises(asyncio.CancelledError): await task From 70a1519ed2feb9214f1c4ac1ab5f3652921f4864 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 13:19:49 +0200 Subject: [PATCH 16/24] =?UTF-8?q?=E2=9C=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index e68cf09a6..5505223de 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -7,11 +7,12 @@ import pytest from fastapi.testclient import TestClient from iamlistening import Listener +from loguru import logger from tt.bot import app from tt.config import settings from tt.plugins.plugin_manager import BasePlugin, PluginManager -from tt.utils import send_notification, start_bot, start_plugins +from tt.utils import run_bot, send_notification, start_plugins @pytest.fixture(scope="session", autouse=True) @@ -76,20 +77,16 @@ async def test_start_plugins(): @pytest.mark.asyncio -async def test_run_bot(listener_obj, plugin_manager_obj, caplog): - plugin_manager_obj.start_plugins = AsyncMock() - plugin_manager_obj.start_all_plugins = AsyncMock() - assert isinstance(listener_obj, Listener) - assert isinstance(plugin_manager_obj, PluginManager) - task = asyncio.create_task(start_bot(listener_obj, plugin_manager_obj)) - print(start_bot) +async def test_run_bot(caplog): + task = asyncio.create_task(run_bot()) + print(run_bot) print(task) assert task is not None - assert start_bot is not None - # plugin_manager_obj.start_plugins.assert_awaited_once() + assert run_bot is not None # assert any("Plugin loaded" in record.message for record in caplog.records) task.cancel() with pytest.raises(asyncio.CancelledError): + # assert task is not None await task From 82f92331ef08f9ac2bc1fdebf2642a847baec070 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 14:13:39 +0200 Subject: [PATCH 17/24] =?UTF-8?q?=E2=9C=85=20start=20and=20run=20unit=20te?= =?UTF-8?q?st?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 20 ++++++++++++-------- tt/utils.py | 2 +- 2 files changed, 13 insertions(+), 9 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 5505223de..e184727af 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -7,12 +7,11 @@ import pytest from fastapi.testclient import TestClient from iamlistening import Listener -from loguru import logger from tt.bot import app from tt.config import settings from tt.plugins.plugin_manager import BasePlugin, PluginManager -from tt.utils import run_bot, send_notification, start_plugins +from tt.utils import run_bot, send_notification, start_bot, start_plugins @pytest.fixture(scope="session", autouse=True) @@ -75,18 +74,23 @@ async def test_start_plugins(): await start_plugins(plugin_manager) plugin_manager.load_plugins.assert_called_once() +@pytest.mark.asyncio +async def test_start_bot(listener_obj, plugin_manager_obj): + start = AsyncMock() + task = asyncio.create_task(start_bot(listener_obj, plugin_manager_obj)) + task.cancel() + with pytest.raises(asyncio.CancelledError): + start.assert_awaited + await task + @pytest.mark.asyncio async def test_run_bot(caplog): + start_bot = AsyncMock() task = asyncio.create_task(run_bot()) - print(run_bot) - print(task) - assert task is not None - assert run_bot is not None - # assert any("Plugin loaded" in record.message for record in caplog.records) + start_bot.assert_awaited task.cancel() with pytest.raises(asyncio.CancelledError): - # assert task is not None await task diff --git a/tt/utils.py b/tt/utils.py index f598fe956..7767041cd 100644 --- a/tt/utils.py +++ b/tt/utils.py @@ -39,7 +39,7 @@ async def start_plugins(plugin_manager): loop = asyncio.get_running_loop() loop.create_task(plugin_manager.start_all_plugins()) - + async def start_bot(listener, plugin_manager): """ Listen to the message in the bot channel From e780c9992d8b985c1861698e8c4a2aa04bf35107 Mon Sep 17 00:00:00 2001 From: mraniki Date: Tue, 25 Jul 2023 12:15:51 +0000 Subject: [PATCH 18/24] Update Requirements --- .requirements/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.requirements/requirements.txt b/.requirements/requirements.txt index 14e318654..963025f4b 100644 --- a/.requirements/requirements.txt +++ b/.requirements/requirements.txt @@ -13,7 +13,7 @@ asyncz==0.4.0 ; python_version >= "3.10" and python_version < "4.0" attrs==23.1.0 ; python_version >= "3.10" and python_version < "4.0" beautifulsoup4==4.12.2 ; python_version >= "3.10" and python_version < "4.0" bitarray==2.8.0 ; python_version >= "3.10" and python_version < "4" -ccxt==4.0.38 ; python_version >= "3.10" and python_version < "4.0" +ccxt==4.0.39 ; python_version >= "3.10" and python_version < "4.0" certifi==2023.7.22 ; python_version >= "3.10" and python_version < "4.0" cffi==1.15.1 ; python_version >= "3.10" and python_version < "4.0" charset-normalizer==3.2.0 ; python_version >= "3.10" and python_version < "4.0" From a27b6fe732600caad66ba36eb4167fba3ee163ea Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 14:32:54 +0200 Subject: [PATCH 19/24] =?UTF-8?q?=E2=9C=85=20unit=20test=20increase?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/tests/test_unit.py b/tests/test_unit.py index e184727af..5f9a81129 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -19,6 +19,10 @@ def set_test_settings(): settings.configure(FORCE_ENV_FOR_DYNACONF="testing") +@pytest.fixture(name="message") +def message(): + return "Test message" + @pytest.fixture(name="listener_obj") def listener_test(): return Listener() @@ -83,6 +87,11 @@ async def test_start_bot(listener_obj, plugin_manager_obj): start.assert_awaited await task +@pytest.mark.asyncio +async def test_get_latest_message(listener_obj, message): + await listener_obj.start() + await listener_obj.handler.handle_message(message) + assert await listener_obj.handler.get_latest_message() == message @pytest.mark.asyncio async def test_run_bot(caplog): @@ -94,6 +103,19 @@ async def test_run_bot(caplog): await task +@pytest.mark.asyncio +async def test_listener_telegram(): + listener_test = Listener() + print(listener_test) + assert listener_test is not None + assert isinstance(listener_test, Listener) + await listener_test.start() + await listener_test.handler.handle_message("hello") + msg = await listener_test.handler.get_latest_message() + print(msg) + assert msg == "hello" + + @pytest.mark.asyncio async def test_baseplugins(): plugin = BasePlugin @@ -102,3 +124,4 @@ async def test_baseplugins(): assert callable(plugin.send_notification) assert callable(plugin.should_handle) assert callable(plugin.handle_message) + From 6bc9006cd32862fb1904cb795b7ee9121012a75c Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 14:34:37 +0200 Subject: [PATCH 20/24] =?UTF-8?q?=F0=9F=9A=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 5f9a81129..606ffa667 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -20,7 +20,7 @@ def set_test_settings(): @pytest.fixture(name="message") -def message(): +def message_test(): return "Test message" @pytest.fixture(name="listener_obj") From 90ab01ef87419cf0719d4894f3d07dd40232c71e Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 15:14:57 +0200 Subject: [PATCH 21/24] =?UTF-8?q?=E2=9C=85=20added=20listener=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 17 +++++++++++------ tt/talky_settings.toml | 8 ++++---- 2 files changed, 15 insertions(+), 10 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 606ffa667..8ee91f1b1 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -87,11 +87,6 @@ async def test_start_bot(listener_obj, plugin_manager_obj): start.assert_awaited await task -@pytest.mark.asyncio -async def test_get_latest_message(listener_obj, message): - await listener_obj.start() - await listener_obj.handler.handle_message(message) - assert await listener_obj.handler.get_latest_message() == message @pytest.mark.asyncio async def test_run_bot(caplog): @@ -104,7 +99,17 @@ async def test_run_bot(caplog): @pytest.mark.asyncio -async def test_listener_telegram(): +async def test_get_latest_message(message): + listener = Listener() + assert listener is not None + assert settings.VALUE == "On Testing" + await listener.start() + await listener.handler.handle_message(message) + assert await listener.handler.get_latest_message() == message + + +@pytest.mark.asyncio +async def test_listener_handler(): listener_test = Listener() print(listener_test) assert listener_test is not None diff --git a/tt/talky_settings.toml b/tt/talky_settings.toml index ef43093c0..e82d9f22b 100644 --- a/tt/talky_settings.toml +++ b/tt/talky_settings.toml @@ -244,10 +244,10 @@ apprise_config = "" apprise_url = "json://localhost" apprise_format = "NotifyFormat.MARKDOWN" iamlistening_enabled = true -bot_token = "" -bot_channel_id = "" -telethon_api_id = "" -telethon_api_hash = "" +bot_token = "1234556" +bot_channel_id = "1234556" +telethon_api_id = "1234556" +telethon_api_hash = "1234556" matrix_hostname = "" matrix_user = "" matrix_pass = "" From e8e4b8f90382e7f133bc413abf511d057d617317 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 15:25:05 +0200 Subject: [PATCH 22/24] =?UTF-8?q?=F0=9F=90=9B=20=E2=9C=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index 8ee91f1b1..af4364158 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -78,6 +78,7 @@ async def test_start_plugins(): await start_plugins(plugin_manager) plugin_manager.load_plugins.assert_called_once() + @pytest.mark.asyncio async def test_start_bot(listener_obj, plugin_manager_obj): start = AsyncMock() @@ -101,9 +102,9 @@ async def test_run_bot(caplog): @pytest.mark.asyncio async def test_get_latest_message(message): listener = Listener() + await listener.start() assert listener is not None assert settings.VALUE == "On Testing" - await listener.start() await listener.handler.handle_message(message) assert await listener.handler.get_latest_message() == message @@ -129,4 +130,3 @@ async def test_baseplugins(): assert callable(plugin.send_notification) assert callable(plugin.should_handle) assert callable(plugin.handle_message) - From a94a5414c10e6182e63dc3820473e09ee4062358 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 15:48:16 +0200 Subject: [PATCH 23/24] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20refactor=20to=20simp?= =?UTF-8?q?lify=20utils.py=20and=20apprise?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tt/plugins/plugin_manager.py | 2 +- tt/talky_settings.toml | 4 ---- tt/utils.py | 45 ++++++++++++++---------------------- 3 files changed, 18 insertions(+), 33 deletions(-) diff --git a/tt/plugins/plugin_manager.py b/tt/plugins/plugin_manager.py index 36e1d7048..850e05a25 100644 --- a/tt/plugins/plugin_manager.py +++ b/tt/plugins/plugin_manager.py @@ -35,7 +35,7 @@ def load_plugin(self, module, plugin_name): and obj is not BasePlugin): plugin_instance = obj() self.plugins.append(plugin_instance) - logger.info("Plugin loaded: {}", name) + logger.debug("Plugin loaded: {}", name) async def start_all_plugins(self): """ Start all plugins """ diff --git a/tt/talky_settings.toml b/tt/talky_settings.toml index e82d9f22b..9d50eb76e 100644 --- a/tt/talky_settings.toml +++ b/tt/talky_settings.toml @@ -47,8 +47,6 @@ webhook_secret = "123abc" ### APPRISE SETTINGS ### ######################################## -apprise_api_endpoint = "" -apprise_config = "" apprise_url = "json://localhost" apprise_format = "NotifyFormat.MARKDOWN" # TT_APPRISE_URL='tgram://TOKEN/CHAT_ID' @@ -239,8 +237,6 @@ loglevel = "DEBUG" host = "0.0.0.0" port = 8080 webhook_secret = "123abc" -apprise_api_endpoint = "" -apprise_config = "" apprise_url = "json://localhost" apprise_format = "NotifyFormat.MARKDOWN" iamlistening_enabled = true diff --git a/tt/utils.py b/tt/utils.py index 7767041cd..c71b9ed58 100644 --- a/tt/utils.py +++ b/tt/utils.py @@ -9,7 +9,7 @@ from apprise import Apprise, NotifyFormat from iamlistening import Listener -from tt.config import logger, settings +from tt.config import settings from tt.plugins.plugin_manager import PluginManager @@ -17,22 +17,25 @@ async def send_notification(msg): """ 💬 Notification via Apprise """ - aobj = Apprise() - if settings.apprise_api_endpoint: - aobj.add(settings.apprise_api_endpoint) - elif settings.apprise_config: - aobj.add(settings.apprise_config) - elif settings.apprise_url: - aobj.add(settings.apprise_url) + aobj = Apprise(settings.apprise_url) msg_format = settings.apprise_format or NotifyFormat.MARKDOWN await aobj.async_notify( body=msg, body_format=msg_format) +async def run_bot(max_iterations=None): + """ + 🤖 Run the chat bot & the plugins. + """ + listener = Listener() + plugin_manager = PluginManager() + await asyncio.gather(start_bot(listener, plugin_manager)) + + async def start_plugins(plugin_manager): """ - Start all plugins. + 🔌 Start all plugins. """ if settings.plugin_enabled: plugin_manager.load_plugins() @@ -42,27 +45,13 @@ async def start_plugins(plugin_manager): async def start_bot(listener, plugin_manager): """ - Listen to the message in the bot channel - and dispatch to plugins + 👂 Listen to the bot and dispatch to plugins """ await listener.start() await start_plugins(plugin_manager) while True: - try: - msg = await listener.handler.get_latest_message() - if msg and settings.plugin_enabled: - logger.debug("👂 listener: {}", msg) - await plugin_manager.process_message(msg) - except Exception as error: - logger.error("👂 listener: {}", error) + msg = await listener.handler.get_latest_message() + if msg and settings.plugin_enabled: + await plugin_manager.process_message(msg) - await asyncio.sleep(1) - - -async def run_bot(max_iterations=None): - """ - Run the chat bot & the plugins. - """ - listener = Listener() - plugin_manager = PluginManager() - await asyncio.gather(start_bot(listener, plugin_manager)) \ No newline at end of file + await asyncio.sleep(1) From 8f4aed764235e642b130966d6b85c39c0709f497 Mon Sep 17 00:00:00 2001 From: mraniki <8766259+mraniki@users.noreply.github.com> Date: Tue, 25 Jul 2023 16:03:08 +0200 Subject: [PATCH 24/24] =?UTF-8?q?=E2=9C=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_unit.py | 42 +++++++++++++++++++++--------------------- 1 file changed, 21 insertions(+), 21 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index af4364158..86adda9ce 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -99,27 +99,27 @@ async def test_run_bot(caplog): await task -@pytest.mark.asyncio -async def test_get_latest_message(message): - listener = Listener() - await listener.start() - assert listener is not None - assert settings.VALUE == "On Testing" - await listener.handler.handle_message(message) - assert await listener.handler.get_latest_message() == message - - -@pytest.mark.asyncio -async def test_listener_handler(): - listener_test = Listener() - print(listener_test) - assert listener_test is not None - assert isinstance(listener_test, Listener) - await listener_test.start() - await listener_test.handler.handle_message("hello") - msg = await listener_test.handler.get_latest_message() - print(msg) - assert msg == "hello" +# @pytest.mark.asyncio +# async def test_get_latest_message(message): +# listener = Listener() +# await listener.start() +# assert listener is not None +# assert settings.VALUE == "On Testing" +# await listener.handler.handle_message(message) +# assert await listener.handler.get_latest_message() == message + + +# @pytest.mark.asyncio +# async def test_listener_handler(): +# listener_test = Listener() +# print(listener_test) +# assert listener_test is not None +# assert isinstance(listener_test, Listener) +# await listener_test.start() +# await listener_test.handler.handle_message("hello") +# msg = await listener_test.handler.get_latest_message() +# print(msg) +# assert msg == "hello" @pytest.mark.asyncio