mirror of
https://github.com/eternnoir/pyTelegramBotAPI.git
synced 2021-05-13 03:08:02 +03:00
get updates and polling done.
This commit is contained in:
@@ -3,6 +3,8 @@
|
||||
import apihelper
|
||||
import json
|
||||
import types
|
||||
import time
|
||||
import threading
|
||||
|
||||
"""
|
||||
Module : telebot
|
||||
@@ -31,11 +33,62 @@ class TeleBot:
|
||||
|
||||
def __init__(self, token):
|
||||
self.token = token
|
||||
self.update_entries = {}
|
||||
self.update_listener = []
|
||||
self.chat_list = {}
|
||||
self.polling_thread = None
|
||||
self.__stop_polling = False
|
||||
self.interval = 3
|
||||
|
||||
def get_update(self):
|
||||
result = apihelper.get_updates(self.token)
|
||||
if result['ok'] is not True:
|
||||
raise Exception('getMe Error.' + json.dumps(result))
|
||||
updates = result['result']
|
||||
notify_updates = []
|
||||
for update in updates:
|
||||
if update['update_id'] in self.update_entries:
|
||||
continue
|
||||
msg = types.Message.de_json(json.dumps(update['message']))
|
||||
self.update_entries[update['update_id']] = msg
|
||||
notify_updates.append(msg)
|
||||
self.__notify_update(notify_updates)
|
||||
|
||||
def __notify_update(self, new_messages):
|
||||
for listener in self.update_listener:
|
||||
t = threading.Thread(target=listener, args=(new_messages))
|
||||
t.start()
|
||||
|
||||
def polling(self, interval):
|
||||
"""
|
||||
Always get updates.
|
||||
:param interval: iterval secs.
|
||||
:return:
|
||||
"""
|
||||
self.__stop_polling = True
|
||||
time.sleep(1)
|
||||
self.__stop_polling = False
|
||||
self.polling_thread = threading.Thread(target=self.__polling, args=())
|
||||
self.polling_thread.start()
|
||||
|
||||
def __polling(self):
|
||||
print 'telegram bot start polling'
|
||||
while not self.__stop_polling:
|
||||
self.get_update()
|
||||
time.sleep(self.interval)
|
||||
|
||||
print 'telegram bot stop polling'
|
||||
|
||||
def stop_polling(self):
|
||||
self.__stop_polling = True
|
||||
|
||||
def set_update_listener(self, listener):
|
||||
self.update_listener.append(listener)
|
||||
|
||||
def get_me(self):
|
||||
result = apihelper.get_me(self.token)
|
||||
if result['ok'] is not True:
|
||||
raise Exception('getMe Error.'+json.dumps(result))
|
||||
raise Exception('getMe Error.' + json.dumps(result))
|
||||
u = types.User.de_json(json.dumps(result['result']))
|
||||
return u
|
||||
|
||||
|
||||
Reference in New Issue
Block a user