TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-28 00:56:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add idle mechanism to gently stop the program

Browse Source
This commit is contained in:
Dan 2017-12-17 13:52:33 +01:00
parent c83daf8771
commit b6607fede0
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pyrogram/client/client.py
View File

@ -26,6 +26,8 @@ import time
from collections import namedtuple
from configparser import ConfigParser
from hashlib import sha256, md5
from signal import signal, SIGINT, SIGTERM, SIGABRT
from threading import Event
from pyrogram.api import functions, types
from pyrogram.api.core import Object
@ -71,6 +73,7 @@ class Client:
self.session = None
self.update_handler = None
self.is_idle = Event()
# TODO: Better update handler
def set_update_handler(self, callback: callable):
@ -79,6 +82,16 @@ class Client:
def send(self, data: Object):
return self.session.send(data)
def signal_handler(self, *args):
self.stop()
self.is_idle.set()
def idle(self, stop_signals: tuple = (SIGINT, SIGTERM, SIGABRT)):
for s in stop_signals:
signal(s, self.signal_handler)
self.is_idle.wait()
def authorize(self):
while True:
phone_number = input("Enter phone number: ")