yannik/tg4xmpp
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

whitespace cleanup

Browse Source
This commit is contained in:
adnidor
2019-03-10 17:05:32 +01:00
parent cea496c808
commit 5f2e2e517d
1 changed files with 149 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

216
xmpp_tg/mtproto.py
View File

@@ -1,18 +1,18 @@
from telethon import TelegramClient from telethon import TelegramClient
from telethon.utils import get_extension from telethon.utils import get_extension
from telethon.tl.types import UpdateShortMessage, UpdateShortChatMessage, UpdateEditMessage, UpdateDeleteMessages, \ from telethon.tl.types import UpdateShortMessage, UpdateShortChatMessage, UpdateEditMessage, UpdateDeleteMessages, \
UpdateNewMessage, UpdateUserStatus, UpdateShort, Updates, UpdateNewChannelMessage,\ UpdateNewMessage, UpdateUserStatus, UpdateShort, Updates, UpdateNewChannelMessage, \
UpdateChannelTooLong, UpdateDeleteChannelMessages, UpdateEditChannelMessage,\ UpdateChannelTooLong, UpdateDeleteChannelMessages, UpdateEditChannelMessage, \
UpdateUserName UpdateUserName
from telethon.tl.types import InputPeerChat, InputPeerUser, InputPeerChannel, InputUser from telethon.tl.types import InputPeerChat, InputPeerUser, InputPeerChannel, InputUser
from telethon.tl.types import MessageMediaDocument, MessageMediaPhoto, MessageMediaUnsupported, MessageMediaContact,\ from telethon.tl.types import MessageMediaDocument, MessageMediaPhoto, MessageMediaUnsupported, MessageMediaContact, \
MessageMediaGeo, MessageMediaEmpty, MessageMediaVenue MessageMediaGeo, MessageMediaEmpty, MessageMediaVenue
from telethon.tl.types import DocumentAttributeAnimated, DocumentAttributeAudio, DocumentAttributeFilename,\ from telethon.tl.types import DocumentAttributeAnimated, DocumentAttributeAudio, DocumentAttributeFilename, \
DocumentAttributeSticker, DocumentAttributeVideo, DocumentAttributeHasStickers DocumentAttributeSticker, DocumentAttributeVideo, DocumentAttributeHasStickers
from telethon.tl.types import Message, MessageService, MessageActionChannelCreate, MessageActionChannelMigrateFrom,\ from telethon.tl.types import Message, MessageService, MessageActionChannelCreate, MessageActionChannelMigrateFrom, \
MessageActionChatCreate, MessageActionChatAddUser, MessageActionChatDeleteUser,\ MessageActionChatCreate, MessageActionChatAddUser, MessageActionChatDeleteUser, \
MessageActionChatEditTitle, MessageActionChatJoinedByLink, MessageActionChatMigrateTo,\ MessageActionChatEditTitle, MessageActionChatJoinedByLink, MessageActionChatMigrateTo, \
MessageActionPinMessage MessageActionPinMessage
from telethon.tl.types import UserStatusOnline, UserStatusOffline, UserStatusRecently from telethon.tl.types import UserStatusOnline, UserStatusOffline, UserStatusRecently
from telethon.tl.types import User, Chat, Channel from telethon.tl.types import User, Chat, Channel
from telethon.tl.types import PeerUser, PeerChat, PeerChannel from telethon.tl.types import PeerUser, PeerChat, PeerChannel
@@ -28,6 +28,8 @@ from xmpp_tg.utils import localtime, display_tg_name
import xmpp_tg.monkey import xmpp_tg.monkey
import traceback import traceback
# modified by adnidor
class TelegramGateClient(TelegramClient): class TelegramGateClient(TelegramClient):
def __init__(self, session, api_id, api_hash, xmpp_gate, jid, phone, proxy=None): def __init__(self, session, api_id, api_hash, xmpp_gate, jid, phone, proxy=None):
@@ -53,7 +55,7 @@ class TelegramGateClient(TelegramClient):
self._del_pts = 0 self._del_pts = 0
def xmpp_update_handler(self, obj): def xmpp_update_handler(self, update_obj):
""" """
Main function: Telegram update handler. Main function: Telegram update handler.
@@ -65,127 +67,127 @@ class TelegramGateClient(TelegramClient):
# we have received some updates, so we're logined and can get <me> object and start mtd / upd threads # # we have received some updates, so we're logined and can get <me> object and start mtd / upd threads #
if not self.me: if not self.me:
me = self.get_me() me = self.get_me()
self.me = InputPeerUser(me.id, me.access_hash) self.me = InputPeerUser(me.id, me.access_hash)
self._media_thread.start() self._media_thread.start()
self._status_update_thread.start() self._status_update_thread.start()
nl = '\n' nl = '\n'
try: try:
# message from normal chat # # message from normal chat #
if type(obj) in [UpdateShortMessage] and not obj.out: if type(update_obj) in [UpdateShortMessage] and not update_obj.out:
fwd_from = self._process_forward_msg(obj) if obj.fwd_from else '' # process forward messages fwd_from = self._process_forward_msg(update_obj) if update_obj.fwd_from else '' # process forward messages
self.gate_send_message( mfrom='u' + str(obj.user_id), mbody = '[MSG {}] {}{}'.format(obj.id, fwd_from, obj.message) ) self.xmpp_send_message(mfrom='u' + str(update_obj.user_id), mbody ='[MSG {}] {}{}'.format(update_obj.id, fwd_from, update_obj.message))
usr = self._get_user_information(obj.user_id) # get peer information usr = self._get_user_information(update_obj.user_id) # get peer information
self.invoke(ReadHistoryRequest( InputPeerUser(usr.id, usr.access_hash), obj.id )) # delivery report self.invoke(ReadHistoryRequest(InputPeerUser(usr.id, usr.access_hash), update_obj.id)) # delivery report
# message from normal group # # message from normal group #
if type(obj) in [UpdateShortChatMessage] and not obj.out: if type(update_obj) in [UpdateShortChatMessage] and not update_obj.out:
fwd_from = self._process_forward_msg(obj) if obj.fwd_from else '' # process forward messages fwd_from = self._process_forward_msg(update_obj) if update_obj.fwd_from else '' # process forward messages
usr = self._get_user_information(obj.from_id) usr = self._get_user_information(update_obj.from_id)
nickname = display_tg_name(usr) nickname = display_tg_name(usr)
# send message # send message
self.gate_send_message(mfrom='g' + str(obj.chat_id), mbody ='[MSG {}] [User: {}] {}{}'.format(obj.id, nickname, fwd_from, obj.message) ) self.xmpp_send_message(mfrom='g' + str(update_obj.chat_id), mbody ='[MSG {}] [User: {}] {}{}'.format(update_obj.id, nickname, fwd_from, update_obj.message))
self.invoke(ReadHistoryRequest(InputPeerChat(obj.chat_id), obj.id)) self.invoke(ReadHistoryRequest(InputPeerChat(update_obj.chat_id), update_obj.id))
# message from supergroup or media message # # message from supergroup or media message #
if type(obj) in [UpdateNewMessage, UpdateNewChannelMessage, UpdateEditMessage, UpdateEditChannelMessage] and not obj.message.out: if type(update_obj) in [UpdateNewMessage, UpdateNewChannelMessage, UpdateEditMessage, UpdateEditChannelMessage] and not update_obj.message.out:
cid = None cid = None
msg = '' msg = ''
fwd_from = '' fwd_from = ''
mid = obj.message.id mid = update_obj.message.id
# detect message type # detect message type
is_user = type(obj.message.to_id) is PeerUser is_user = type(update_obj.message.to_id) is PeerUser
is_group = type(obj.message.to_id) is PeerChat is_group = type(update_obj.message.to_id) is PeerChat
is_supergroup = type(obj.message.to_id) is PeerChannel is_supergroup = type(update_obj.message.to_id) is PeerChannel
# detect from id # detect from id
if is_user: if is_user:
cid = obj.message.from_id cid = update_obj.message.from_id
user = self._get_user_information(cid) user = self._get_user_information(cid)
peer = InputPeerUser(user.id, user.access_hash) peer = InputPeerUser(user.id, user.access_hash)
prefix = 'u' prefix = 'u'
prefix = 'b' if user.bot else prefix prefix = 'b' if user.bot else prefix
elif is_group: elif is_group:
cid = obj.message.to_id.chat_id cid = update_obj.message.to_id.chat_id
peer = InputPeerChat(cid) peer = InputPeerChat(cid)
prefix = 'g' prefix = 'g'
elif is_supergroup: elif is_supergroup:
cid = obj.message.to_id.channel_id cid = update_obj.message.to_id.channel_id
peer = InputPeerChannel(cid, self.xmpp_gate.tg_dialogs[self.jid]['supergroups'][cid].access_hash) if cid in self.xmpp_gate.tg_dialogs[self.jid]['supergroups'] else None peer = InputPeerChannel(cid, self.xmpp_gate.tg_dialogs[self.jid]['supergroups'][cid].access_hash) if cid in self.xmpp_gate.tg_dialogs[self.jid]['supergroups'] else None
prefix = 's' prefix = 's'
# our message # # our message #
if type(obj.message) == MessageService: if type(update_obj.message) == MessageService:
obj.message.fwd_from, obj.message.post, obj.message.edit_date, obj.message.media = None, None, None, None update_obj.message.fwd_from, update_obj.message.post, update_obj.message.edit_date, update_obj.message.media = None, None, None, None
msg = self._process_info_msg(obj.message, peer) msg = self._process_info_msg(update_obj.message, peer)
elif type(obj.message) == Message: elif type(update_obj.message) == Message:
msg = obj.message.message msg = update_obj.message.message
# is forwarded? # is forwarded?
if obj.message.fwd_from: if update_obj.message.fwd_from:
fwd_from = self._process_forward_msg(obj.message) fwd_from = self._process_forward_msg(update_obj.message)
# maybe its channel? # # maybe its channel? #
if obj.message.post: if update_obj.message.post:
prefix = 'c' prefix = 'c'
# get sender information from chat info # # get sender information from chat info #
if not is_user and not obj.message.post: if not is_user and not update_obj.message.post:
usr = self._get_user_information(obj.message.from_id) usr = self._get_user_information(update_obj.message.from_id)
nickname = display_tg_name(usr) nickname = display_tg_name(usr)
msg = '[User: {}] {}'.format(nickname, msg) msg = '[User: {}] {}'.format(nickname, msg)
# message media # # message media #
if obj.message.media: if update_obj.message.media:
msg = '{} {}'.format( msg, self._process_media_msg(obj.message.media) ) msg = '{} {}'.format(msg, self._process_media_msg(update_obj.message.media))
# edited # # edited #
if obj.message.edit_date: if update_obj.message.edit_date:
msg = '[Edited] {}'.format(msg) msg = '[Edited] {}'.format(msg)
# send message # # send message #
self.gate_send_message(prefix + str(cid), mbody = '[MSG {}] {}{}'.format(mid, fwd_from, msg) ) self.xmpp_send_message(prefix + str(cid), mbody ='[MSG {}] {}{}'.format(mid, fwd_from, msg))
# delivery report # delivery report
if is_supergroup: if is_supergroup:
self.invoke(ReadHistoryChannel(peer, mid)) self.invoke(ReadHistoryChannel(peer, mid))
else: else:
self.invoke(ReadHistoryRequest(peer, mid)) self.invoke(ReadHistoryRequest(peer, mid))
# Status Updates # # Status Updates #
if type(obj) is UpdateUserStatus: if type(update_obj) is UpdateUserStatus:
# process status update # # process status update #
if type(obj.status) is UserStatusOnline: if type(update_obj.status) is UserStatusOnline:
self._status_updates[str(obj.user_id)] = { 'status': None, 'message': 'Online' } self._status_updates[str(update_obj.user_id)] = {'status': None, 'message': 'Online'}
elif type(obj.status) is UserStatusOffline: elif type(update_obj.status) is UserStatusOffline:
status = 'away' if datetime.datetime.utcnow() - obj.status.was_online < datetime.timedelta(hours = self.xmpp_gate.accounts[self.jid]['status_xa_interval'] ) else 'xa' status = 'away' if datetime.datetime.utcnow() - update_obj.status.was_online < datetime.timedelta(hours = self.xmpp_gate.accounts[self.jid]['status_xa_interval']) else 'xa'
self._status_updates[str(obj.user_id)] = { 'status': status, 'message': localtime(obj.status.was_online).strftime('Last seen at %H:%M %d/%m/%Y') } self._status_updates[str(update_obj.user_id)] = {'status': status, 'message': localtime(update_obj.status.was_online).strftime('Last seen at %H:%M %d/%m/%Y')}
elif type(obj.status) is UserStatusRecently: elif type(update_obj.status) is UserStatusRecently:
self._status_updates[str(obj.user_id)] = { 'status': 'dnd', 'message': 'Last seen recently' } self._status_updates[str(update_obj.user_id)] = {'status': 'dnd', 'message': 'Last seen recently'}
else: else:
pass pass
except Exception: except Exception:
print('Exception occurs!') print('Exception occurs!')
print(traceback.format_exc()) print(traceback.format_exc())
def gate_send_message(self, mfrom, mbody): def xmpp_send_message(self, mfrom, mbody):
tg_from = int(mfrom[1:]) tg_from = int(mfrom[1:])
if not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['users'] and not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['groups'] and not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['supergroups']: # new contact appeared if not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['users'] and not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['groups'] and not tg_from in self.xmpp_gate.tg_dialogs[self.jid]['supergroups']: # new contact appeared
self.xmpp_gate.tg_process_dialogs( self.jid ) self.xmpp_gate.tg_process_dialogs( self.jid )
self.xmpp_gate.send_message( mto=self.jid, mfrom=mfrom + '@' + self.xmpp_gate.config['jid'], mtype='chat', mbody=mbody) self.xmpp_gate.send_message( mto=self.jid, mfrom=mfrom + '@' + self.xmpp_gate.config['jid'], mtype='chat', mbody=mbody)
@@ -230,15 +232,15 @@ class TelegramGateClient(TelegramClient):
def _get_user_information(self, uid): def _get_user_information(self, uid):
if uid in self.xmpp_gate.tg_dialogs[self.jid]['users']: if uid in self.xmpp_gate.tg_dialogs[self.jid]['users']:
return self.xmpp_gate.tg_dialogs[self.jid]['users'][uid] return self.xmpp_gate.tg_dialogs[self.jid]['users'][uid]
entity = self.get_entity(uid) entity = self.get_entity(uid)
if entity.access_hash: if entity.access_hash:
self.xmpp_gate.tg_dialogs[self.jid]['users'][uid] = entity self.xmpp_gate.tg_dialogs[self.jid]['users'][uid] = entity
return entity return entity
else: else:
return {'first_name': 'Unknown', 'last_name': 'user', 'access_hash': -1, 'id': 0, 'bot': False} return {'first_name': 'Unknown', 'last_name': 'user', 'access_hash': -1, 'id': 0, 'bot': False}
def _process_forward_msg(self, message): def _process_forward_msg(self, message):
@@ -326,7 +328,7 @@ class TelegramGateClient(TelegramClient):
msg = '{} {}'.format(media.caption, msg) msg = '{} {}'.format(media.caption, msg)
elif type(media) is MessageMediaContact: # contact elif type(media) is MessageMediaContact: # contact
msg = 'First name: {} / Last name: {} / Phone: {}'\ msg = 'First name: {} / Last name: {} / Phone: {}' \
.format(media.first_name, media.last_name, media.phone_number) .format(media.first_name, media.last_name, media.phone_number)
elif type(media) in [MessageMediaGeo, MessageMediaVenue]: # address elif type(media) in [MessageMediaGeo, MessageMediaVenue]: # address
map_link_template = 'https://maps.google.com/maps?q={0:.4f},{1:.4f}&ll={0:.4f},{1:.4f}&z=16' map_link_template = 'https://maps.google.com/maps?q={0:.4f},{1:.4f}&ll={0:.4f},{1:.4f}&z=16'
@@ -362,12 +364,12 @@ class TelegramGateClient(TelegramClient):
elif type(message.action) is MessageActionChatAddUser: elif type(message.action) is MessageActionChatAddUser:
added_users = [] added_users = []
for user_id in message.action.users: for user_id in message.action.users:
usr = self._get_user_information(user_id) usr = self._get_user_information(user_id)
added_users.append(display_tg_name(usr)) added_users.append(display_tg_name(usr))
msg = 'User [{}] has just invited [{}]'.format(nickname, ','.join(added_users)) msg = 'User [{}] has just invited [{}]'.format(nickname, ','.join(added_users))
# user exit # # user exit #
elif type(message.action) is MessageActionChatDeleteUser: elif type(message.action) is MessageActionChatDeleteUser:
usr = self._get_user_information(message.action.user_id) usr = self._get_user_information(message.action.user_id)
msg = 'User [{}] has just left the room'.format(display_tg_name(usr)) msg = 'User [{}] has just left the room'.format(display_tg_name(usr))