chgksuite 0.24.0b3__py3-none-any.whl → 0.24.2__py3-none-any.whl

chgksuite/composer/telegram_bot.py

@@ -0,0 +1,115 @@
+import asyncio
+import json
+import os
+import sqlite3
+import threading
+from datetime import datetime
+
+import requests
+import toml
+from telegram import Update
+from telegram.ext import Application, ContextTypes, MessageHandler, filters
+
+from chgksuite.common import get_chgksuite_dir
+
+
+class TelegramSidecarBot:
+    def __init__(self, bot_token, db_path):
+        self.db_path = db_path
+        self._local = threading.local()
+        self.token = bot_token
+
+    @property
+    def conn(self):
+        if not hasattr(self._local, "connection"):
+            self._local.connection = sqlite3.connect(self.db_path)
+        return self._local.connection
+
+    async def handle_message(self, update: Update, _: ContextTypes.DEFAULT_TYPE):
+        cursor = self.conn.cursor()
+        raw_data = json.dumps(update.to_dict(), ensure_ascii=False)
+        cursor.execute(
+            "INSERT INTO messages (raw_data, chat_id, created_at) VALUES (?, ?, ?)",
+            (raw_data, update.message.chat.id, datetime.now().isoformat()),
+        )
+        self.conn.commit()
+
+    async def error_handler(self, update, context):
+        print(f"Update {update} caused error: {context.error}")
+
+    async def check_connectivity(self):
+        url = f"https://api.telegram.org/bot{self.token}/getMe"
+        req = requests.get(url)
+        cursor = self.conn.cursor()
+        if req.status_code == 200 and "ok" in req.json():
+            cursor.execute(
+                "INSERT INTO bot_status (raw_data, created_at) VALUES (?, ?)",
+                (json.dumps({"status": "ok"}), datetime.now().isoformat()),
+            )
+            self.conn.commit()
+        else:
+            print(f"couldn't check status, req: {req.text}")
+            cursor.execute(
+                "INSERT INTO bot_status (raw_data, created_at) VALUES (?, ?)",
+                (
+                    json.dumps(
+                        {
+                            "status": "bad",
+                            "error": req.text,
+                            "status_code": req.status_code,
+                        }
+                    ),
+                    datetime.now().isoformat(),
+                ),
+            )
+            self.conn.commit()
+        return True
+
+    def run(self):
+        loop = asyncio.get_event_loop()
+        application = (
+            Application.builder()
+            .token(self.token)
+            .connect_timeout(30.0)
+            .read_timeout(30.0)
+            .write_timeout(30.0)
+            .build()
+        )
+        application.add_handler(MessageHandler(filters.ALL, self.handle_message))
+        application.add_error_handler(self.error_handler)
+        loop.run_until_complete(application.initialize())
+        loop.run_until_complete(application.start())
+        loop.run_until_complete(
+            application.updater.start_polling(
+                allowed_updates=Update.ALL_TYPES, drop_pending_updates=True
+            )
+        )
+        loop.run_forever()
+
+
+def run_bot_in_thread(bot_token, db_path):
+    """Run the bot in a daemon thread."""
+
+    def thread_function():
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        bot = TelegramSidecarBot(bot_token, db_path)
+        connectivity_ok = loop.run_until_complete(bot.check_connectivity())
+        if not connectivity_ok:
+            raise Exception("bot couldn't connect")
+        bot.run()
+
+    bot_thread = threading.Thread(target=thread_function, daemon=True)
+    bot_thread.start()
+    return bot_thread
+
+
+def main():
+    toml_path = os.path.join(get_chgksuite_dir(), "telegram.toml")
+    with open(toml_path, "r", encoding="utf8") as f:
+        bot_token = toml.load(f)["bot_token"]
+    run_bot_in_thread(bot_token, "test_bot.db")
+
+
+if __name__ == "__main__":
+    main()

chgksuite/version.py

@@ -1 +1 @@
-__version__ = "0.24.0b3"
+__version__ = "0.24.2"

