telegram_libs 0.1.13__tar.gz → 0.1.15__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: telegram_libs
3
- Version: 0.1.13
3
+ Version: 0.1.15
4
4
  Summary: Common libraries for Telegram bots
5
5
  Author: Andrey Gritsaenko gricaenko.95a@gmail.com
6
6
  Requires-Python: >=3.11,<4.0
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "telegram_libs"
3
- version = "0.1.13"
3
+ version = "0.1.15"
4
4
  description = "Common libraries for Telegram bots"
5
5
  authors = ["Andrey Gritsaenko gricaenko.95a@gmail.com"]
6
6
  readme = "README.md"
@@ -2,4 +2,4 @@
2
2
  Telegram Libs - Common libraries for Telegram bots
3
3
  """
4
4
 
5
- __version__ = "0.1.13"
5
+ __version__ = "0.1.15"
@@ -5,6 +5,7 @@ required_constants = []
5
5
  BOTS_AMOUNT = os.getenv("BOTS_AMOUNT")
6
6
  MONGO_URI = os.getenv("MONGO_URI")
7
7
  SUBSCRIPTION_DB_NAME = os.getenv("SUBSCRIPTION_DB_NAME")
8
+ DEBUG = os.getenv("DEBUG", "False").lower() in ("true", "1", "yes")
8
9
 
9
10
  required_constants.append(("BOTS_AMOUNT", BOTS_AMOUNT))
10
11
  required_constants.append(("MONGO_URI", MONGO_URI))
@@ -9,11 +9,7 @@
9
9
  "info": "Buying a subscription you will get unlimited access to other {0} bots, to see all bots click /more"
10
10
  },
11
11
  "support": {
12
- "message": "If you have any questions or need help, please write it now and we will solve your issue as soon as possible.",
12
+ "message": "Write down any questions, issues or suggestions you have, and we will resolve them as soon as possible 👇 ",
13
13
  "response": "Thank you! Our support team will contact you soon."
14
- },
15
- "feedback": {
16
- "message": "We appreciate your feedback! Please send your suggestions or issues and we will review them as soon as possible.",
17
- "response": "Thank you for your feedback!"
18
14
  }
19
15
  }
@@ -9,11 +9,7 @@
9
9
  "info": "Купив подписку, вы получите неограниченный доступ к другим {0} ботам, чтобы увидеть всех ботов, нажмите /more"
10
10
  },
11
11
  "support": {
12
- "message": "Если у вас есть вопросы или нужна помощь, пожалуйста, напишите сейчас, и мы решим вашу проблему как можно скорее.",
12
+ "message": "Напишите любые вопросы, проблемы, или предложения и мы решим их как можно скорее 👇 ",
13
13
  "response": "Спасибо! Наша служба поддержки свяжется с вами в ближайшее время."
14
- },
15
- "feedback": {
16
- "message": "Мы ценим ваш отзыв! Пожалуйста, присылайте ваши предложения или проблемы, и мы рассмотрим их как можно скорее.",
17
- "response": "Спасибо за ваш отзыв!"
18
14
  }
19
15
  }
@@ -0,0 +1,57 @@
1
+ from functools import partial
2
+ from datetime import datetime
3
+ from telegram import Update
4
+ from telegram.ext import ContextTypes, Application, CommandHandler, MessageHandler, filters
5
+ from telegram.ext.filters import BaseFilter
6
+ from telegram_libs.mongo import mongo_client
7
+ from telegram_libs.translation import t
8
+
9
+
10
+ SUPPORT_WAITING = "support_waiting"
11
+
12
+
13
+ async def handle_support_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
14
+ """Support command handler"""
15
+ await update.message.reply_text(
16
+ t("support.message", update.effective_user.language_code, common=True)
17
+ )
18
+ context.user_data[SUPPORT_WAITING] = True
19
+
20
+
21
+ async def _handle_user_response(update: Update, context: ContextTypes.DEFAULT_TYPE, bot_name: str) -> None:
22
+ """Handle user's support message"""
23
+ if context.user_data.get(SUPPORT_WAITING):
24
+ db_name = "support"
25
+ collection_name = "support"
26
+ message_key = "support.response"
27
+ doc_field_name = "message"
28
+ context_key = SUPPORT_WAITING
29
+ extra_fields = {"resolved": False}
30
+ else:
31
+ # Should not happen if filter is correct
32
+ return
33
+
34
+ db = mongo_client[db_name]
35
+ collection = db[collection_name]
36
+ doc = {
37
+ "user_id": update.effective_user.id,
38
+ "username": update.effective_user.username,
39
+ doc_field_name: update.message.text,
40
+ "bot_name": bot_name,
41
+ "timestamp": datetime.now().isoformat(),
42
+ }
43
+ doc.update(extra_fields)
44
+ collection.insert_one(doc)
45
+ await update.message.reply_text(t(message_key, update.effective_user.language_code, common=True))
46
+ context.user_data[context_key] = False
47
+
48
+
49
+ class SupportFilter(BaseFilter):
50
+ def __call__(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> bool:
51
+ return context.user_data.get(SUPPORT_WAITING, False)
52
+
53
+
54
+ def register_support_handlers(app: Application, bot_name: str) -> None:
55
+ """Register support handlers for the bot"""
56
+ app.add_handler(CommandHandler("support", handle_support_command))
57
+ app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND & SupportFilter(), partial(_handle_user_response, bot_name=bot_name)))
@@ -0,0 +1,65 @@
1
+ from logging import basicConfig, getLogger, INFO
2
+ from telegram import (
3
+ InlineKeyboardButton,
4
+ InlineKeyboardMarkup,
5
+ )
6
+ from telegram import Update
7
+ from telegram.ext import ContextTypes, Application, CommandHandler
8
+ from telegram_libs.constants import BOTS_AMOUNT
9
+ from telegram_libs.translation import t
10
+ from telegram_libs.support_handlers import register_support_handlers
11
+
12
+
13
+ basicConfig(
14
+ format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=INFO
15
+ )
16
+ logger = getLogger(__name__)
17
+
18
+
19
+ async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboardMarkup:
20
+ """Get subscription keyboard
21
+
22
+ Args:
23
+ update (Update): Update object
24
+ lang (str): Language code
25
+
26
+ Returns:
27
+ InlineKeyboardMarkup: Inline keyboard markup
28
+ """
29
+ await update.message.reply_text(
30
+ t("subscription.info", lang, common=True).format(int(BOTS_AMOUNT) - 1)
31
+ )
32
+ return [
33
+ [
34
+ InlineKeyboardButton(
35
+ t("subscription.plans.1month", lang, common=True), callback_data="sub_1month"
36
+ ),
37
+ InlineKeyboardButton(
38
+ t("subscription.plans.3months", lang, common=True), callback_data="sub_3months"
39
+ ),
40
+ ],
41
+ [
42
+ InlineKeyboardButton(
43
+ t("subscription.plans.1year", lang, common=True), callback_data="sub_1year"
44
+ ),
45
+ ],
46
+ ]
47
+
48
+
49
+ async def more_bots_list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
50
+ message = """Here is the list of all bots:
51
+
52
+
53
+ - <a href="https://t.me/MagMediaBot">Remove Background</a>
54
+ - <a href="https://t.me/UpscaleImageGBot">Upscale Image</a>
55
+ - <a href="https://t.me/GenerateBackgroundGBot">Generate a Background</a>
56
+ - <a href="https://t.me/kudapoyti_go_bot">Recommend a place to visit</a>
57
+ - <a href="https://t.me/TryOnOutfitGBot">Try On Outfit</a>
58
+ """
59
+ await update.message.reply_text(message, disable_web_page_preview=True, parse_mode='HTML')
60
+
61
+
62
+ def register_common_handlers(app: Application, bot_name: str) -> None:
63
+ """Register common handlers for the bot"""
64
+ app.add_handler(CommandHandler("more", more_bots_list_command))
65
+ register_support_handlers(app, bot_name)
@@ -1,134 +0,0 @@
1
- from functools import partial
2
- from logging import basicConfig, getLogger, INFO
3
- from datetime import datetime
4
- from telegram import (
5
- InlineKeyboardButton,
6
- InlineKeyboardMarkup,
7
- )
8
- from telegram import Update
9
- from telegram.ext import ContextTypes, Application, CommandHandler, MessageHandler, filters
10
- from telegram.ext.filters import BaseFilter
11
- from telegram_libs.constants import BOTS_AMOUNT
12
- from telegram_libs.translation import t
13
- from telegram_libs.mongo import mongo_client
14
-
15
-
16
-
17
- basicConfig(
18
- format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=INFO
19
- )
20
- logger = getLogger(__name__)
21
-
22
-
23
- FEEDBACK_WAITING = "feedback_waiting"
24
- SUPPORT_WAITING = "support_waiting"
25
-
26
-
27
- async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboardMarkup:
28
- """Get subscription keyboard
29
-
30
- Args:
31
- update (Update): Update object
32
- lang (str): Language code
33
-
34
- Returns:
35
- InlineKeyboardMarkup: Inline keyboard markup
36
- """
37
- await update.message.reply_text(
38
- t("subscription.info", lang, common=True).format(int(BOTS_AMOUNT) - 1)
39
- )
40
- return [
41
- [
42
- InlineKeyboardButton(
43
- t("subscription.plans.1month", lang, common=True), callback_data="sub_1month"
44
- ),
45
- InlineKeyboardButton(
46
- t("subscription.plans.3months", lang, common=True), callback_data="sub_3months"
47
- ),
48
- ],
49
- [
50
- InlineKeyboardButton(
51
- t("subscription.plans.1year", lang, common=True), callback_data="sub_1year"
52
- ),
53
- ],
54
- ]
55
-
56
-
57
- async def more_bots_list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
58
- message = """Here is the list of all bots: \n\n
59
- - <a href="https://t.me/MagMediaBot">Remove Background</a>
60
- - <a href="https://t.me/UpscaleImageGBot">Upscale Image</a>
61
- - <a href="https://t.me/GenerateBackgroundGBot">Generate a Background</a>
62
- - <a href="https://t.me/kudapoyti_go_bot">Recommend a place to visit</a>
63
- - <a href="https://t.me/TryOnOutfitGBot">Try On Outfit</a>
64
- """
65
- await update.message.reply_text(message, disable_web_page_preview=True, parse_mode='HTML')
66
-
67
-
68
- async def handle_support_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
69
- """Support command handler"""
70
- await update.message.reply_text(
71
- t("support.message", update.effective_user.language_code, common=True)
72
- )
73
- context.user_data[SUPPORT_WAITING] = True
74
-
75
-
76
- async def _handle_user_response(update: Update, context: ContextTypes.DEFAULT_TYPE, bot_name: str) -> None:
77
- """Handle user's support or feedback message"""
78
- if context.user_data.get(FEEDBACK_WAITING):
79
- db_name = "feedback"
80
- collection_name = "feedback"
81
- message_key = "feedback.response"
82
- doc_field_name = "feedback"
83
- context_key = FEEDBACK_WAITING
84
- extra_fields = {}
85
- elif context.user_data.get(SUPPORT_WAITING):
86
- db_name = "support"
87
- collection_name = "support"
88
- message_key = "support.response"
89
- doc_field_name = "message"
90
- context_key = SUPPORT_WAITING
91
- extra_fields = {"resolved": False}
92
- else:
93
- # Should not happen if filter is correct
94
- return
95
-
96
- db = mongo_client[db_name]
97
- collection = db[collection_name]
98
- doc = {
99
- "user_id": update.effective_user.id,
100
- "username": update.effective_user.username,
101
- doc_field_name: update.message.text,
102
- "bot_name": bot_name,
103
- "timestamp": datetime.now().isoformat(),
104
- }
105
- doc.update(extra_fields)
106
- collection.insert_one(doc)
107
- await update.message.reply_text(t(message_key, update.effective_user.language_code, common=True))
108
- context.user_data[context_key] = False
109
-
110
-
111
- async def handle_feedback_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
112
- """Feedback command handler"""
113
- await update.message.reply_text(
114
- t("feedback.message", update.effective_user.language_code, common=True)
115
- )
116
- context.user_data[FEEDBACK_WAITING] = True
117
-
118
-
119
- class CombinedFeedbackSupportFilter(BaseFilter):
120
- def __call__(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> bool:
121
- return context.user_data.get(FEEDBACK_WAITING, False) or context.user_data.get(SUPPORT_WAITING, False)
122
-
123
-
124
- def register_feedback_and_support_handlers(app: Application, bot_name: str) -> None:
125
- """Register feedback and support handlers for the bot"""
126
- app.add_handler(CommandHandler("feedback", handle_feedback_command))
127
- app.add_handler(CommandHandler("support", handle_support_command))
128
- app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND & CombinedFeedbackSupportFilter(), partial(_handle_user_response, bot_name=bot_name)))
129
-
130
-
131
- def register_common_handlers(app: Application, bot_name: str) -> None:
132
- """Register common handlers for the bot"""
133
- app.add_handler(CommandHandler("more", more_bots_list_command))
134
- register_feedback_and_support_handlers(app, bot_name)
File without changes
File without changes