From 94adeca045247f5d92d9bc1a9cd1e9b42553d5e0 Mon Sep 17 00:00:00 2001 From: joey0629 Date: Thu, 29 Feb 2024 19:57:51 +0800 Subject: [PATCH] PCE scheduling broadcast update --- main.py | 132 ++++++++++++++++++-------------------------------------- 1 file changed, 43 insertions(+), 89 deletions(-) diff --git a/main.py b/main.py index 4b21dc4..887b2dc 100644 --- a/main.py +++ b/main.py @@ -1,84 +1,23 @@ - -import re import telebot from config import Config -from ai_assistant import AI_assistant import pdfplumber import time -# 使用Token來初始化一個telebot.TeleBot物件，並將其儲存到bot這個變數中 +import requests +from datetime import datetime +import schedule BOT_TOKEN = '6701395239:AAFE30dqvNihDdni9vYoAbWssO-X5yAmwho' # BOT_TOKEN = "6746720034:AAEMaoV2FwIZ8pz_PF18-bo2a6gFC1eVtVs" #BOT_TOKEN = '6589162555:AAHGhrTQ0wYNtIUySMohnfpxQl1d6blr24Q' -bot = telebot.TeleBot(BOT_TOKEN) - -user_url = None -user_answer = None - - - - -# 定義消息處理器，當收到網址時 -# 考慮到medium也有提供短網址，所以暫不把qffers判斷寫進去 -@bot.message_handler(func=lambda msg: re.search(r'http[s]?://(www\.)?(link\.)?medium\.com/', msg.text) if msg.text else False) -def handle_medium_url(message): - # bot.reply_to(message, "想了解什麼訊息呢?問問我吧！") # 這邊改到上面不確定會不會比較好 - - global user_url, identify - user_url = message.text - #contents, lang, identify = get_contents(user_url) - #save_to_storage(contents, identify) - bot.reply_to(message, "想了解什麼訊息呢?問問我吧！") - -@bot.message_handler(func=lambda msg: re.search(r'http[s]?://', msg.text) if msg.text else False) -def handle_other_url(message): - bot.reply_to(message, "此網頁不支援唷😂😂\n請試試看輸入 https://link.medium.com/rxe98Z708Db ", disable_web_page_preview=True) - - - -# 定義消息處理器，當用戶發送一個消息時，機器人將回覆相同的消息 -@bot.message_handler(func=lambda msg: msg.text.lower() in ["hi", "hello", "嗨", "你好", "早上好", "晚上好", "早安", "晚安", "介紹", "誰"]) -def reply_all(message): - print(message) #你可以出來看看會有啥 - print(message.text) #單純擷取文字 - user_first_name = message.from_user.first_name - intro = f'嗨, {user_first_name}！👋👋\n我們是睿富者（QFFERS）\n歡迎到我們的Medium文章看看~ https://link.medium.com/rxe98Z708Db \n \n也歡迎試著貼上Medium文章網址問個問題吧！😂😂\n' - result = start() - print(result) - bot.reply_to(message, result, disable_web_page_preview=True) -@bot.message_handler(func=lambda msg: True) -def handle_user_answer(message): - global user_answer , startime - user_answer = message.text - # result = answering(user_answer) - - result = start() - print("This is result", result) - end =time.time() - print("Time: ", end-startime) - bot.reply_to(message, result) - -# Using assistant API to answer question -def answering(query): - files = ai_assistant.get_files() - ai_assistant.create_assistant( - name="QFFERS Bot", - instructions="你是一個天問Bot機器人，你的任務是請基於用戶上傳的PDF上找尋用戶所要找尋的答案、數值。" - "任務說明:用戶提問時，請仔細分析問題並提供基於上傳PDF。如果答案來自PDF檔案請提供該篇PDF的段落，若沒有資料請回答:我不知道", - tools=[{"type": "retrieval"}], - files=files - ) - ai_assistant.create_thread() - ai_assistant.add_message_to_thread( - role="user", - content=query +def broadcast_message(message:str,chat_id:str): + r = requests.post(f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage", + json={ + "chat_id": chat_id, + "text": message, + }, ) - ai_assistant.run_assistant( - instructions="Please user's language to answer the question. You can only answer according to the uploaded files.") - ai_assistant.check_run(thread_id=ai_assistant.thread.id, run_id=ai_assistant.run.id) - total_price , content = ai_assistant.process_messages() - return content + def start(): global startime @@ -112,13 +51,6 @@ def find_indicate(url): # 是一個持續運行的迴圈，不斷從Telegram伺服器抓取新的消息 # 然後使用上面定義的消息處理器來處理這些消息。 -def tg_bot(cfg: Config): - """Run the Telegram bot.""" - global ai_assistant, storage ,bot - print("Starting Telegram bot...") - # ai_assistant = AI_assistant(cfg) - # 啟動Telegram Bot - bot.infinity_polling() #非農就業人數 def read_pdf_nonfarm(month, year): pdf = pdfplumber.open(f"empsit/empsit_{month}_{year}.pdf") @@ -136,16 +68,38 @@ def read_nonfarm(): endtimee = time.time() print("Time: ", endtimee-startimee) # print(text.split('\n')[7:9]) -if __name__ == "__main__": - #非農 - startimee = time.time() - print(f"2023年7月非農就業人數: ", end= "" ) - print(read_pdf_nonfarm(7, 23)) - endtimee = time.time() - print("Time_NonFarm: ", endtimee-startimee) + +def read_PCE(): + message = find_indicate_bs4("https://www.bea.gov/data/personal-consumption-expenditures-price-index") + + return message +# def read_PCE_test(): +# from bs4 import BeautifulSoup +# import requests +# response = requests.get("http://127.0.0.1:5000") +# soup = BeautifulSoup(response.text, 'html.parser') +# +# # 使用CSS選擇器找到元素 +# element = soup.select("p") +# print(element[0].text) +# return element[0].text +def broadcast_all(target:str): startimee = time.time() - print(find_indicate_bs4("https://www.bea.gov/data/personal-consumption-expenditures-price-index")) + message = read_PCE() + while message.split(' ')[0] != target: + message = read_PCE() + broadcast_message(message, "-1002033782195") endtimee = time.time() - print("Time_NonFarm: ", endtimee-startimee) - # cfg = Config() - # tg_bot(cfg) \ No newline at end of file + broadcast_message( + f"Now :{datetime.fromtimestamp(time.time())} , Spend time :{str(round(endtimee - startimee, 3))} s", + "-1002033782195") + +if __name__ == "__main__": + #PCE + print("Start Time:" , datetime.fromtimestamp(time.time())) + schedule.every().day.at("20:05").do(broadcast_all) + while True: + if datetime.now().strftime("%Y-%m-%d %H:%M") == "2024-02-29 20:05": + schedule.run_pending() + time.sleep(0.1) # Check every 0.1 seconds + broadcast_all("December")