cli2telegram 2.0.0__tar.gz → 2.0.2__tar.gz

{cli2telegram-2.0.0 → cli2telegram-2.0.2}/PKG-INFO

@@ -1,23 +1,29 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: cli2telegram
-Version: 2.0.0
+Version: 2.0.2
 Summary: Small utility to send Telegram messages from the CLI.
-Home-page: https://github.com/markusressel/cli2telegram
+License: AGPL-3.0-or-later
+License-File: LICENSE
+Keywords: cli,telegram,notification
 Author: Markus Ressel
 Author-email: mail@markusressel.de
-License: AGPLv3+
-Platform: UNKNOWN
+Requires-Python: >=3.12,<4.0
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
-Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
+Classifier: Programming Language :: Python :: 3 :: Only
+Requires-Dist: aiofiles
+Requires-Dist: click
+Requires-Dist: container-app-conf
+Requires-Dist: emoji
+Requires-Dist: python-telegram-bot
+Project-URL: Homepage, https://github.com/markusressel/cli2telegram
+Project-URL: Repository, https://github.com/markusressel/cli2telegram
 Description-Content-Type: text/markdown
-License-File: LICENSE
 
 # cli2telegram
 Small utility to send Telegram messages from the CLI.
@@ -167,4 +173,3 @@ You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.
 ```
 
-

{cli2telegram-2.0.0 → cli2telegram-2.0.2}/cli2telegram/cli.py

@@ -13,17 +13,19 @@
 #
 #  You should have received a copy of the GNU Affero General Public License
 #  along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
+import asyncio
 import logging
 import sys
 from typing import Tuple, List
 
 import click
+from telegram.ext import Application
 
+from cli2telegram import util
 from cli2telegram.config import Config
 from cli2telegram.const import CODEBLOCK_MARKER_END, CODEBLOCK_MARKER_START, TELEGRAM_MESSAGE_LENGTH_LIMIT
 from cli2telegram.daemon import Daemon
-from cli2telegram.util import split_message, prepare_code_message, _try_send_message
+from cli2telegram.util import split_message, prepare_code_message
 
 LOGGER = logging.getLogger(__name__)
 # LOGGER.setLevel(logging.DEBUG)
@@ -86,15 +88,9 @@ def cli(bot_token: str or None, chat_id: str or None, code_block: bool, lines: T
     text = "".join(lines)
     messages = prepare_messages(text, code_block)
 
-    for message in messages:
-        _try_send_message(
-            bot_token=CONFIG.TELEGRAM_BOT_TOKEN.value,
-            chat_id=CONFIG.TELEGRAM_CHAT_ID.value,
-            message=message,
-            retry=CONFIG.RETRY_ENABLED.value,
-            retry_timeout=CONFIG.RETRY_TIMEOUT.value,
-            give_up_after=CONFIG.RETRY_GIVE_UP_AFTER.value
-        )
+    app = Application.builder().token(CONFIG.TELEGRAM_BOT_TOKEN.value).build()
+
+    asyncio.run(_send_messages(app, messages))
 
 
 def prepare_messages(text: str, code_block: bool) -> List[str]:
@@ -114,5 +110,17 @@ def prepare_messages(text: str, code_block: bool) -> List[str]:
     return result
 
 
+async def _send_messages(app, messages):
+    for message in messages:
+        await util.try_send_message(
+            app=app,
+            chat_id=CONFIG.TELEGRAM_CHAT_ID.value,
+            message=message,
+            retry=CONFIG.RETRY_ENABLED.value,
+            retry_timeout=CONFIG.RETRY_TIMEOUT.value,
+            give_up_after=CONFIG.RETRY_GIVE_UP_AFTER.value
+        )
+
+
 if __name__ == '__main__':
     cli()

{cli2telegram-2.0.0 → cli2telegram-2.0.2}/cli2telegram/daemon.py

@@ -13,20 +13,19 @@
 #
 #  You should have received a copy of the GNU Affero General Public License
 #  along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
+import asyncio
 import logging
 import os
-import queue
 import sys
-import threading
 from signal import signal, SIGINT, SIGTERM
 from time import time, sleep
 
-from telegram.ext import Updater
+import aiofiles
+from telegram.ext import ApplicationBuilder
 
 from cli2telegram import RetryLimitReachedException
 from cli2telegram.config import Config
-from cli2telegram.util import _try_send_message, prepare_code_message
+from cli2telegram.util import try_send_message, prepare_code_message
 
 LOGGER = logging.getLogger(__name__)
 HANDLER = logging.StreamHandler(sys.stdout)
@@ -40,8 +39,8 @@ class Daemon:
         self.config = config
         self.running = True
         self.pipe_file_path = config.DAEMON_PIPE_PATH.value
-        self.message_queue = queue.Queue()
-        self.updater = Updater(token=config.TELEGRAM_BOT_TOKEN.value, use_context=True)
+        self.message_queue = asyncio.Queue()
+        self._app = ApplicationBuilder().token(config.TELEGRAM_BOT_TOKEN.value).build()
 
     def run(self):
         if os.path.exists(self.pipe_file_path):
@@ -55,22 +54,27 @@ class Daemon:
         signal(SIGINT, self.signal_handler)  # ctrl+c
         signal(SIGTERM, self.signal_handler)  # systemctl stop
 
-        threading.Thread(target=self.send_message_worker, daemon=True).start()
+        asyncio.run(self._run_workers())
 
-        LOGGER.info(f"Daemon is listening for input on {self.pipe_file_path}...")
-        self.read_pipe_loop(self.pipe_file_path)
+    async def _run_workers(self):
+        await asyncio.gather(
+            self.send_message_worker(),
+            self.read_pipe_loop(),
+        )
 
-    def read_pipe_loop(self, path: str):
+    async def read_pipe_loop(self):
+        LOGGER.info(f"Daemon is listening for input on {self.pipe_file_path}...")
         while True:
             LOGGER.debug("Waiting for new input on pipe")
-            with open(path) as pipe:
+
+            async with aiofiles.open(self.pipe_file_path, 'r') as pipe:
                 pipe_buffer = ''
                 while True:
-                    data = pipe.read()
+                    data = await pipe.read()
                     LOGGER.debug(f"received: {data}")
                     if len(data) == 0 and self.running:
                         LOGGER.debug("No new data from pipe, adding current buffer to message queue")
-                        self.message_queue.put(pipe_buffer)
+                        await self.message_queue.put(pipe_buffer)
                         break
                     elif not self.running:
                         break
@@ -80,21 +84,23 @@ class Daemon:
             if not self.running:
                 break
 
-    def send_message_worker(self):
+    async def send_message_worker(self):
         while True:
-            message = self.message_queue.get()
+            LOGGER.debug("Waiting for item in message queue...")
+            message = await self.message_queue.get()
+            LOGGER.debug(f"Got new message: {message}")
 
             if not message.strip():
                 LOGGER.warning("Message is empty, ignoring")
                 self.message_queue.task_done()
                 continue
 
-            prepared_message = prepare_code_message(message.splitlines())
+            prepared_message = prepare_code_message(message)
             LOGGER.debug(f"Trying to send message from queue...")
 
             try:
-                _try_send_message(
-                    bot_token=self.config.TELEGRAM_BOT_TOKEN.value,
+                await try_send_message(
+                    app=self._app,
                     chat_id=self.config.TELEGRAM_CHAT_ID.value,
                     message=prepared_message,
                     retry=self.config.RETRY_ENABLED.value,
@@ -120,7 +126,8 @@ class Daemon:
                 LOGGER.warning("Unable to clean up named pipe")
 
         stop = time() + 5
-        while self.message_queue.unfinished_tasks and time() < stop:
+
+        while self.message_queue._unfinished_tasks and time() < stop:
             sleep(1)
 
         sys.exit(0)

{cli2telegram-2.0.0 → cli2telegram-2.0.2}/cli2telegram/util.py

@@ -18,7 +18,7 @@ import time
 from datetime import datetime, timedelta
 
 from telegram import Bot, InlineKeyboardMarkup, Message
-from telegram.ext import Updater
+from telegram.ext import Application
 
 from cli2telegram import RetryLimitReachedException
 from cli2telegram.const import CODEBLOCK_MARKER_START, CODEBLOCK_MARKER_END
@@ -26,7 +26,7 @@ from cli2telegram.const import CODEBLOCK_MARKER_START, CODEBLOCK_MARKER_END
 LOGGER = logging.getLogger(__name__)
 
 
-def send_message(
+async def send_message(
         bot: Bot, chat_id: str, message: str, parse_mode: str = None, reply_to: int = None,
         menu: InlineKeyboardMarkup = None
 ) -> Message:
@@ -40,10 +40,10 @@ def send_message(
     :param menu: inline keyboard menu markup
     """
     from emoji import emojize
-    emojized_text = emojize(message, use_aliases=True)
-    return bot.send_message(
+    emojized_text = emojize(message, language='alias')
+    return await bot.send_message(
         chat_id=chat_id, parse_mode=parse_mode, text=emojized_text, reply_to_message_id=reply_to, reply_markup=menu,
-        timeout=10
+        connect_timeout=10, read_timeout=10, write_timeout=10
     )
 
 
@@ -73,14 +73,14 @@ def prepare_code_message(message: str) -> str:
     return result
 
 
-def _try_send_message(
-        bot_token: str,
+async def try_send_message(
+        app: Application,
         chat_id: str, message: str,
         retry: bool, retry_timeout: timedelta, give_up_after: timedelta
 ):
     """
     Sends a message
-    :param bot_token: telegram bot token
+    :param app: telegram application
     :param chat_id: chat id
     :param message: the message to send
     :param retry: whether to retry if something fails
@@ -91,8 +91,7 @@ def _try_send_message(
     success = False
     while not success:
         try:
-            updater = Updater(token=bot_token, use_context=True)
-            send_message(bot=updater.bot, chat_id=chat_id, message=message, parse_mode="markdown")
+            await send_message(bot=app.bot, chat_id=chat_id, message=message, parse_mode="markdown")
             success = True
         except Exception as ex:
             LOGGER.exception(ex)

cli2telegram-2.0.2/pyproject.toml

@@ -0,0 +1,51 @@
+[tool.poetry]
+name = "cli2telegram"
+version = "2.0.2"
+description = "Small utility to send Telegram messages from the CLI."
+
+license = "AGPL-3.0-or-later"
+
+authors = [
+    "Markus Ressel <mail@markusressel.de>",
+]
+
+readme = 'README.md'
+
+repository = "https://github.com/markusressel/cli2telegram"
+homepage = "https://github.com/markusressel/cli2telegram"
+
+keywords = ['cli', 'telegram', 'notification']
+
+classifiers = [
+    "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Development Status :: 5 - Production/Stable"
+]
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.dependencies]
+python = "^3.12"  # Compatible python versions must be declared here
+
+container-app-conf = "*"
+click = "*"
+emoji = "*"
+python-telegram-bot = "*"
+aiofiles = "*"
+
+[tool.poetry.dev-dependencies]
+pytest = "*"
+
+[tool.pytest.ini_options]
+testpaths = [
+    "tests",
+]
+
+[tool.poetry.scripts]
+cli2telegram = 'cli2telegram.cli:cli'

cli2telegram-2.0.0/cli2telegram.egg-info/PKG-INFO

@@ -1,170 +0,0 @@
-Metadata-Version: 2.1
-Name: cli2telegram
-Version: 2.0.0
-Summary: Small utility to send Telegram messages from the CLI.
-Home-page: https://github.com/markusressel/cli2telegram
-Author: Markus Ressel
-Author-email: mail@markusressel.de
-License: AGPLv3+
-Platform: UNKNOWN
-Classifier: Development Status :: 5 - Production/Stable
-Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-
-# cli2telegram
-Small utility to send Telegram messages from the CLI.
-
-This can f.ex. be used to
-* quickly send arbitrary messages to a Telegram chat of your choice
-* use it as a replacement for the "mail" program on linux
-* use Telegram as a notification backend for the ZFS Event Daemon (zed)
-
-## Features
-* [x] Read messages from argument or STDIN
-* [x] (Optional) Configuration file
-* [x] Retry sending messages for a specified amount of time
-* [x] Handling of long messages
-* [x] Run as a daemon and echo messages into a linux pipe
-
-## Examples
-
-```shell
-cli2telegram -h
-
-# From arguments
-cli2telegram "This is a message"
-
-cli2telegram "Header" "This is a multiline message."
-
-# From STDIN
-echo My Message | cli2telegram
-
-printf "Header\nThis is a multiline message." | cli2telegram
-
-# Config via parameters
-printf "Message" | cli2telegram -b "123456789:ABCDEFG" -c "123456789"
-
-# Config via parameters and send message in a code block
-printf "Message" | cli2telegram -b "123456789:ABCDEFG" -c "123456789" -C
-
-# as a Daemon
-cli2telegram -d -p "/tmp/cli2telegram"
-echo "hello world!" > /tmp/cli2telegram
-``` 
-
-# Install
-
-To use this utility install it either using:
-```
-pip install cli2telegram
-```
-
-or - if you don't want to install it globally - using f.ex. [venv-install](https://github.com/markusressel/venv-install):
-```
-venv-install cli2telegram cli2telegram
-```
-
-or your custom `venv` manager of choice.
-
-# Configuration
-
-To be able to send you messages you have to provide a **bot token** and a **chat id**.
-You can configure cli2telegram using cli parameters, a configuration file,
-environment variables, or a combination of them.
-
-## Parameters
-
-| Name                 | Type   | Description                               |
-|----------------------|--------|-------------------------------------------|
-| `-b`, `--bot-token`  | String | Telegram Bot Token                        |
-| `-c`, `--chat-id`    | String | Telegram Chat ID                          |
-| `-C`, `--code-block` | String | Send message inside a code block          |
-| `-d`, `--daemon`     | Flag   | Run as a daemon                           |
-| `-p`, `--pipe`       | String | File path to the pipe used in daemon mode |
-
-## File
-cli2telegram uses [container-app-conf](https://github.com/markusressel/container-app-conf) so you can use YAML, TOML, or ENV to set those.
-
-Have a look at the [cli2telegram.toml_example](cli2telegram.toml_example) file to get an idea.
-
-# Daemon
-
-When running cli2telegram as a daemon, the pipe will close for a brief amount of time between receiving input messages.
-If you are sending multiple messages to the pipe using f.ex. a script, make sure to wait a bit (f.ex. 0.5 seconds)
-between sending messages, otherwise:
-* multiple messages may be received as one
-* messages may get lost
-* you may receive a `Broken pipe` error
-
-# Use Cases
-
-## ZFS Event Daemon (ZED)
-To make `zed` call cli2telegram we will trick it and make it use cli2telegram as an E-Mail client.
-
-Edit `/etc/zfs/zed.d/zed.rc` as root:
-```bash
-sudo nano -w /etc/zfs/zed.d/zed.rc
-```
-
-and
-* uncomment `ZED_EMAIL_ADDR`, the value does not matter since we use our own email script, but **it is necessary to set a value to make ZED send 'emails'**
-* set `ZED_EMAIL_PROG` to the path of the script, f.ex. `/usr/bin/cli2telegram`
-  * it is important to note that zed does not seem to work if your command needs arguments to run
-```
-# this must not be empty!
-ZED_EMAIL_ADDR="root"
-
-[...]
-
-ZED_EMAIL_PROG="/usr/bin/cli2telegram"
-
-[...]
-
-# this must not be empty!
-ZED_EMAIL_OPTS="#zfs #$(hostname)"
-
-[...]
-
-# If you want to receive email no matter the state of your pool, you’ll want to set:
-ZED_NOTIFY_VERBOSE=1
-
-[...]
-```
-
-Since `zed` will run your scripts as root, if you want to use a config file 
-you have to put it in f.ex. `/root/.config/cli2telegram.toml`.
-
-# Contributing
-
-GitHub is for social coding: if you want to write code, I encourage contributions through pull requests from forks
-of this repository. Create GitHub tickets for bugs and new features and comment on the ones that you are interested in.
-
-# License
-
-```text
-cli2telegram by Markus Ressel
-Copyright (C) 2018  Markus Ressel
-
-This program is free software: you can redistribute it and/or modify
-it under the terms of the GNU Affero General Public License as
-published by the Free Software Foundation, either version 3 of the
-License, or (at your option) any later version.
-
-This program is distributed in the hope that it will be useful,
-but WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-GNU Affero General Public License for more details.
-
-You should have received a copy of the GNU Affero General Public License
-along with this program.  If not, see <https://www.gnu.org/licenses/>.
-```
-
-

cli2telegram-2.0.0/cli2telegram.egg-info/SOURCES.txt

@@ -1,17 +0,0 @@
-LICENSE
-README.md
-setup.py
-cli2telegram/__init__.py
-cli2telegram/cli.py
-cli2telegram/config.py
-cli2telegram/const.py
-cli2telegram/daemon.py
-cli2telegram/util.py
-cli2telegram.egg-info/PKG-INFO
-cli2telegram.egg-info/SOURCES.txt
-cli2telegram.egg-info/dependency_links.txt
-cli2telegram.egg-info/entry_points.txt
-cli2telegram.egg-info/requires.txt
-cli2telegram.egg-info/top_level.txt
-tests/__init__.py
-tests/test_message_processing.py

cli2telegram-2.0.0/cli2telegram.egg-info/dependency_links.txt

@@ -1 +0,0 @@
-

cli2telegram-2.0.0/cli2telegram.egg-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-cli2telegram = cli2telegram.cli:cli
-

cli2telegram-2.0.0/cli2telegram.egg-info/requires.txt

@@ -1,4 +0,0 @@
-container-app-conf==5.2.2
-click
-emoji
-python-telegram-bot>=13.10.0

cli2telegram-2.0.0/cli2telegram.egg-info/top_level.txt

@@ -1,2 +0,0 @@
-cli2telegram
-tests

cli2telegram-2.0.0/setup.cfg

@@ -1,4 +0,0 @@
-[egg_info]
-tag_build = 
-tag_date = 0
-

cli2telegram-2.0.0/setup.py

@@ -1,89 +0,0 @@
-#  Copyright (c) 2019 Markus Ressel
-#
-#  Permission is hereby granted, free of charge, to any person obtaining a copy
-#  of this software and associated documentation files (the "Software"), to deal
-#  in the Software without restriction, including without limitation the rights
-#  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-#  copies of the Software, and to permit persons to whom the Software is
-#  furnished to do so, subject to the following conditions:
-#
-#  The above copyright notice and this permission notice shall be included in all
-#  copies or substantial portions of the Software.
-#
-#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-#  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-#  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-#  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-#  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-#  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-#  SOFTWARE.
-#
-
-from setuptools import setup, find_packages
-
-VERSION_NUMBER = "2.0.0"
-
-DEVELOPMENT_STATUS = "Development Status :: 5 - Production/Stable"
-VERSION_NAME = VERSION_NUMBER
-
-
-def readme_type() -> str:
-    import os
-    if os.path.exists("README.rst"):
-        return "text/x-rst"
-    if os.path.exists("README.md"):
-        return "text/markdown"
-
-
-def readme() -> [str]:
-    if readme_type() == "text/markdown":
-        file_name = "README.md"
-    else:
-        file_name = "README.rst"
-
-    with open(file_name) as f:
-        return f.read()
-
-
-def install_requirements() -> [str]:
-    return read_requirements_file("requirements.txt")
-
-
-def test_requirements() -> [str]:
-    return read_requirements_file("test_requirements.txt")
-
-
-def read_requirements_file(file_name: str):
-    with open(file_name, encoding='utf-8') as f:
-        requirements_file = f.readlines()
-    return [r.strip() for r in requirements_file]
-
-
-setup(
-    name='cli2telegram',
-    version=VERSION_NAME,
-    description='Small utility to send Telegram messages from the CLI.',
-    long_description=readme(),
-    long_description_content_type=readme_type(),
-    license='AGPLv3+',
-    author='Markus Ressel',
-    author_email='mail@markusressel.de',
-    url='https://github.com/markusressel/cli2telegram',
-    packages=find_packages(),
-    classifiers=[
-        DEVELOPMENT_STATUS,
-        'License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)',
-        'Programming Language :: Python :: 3 :: Only',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.6',
-        'Programming Language :: Python :: 3.7',
-        'Programming Language :: Python :: 3.8',
-        'Programming Language :: Python :: 3.9',
-        'Programming Language :: Python :: 3.10',
-    ],
-    install_requires=install_requirements(),
-    tests_require=test_requirements(),
-    entry_points={
-        'console_scripts': ['cli2telegram = cli2telegram.cli:cli']
-    }
-)

cli2telegram-2.0.0/tests/__init__.py

@@ -1,22 +0,0 @@
-#   cli2telegram
-#  Copyright (c) 2019.  Markus Ressel
-#
-#  This program is free software: you can redistribute it and/or modify
-#  it under the terms of the GNU Affero General Public License as
-#  published by the Free Software Foundation, either version 3 of the
-#  License, or (at your option) any later version.
-#
-#  This program is distributed in the hope that it will be useful,
-#  but WITHOUT ANY WARRANTY; without even the implied warranty of
-#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-#  GNU Affero General Public License for more details.
-#
-#  You should have received a copy of the GNU Affero General Public License
-#  along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
-import unittest
-
-
-class TestBase(unittest.TestCase):
-    if __name__ == '__main__':
-        unittest.main()

cli2telegram-2.0.0/tests/test_message_processing.py

@@ -1,97 +0,0 @@
-#   cli2telegram
-#  Copyright (c) 2019.  Markus Ressel
-#
-#  This program is free software: you can redistribute it and/or modify
-#  it under the terms of the GNU Affero General Public License as
-#  published by the Free Software Foundation, either version 3 of the
-#  License, or (at your option) any later version.
-#
-#  This program is distributed in the hope that it will be useful,
-#  but WITHOUT ANY WARRANTY; without even the implied warranty of
-#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-#  GNU Affero General Public License for more details.
-#
-#  You should have received a copy of the GNU Affero General Public License
-#  along with this program.  If not, see <https://www.gnu.org/licenses/>.
-from cli2telegram.cli import prepare_messages
-from cli2telegram.util import prepare_code_message, split_message
-from tests import TestBase
-
-
-class MessageProcessingTest(TestBase):
-
-    def test_long_message_with_codeblock_doesnt_exceed_limit(self):
-        std_in = "A" * 4096 + "B" * 4096
-        messages = prepare_messages(std_in, True)
-
-        assert len(messages) == 3
-        for m in messages:
-            assert len(m) <= 4096
-        assert "B" not in messages[0]
-        assert "A" in messages[1]
-        assert "B" in messages[1]
-        assert "A" not in messages[2]
-
-    def test_long_message_doesnt_exceed_limit(self):
-        std_in = "A" * 4096 + "B" * 4096
-        messages = prepare_messages(std_in, False)
-
-        assert len(messages) == 2
-        for i, m in enumerate(messages):
-            if i < len(messages) - 1:
-                assert len(m) == 4096
-            else:
-                assert len(m) <= 4096
-        assert "B" not in messages[0]
-        assert "A" not in messages[1]
-
-    def test_long_message_is_split(self):
-        length = 32
-        std_in = "A" * length + "B" * length
-        messages = split_message(std_in, length)
-
-        assert len(messages) == 2
-        for i, m in enumerate(messages):
-            if i < len(messages) - 1:
-                assert len(m) == length
-            else:
-                assert len(m) <= length
-        assert "B" not in messages[0]
-        assert "A" not in messages[1]
-
-    def test_prepare_code_message(self):
-        message = "Code"
-        output = prepare_code_message(message)
-
-        assert output.startswith("```\n")
-        assert output.endswith("\n```")
-        assert message[0] in output
-
-    def test_newlines_are_retained_in_code_block(self):
-        std_in = "\nCo\nde\n"
-        output = prepare_code_message(std_in)
-
-        assert output == "```\n\nCo\nde\n\n```"
-
-    def test_message_preparation(self):
-        std_in = """ZFS has finished a scrub:\n
-                  \n
-                     eid: 3337\n
-                   class: scrub_finish\n
-                    host: \n
-                    time: 2019-12-01 19:36:01+0100\n
-                    pool: rpool\n
-                   state: ONLINE\n
-                    scan: scrub repaired 0B in 0 days 00:00:24 with 0 errors on Sun Dec  1 19:36:01 2019\n
-                  config:\n
-                  \n
-                  \tNAME                                                     STATE     READ WRITE CKSUM\n
-                  \trpool                                                    ONLINE       0     0     0\n
-                  \t  mirror-0                                               ONLINE       0     0     0\n
-                  \t    ata-Samsung_SSD_860_EVO_500GB_S4XBNF1M807031E-part2  ONLINE       0     0     0\n
-                  \t    ata-Samsung_SSD_860_EVO_500GB_S4XBNF1M807033B-part2  ONLINE       0     0     0\n
-                  \n
-                  errors: No known data errors\n"""
-
-        output = prepare_code_message(std_in)
-        assert output