Kaveh commited on
Commit
2bf8c3e
·
unverified ·
1 Parent(s): ced394c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -6
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import logging
2
  from telegram.ext import Updater, MessageHandler, Filters
3
  from bot.handlers import handle_message
 
4
  import os
 
5
 
6
- # لاگ برای دیباگ راحت‌تر
7
  logging.basicConfig(
8
  format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
9
  level=logging.INFO
@@ -11,22 +12,30 @@ logging.basicConfig(
11
 
12
  logger = logging.getLogger(__name__)
13
 
14
- # گرفتن توکن از متغیر محیطی (امن‌تر)
15
  TOKEN = os.environ.get("BOT_TOKEN")
16
 
 
 
 
 
 
 
 
 
17
  def main():
18
  if not TOKEN:
19
  logger.error("توکن ربات تنظیم نشده. متغیر محیطی BOT_TOKEN را تنظیم کن.")
20
  return
21
 
22
- # ساخت ربات
23
  updater = Updater(token=TOKEN, use_context=True)
24
  dp = updater.dispatcher
25
 
26
- # گرفتن همه پیام‌ها و فرستادنشون به handler
27
- dp.add_handler(MessageHandler(Filters.text & (~Filters.command), handle_message))
 
 
 
28
 
29
- # شروع ربات
30
  updater.start_polling()
31
  updater.idle()
32
 
 
1
  import logging
2
  from telegram.ext import Updater, MessageHandler, Filters
3
  from bot.handlers import handle_message
4
+ from bot.fetcher import save_message
5
  import os
6
+ from datetime import datetime
7
 
 
8
  logging.basicConfig(
9
  format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
10
  level=logging.INFO
 
12
 
13
  logger = logging.getLogger(__name__)
14
 
 
15
  TOKEN = os.environ.get("BOT_TOKEN")
16
 
17
+ def save_incoming_message(update, context):
18
+ chat_id = update.effective_chat.id
19
+ message = update.message.text
20
+ timestamp = datetime.fromtimestamp(update.message.date.timestamp())
21
+
22
+ if message:
23
+ save_message(chat_id, message, timestamp)
24
+
25
  def main():
26
  if not TOKEN:
27
  logger.error("توکن ربات تنظیم نشده. متغیر محیطی BOT_TOKEN را تنظیم کن.")
28
  return
29
 
 
30
  updater = Updater(token=TOKEN, use_context=True)
31
  dp = updater.dispatcher
32
 
33
+ # ذخیره‌سازی هر پیام متنی که میاد
34
+ dp.add_handler(MessageHandler(Filters.text & (~Filters.command), save_incoming_message), group=0)
35
+
36
+ # هندل پیام‌های فرمان خلاصه
37
+ dp.add_handler(MessageHandler(Filters.text & (~Filters.command), handle_message), group=1)
38
 
 
39
  updater.start_polling()
40
  updater.idle()
41