{chgksuite-0.24.0b3.dist-info → chgksuite-0.24.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chgksuite
-Version: 0.24.0b3
+Version: 0.24.2
 Summary: A package for chgk automation
 Home-page: https://gitlab.com/peczony/chgksuite
 Author: Alexander Pecheny
@@ -24,8 +24,7 @@ Requires-Dist: pyperclip
 Requires-Dist: python-docx
 Requires-Dist: python-pptx
 Requires-Dist: requests
-Requires-Dist: telethon
-Requires-Dist: TgCrypto
+Requires-Dist: python-telegram-bot
 Requires-Dist: toml
 Dynamic: author
 Dynamic: author-email

{chgksuite-0.24.0b3.dist-info → chgksuite-0.24.2.dist-info}/RECORD

@@ -1,25 +1,26 @@
 chgksuite/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chgksuite/__main__.py,sha256=0-_jfloveTW3SZYW5XEagbyaHKGCiDhGNgcLxsT_dMs,140
 chgksuite/cli.py,sha256=8YoHWfc-wv6gdYMPH3X8HxV9SG70S7M6IQqavJaBXV4,31990
-chgksuite/common.py,sha256=lHQG72HghaBudr1X6ognGbrKmStW5k9nGSLdoxydMSA,10834
+chgksuite/common.py,sha256=FkEntLcVTApjLtTFTDf2QiMOath05i322wcmJIFjztQ,11093
 chgksuite/parser.py,sha256=AVNeTUgv0aGHsxWunV_7bM4Ul2hbDslkzqg0dU49lic,37285
 chgksuite/parser_db.py,sha256=W1--OcDnx18mehH1T2ISHu3Saeq-9mqHo-xJopNySXI,11135
 chgksuite/trello.py,sha256=BG1Qb_W7Uu4o3Mfc_tK71ElU8ysdSplGlj_sAKfvUn4,14730
 chgksuite/typotools.py,sha256=Jdk65Wn_bXqpQtOT7PkBZyD2ZG1MBeeZFPMzcHEPkf4,12771
-chgksuite/version.py,sha256=_g51twxO95ihIudz5fXK0W3t40T0co_jzKU7dVW0AN0,25
+chgksuite/version.py,sha256=UBF3OYTcBAovta3ux5ybxb0MZYAGpGO79WH_ax2NGeI,23
 chgksuite/vulture_whitelist.py,sha256=P__p_X0zt10ivddIf81uyxsobV14vFg8uS2lt4foYpc,3582
 chgksuite/composer/__init__.py,sha256=MAOVZIYXmZmD6nNQSo9DueV6b5RgxF7_HGeLvsAhMJs,6490
-chgksuite/composer/chgksuite_parser.py,sha256=gOFAqPXRh-I5_GAPM6j1Jki-Wcdald5vPp2vFU1NwOM,8303
+chgksuite/composer/chgksuite_parser.py,sha256=MFcLUWbccMqo3OYEuaAIA0loEvWM_PNS9vR7c1z_c60,8843
 chgksuite/composer/composer_common.py,sha256=S5ipehxep6LlGIZ9dcBnifbVaMYXijMhq6-pUHRISo8,15309
 chgksuite/composer/db.py,sha256=71cINE_V8s6YidvqpmBmmlWbcXraUEGZA1xpVFAUENw,8173
 chgksuite/composer/docx.py,sha256=5MASXACM-ztWrr3VdO8HZ-W-hWWQ5TY1jXMsCQIufGc,18346
 chgksuite/composer/latex.py,sha256=WtLdUICxeX4_5vHEJRF0YhFLpTsOUwBkQFunQS488FA,9248
 chgksuite/composer/lj.py,sha256=nty3Zs3N1H0gNK378U04aAHo71_5cABhCM1Mm9jiUEA,15213
 chgksuite/composer/openquiz.py,sha256=CWGxb_cPG2Qh3dQ3hu0pS5fOfhKc8klZuO38Lo9hUF8,7020
-chgksuite/composer/pptx.py,sha256=05JxulgU80jBmxRqC3cFh0q_B3bMUCe31JzLYvnNZYM,19119
+chgksuite/composer/pptx.py,sha256=lmLld5CCNfndmLwWz2rRqZFyPIUxl0Dvk6wO1jW_c5s,19116
 chgksuite/composer/reddit.py,sha256=-Eg4CqMHhyGGfCteVwdQdtE1pfUXQ42XcP5OYUrBXmo,3878
 chgksuite/composer/stats.py,sha256=GbraSrjaZ8Mc2URs5aGAsI4ekboAKzlJJOqsbe96ELA,3995
-chgksuite/composer/telegram.py,sha256=Af0LJIjz3vO5OiQikV7XQObx4ABJewBp7YqHRt0P4GY,25670
+chgksuite/composer/telegram.py,sha256=H65LS79JZ3aCcqeq_SrnUU4E4yro1f-EZxCM3IrIyyQ,46526
+chgksuite/composer/telegram_bot.py,sha256=xT5D39m4zGmIbHV_ZfyQ9Rc8PAmG2V5FGUeDKpkgyTw,3767
 chgksuite/composer/telegram_parser.py,sha256=50WqOuvzzdMJJm5wsSLS49oURAQRYToPnbPJjQbMYC4,8096
 chgksuite/resources/cheader.tex,sha256=Jfe3LESk0VIV0HCObbajSQpEMljaIDAIEGSs6YY9rTk,3454
 chgksuite/resources/fix-unnumbered-sections.sty,sha256=FN6ZSWC6MvoRoThPm5AxCF98DdgcxbxyBYG6YImM05s,1409
@@ -42,10 +43,11 @@ chgksuite/resources/regexes_ua.json,sha256=IXtJtoUY15OoBu5Y5yW6adhJUhsDRkwfxWsPY
 chgksuite/resources/regexes_uz_cyr.json,sha256=mIvWjyZZXQ3lr_ntuC7-ybknnYJxSSQEKxqnGluTvow,1106
 chgksuite/resources/template.docx,sha256=Do29TAsg3YbH0rRSaXhVzKEoh4pwXkklW_idWA34HVE,11189
 chgksuite/resources/template.pptx,sha256=hEFWqE-yYpwZ8ejrMCJIPEyoMT3eDqaqtiEeQ7I4fyk,29777
+chgksuite/resources/template_shorin.pptx,sha256=hEFWqE-yYpwZ8ejrMCJIPEyoMT3eDqaqtiEeQ7I4fyk,29777
 chgksuite/resources/trello.json,sha256=M5Q9JR-AAJF1u16YtNAxDX-7c7VoVTXuq4POTqYvq8o,555
-chgksuite-0.24.0b3.dist-info/licenses/LICENSE,sha256=_a1yfntuPmctLsuiE_08xMSORuCfGS8X5hQph2U_PUw,1081
-chgksuite-0.24.0b3.dist-info/METADATA,sha256=onWj7gdTXF2zHFbAeD7CSg09Tx619sywCW33HtvNRFk,1309
-chgksuite-0.24.0b3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-chgksuite-0.24.0b3.dist-info/entry_points.txt,sha256=lqjX6ULQZGDt0rgouTXBuwEPiwKkDQkSiNsT877A_Jg,54
-chgksuite-0.24.0b3.dist-info/top_level.txt,sha256=cSWiRBOGZW9nIO6Rv1IrEfwPgV2ZWs87QV9wPXeBGqM,10
-chgksuite-0.24.0b3.dist-info/RECORD,,
+chgksuite-0.24.2.dist-info/licenses/LICENSE,sha256=_a1yfntuPmctLsuiE_08xMSORuCfGS8X5hQph2U_PUw,1081
+chgksuite-0.24.2.dist-info/METADATA,sha256=EfR9tuFPhFwdMjYtaExqh1P-CuoXBPSuU-_7avOAvYs,1294
+chgksuite-0.24.2.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+chgksuite-0.24.2.dist-info/entry_points.txt,sha256=lqjX6ULQZGDt0rgouTXBuwEPiwKkDQkSiNsT877A_Jg,54
+chgksuite-0.24.2.dist-info/top_level.txt,sha256=cSWiRBOGZW9nIO6Rv1IrEfwPgV2ZWs87QV9wPXeBGqM,10
+chgksuite-0.24.2.dist-info/RECORD,,

{chgksuite-0.24.0b3.dist-info → chgksuite-0.24.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (79.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any