def test_equality(self):
a = ChatPhoto(
self.chatphoto_small_file_id,
self.chatphoto_big_file_id,
self.chatphoto_small_file_unique_id,
self.chatphoto_big_file_unique_id,
)
b = ChatPhoto(
self.chatphoto_small_file_id,
self.chatphoto_big_file_id,
self.chatphoto_small_file_unique_id,
self.chatphoto_big_file_unique_id,
)
c = ChatPhoto(
'', '', self.chatphoto_small_file_unique_id, self.chatphoto_big_file_unique_id
)
d = ChatPhoto('', '', 0, 0)
e = Voice(self.chatphoto_small_file_id, self.chatphoto_small_file_unique_id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a != c
assert hash(a) != hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def _get_voice(self):
import uuid
from random import randint
return Voice(
file_id=str(uuid.uuid4()),
file_unique_id=str(uuid.uuid4()),
duration=randint(2, 300),
)
def complete_process_voice(bot: Bot, update: Update):
driver = Taxopark.get_driver(tg_id=update.effective_user.id)
voice = update.effective_message.voice.file_id
duration = update.effective_message.voice.duration
dispatcher_chat_id = Taxopark.get_dispatcher_chat_id()
bot.send_message(dispatcher_chat_id,
f'Объяснительная от "{driver.name} {driver.surname}"')
bot.send_voice(dispatcher_chat_id, Voice(voice, duration))
return ConversationHandler.END
def test_equality(self, voice):
a = Voice(voice.file_id, voice.file_unique_id, self.duration)
b = Voice('', voice.file_unique_id, self.duration)
c = Voice(voice.file_id, voice.file_unique_id, 0)
d = Voice('', '', self.duration)
e = Audio(voice.file_id, voice.file_unique_id, self.duration)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_voice(self):
voice = Voice("idyouknow", 12)
u = self.mg.get_message(voice=voice)
self.assertEqual(voice.file_id, u.message.voice.file_id)
cap = "voice file"
u = self.mg.get_message(voice=voice, caption=cap)
self.assertEqual(u.message.caption, cap)
u = self.mg.get_message(voice=True)
self.assertIsInstance(u.message.voice, Voice)
with self.assertRaisesRegexp(BadMessageException, r"telegram\.Voice"):
self.mg.get_message(voice="voice")
def test_equality(self):
a = Animation(self.animation_file_id)
b = Animation(self.animation_file_id)
d = Animation('')
e = Voice(self.animation_file_id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self):
a = Animation(self.animation_file_id, self.height, self.width, self.duration)
b = Animation(self.animation_file_id, self.height, self.width, self.duration)
d = Animation('', 0, 0, 0)
e = Voice(self.animation_file_id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, document):
a = Document(document.file_id)
b = Document(document.file_id)
d = Document('')
e = Voice(document.file_id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, user):
a = ChosenInlineResult(self.result_id, user, "Query", "")
b = ChosenInlineResult(self.result_id, user, "Query", "")
c = ChosenInlineResult(self.result_id, user, "", "")
d = ChosenInlineResult("", user, "Query", "")
e = Voice(self.result_id, "unique_id", 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, video_note):
a = VideoNote(video_note.file_id, video_note.file_unique_id, self.length, self.duration)
b = VideoNote('', video_note.file_unique_id, self.length, self.duration)
c = VideoNote(video_note.file_id, video_note.file_unique_id, 0, 0)
d = VideoNote('', '', self.length, self.duration)
e = Voice(video_note.file_id, video_note.file_unique_id, self.duration)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self):
a = Contact(self.phone_number, self.first_name)
b = Contact(self.phone_number, self.first_name)
c = Contact(self.phone_number, '')
d = Contact('', self.first_name)
e = Voice('', 'unique_id', 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, video):
a = Video(video.file_id, self.width, self.height, self.duration)
b = Video(video.file_id, self.width, self.height, self.duration)
c = Video(video.file_id, 0, 0, 0)
d = Video('', self.width, self.height, self.duration)
e = Voice(video.file_id, self.duration)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self):
a = ShippingOption(self.id, self.title, self.prices)
b = ShippingOption(self.id, self.title, self.prices)
c = ShippingOption(self.id, '', [])
d = ShippingOption(0, self.title, self.prices)
e = Voice(self.id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, bot):
a = File(self.file_id, self.file_unique_id, bot)
b = File('', self.file_unique_id, bot)
c = File(self.file_id, self.file_unique_id, None)
d = File('', '', bot)
e = Voice(self.file_id, self.file_unique_id, 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, audio):
a = Audio(audio.file_id, audio.file_unique_id, audio.duration)
b = Audio('', audio.file_unique_id, audio.duration)
c = Audio(audio.file_id, audio.file_unique_id, 0)
d = Audio('', '', audio.duration)
e = Voice(audio.file_id, audio.file_unique_id, audio.duration)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def test_equality(self, user):
a = ChosenInlineResult(self.result_id, user, 'Query', '')
b = ChosenInlineResult(self.result_id, user, 'Query', '')
c = ChosenInlineResult(self.result_id, user, '', '')
d = ChosenInlineResult('', user, 'Query', '')
e = Voice(self.result_id, 'unique_id', 0)
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a == c
assert hash(a) == hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
def message_to_text(update, context, send_back_voice=True):
if update.message.voice:
with TempFiles(".ogg", ".wav") as (tmp_ogg, tmp_wav):
voice_obj = update.message.voice
voice_file = Voice(voice_obj.file_id,
voice_obj.file_unique_id,
voice_obj.duration,
bot=context.bot)
voice_file.get_file(timeout=100).download(tmp_ogg)
_convert_audio(tmp_ogg, tmp_wav)
text = stt_wav_to_string(tmp_wav)
if send_back_voice:
context.bot.send_message(chat_id=update.effective_chat.id,
text=f"\"{text}\"")
return text
return update.message.text
'caption': 'animation_file'
}, {
'game':
Game('my_game', 'just my game', [
PhotoSize('game_photo_id', 30, 30),
])
}, {
'photo': [PhotoSize('photo_id', 50, 50)],
'caption': 'photo_file'
}, {
'sticker': Sticker('sticker_id', 50, 50)
}, {
'video': Video('video_id', 12, 12, 12),
'caption': 'video_file'
}, {
'voice': Voice('voice_id', 5)
}, {
'video_note': VideoNote('video_note_id', 20, 12)
}, {
'new_chat_members': [User(55, 'new_user', False)]
}, {
'contact': Contact('phone_numner', 'contact_name')
}, {
'location': Location(-23.691288, 46.788279)
}, {
'venue':
Venue(Location(-23.691288, 46.788279), 'some place', 'right here')
}, {
'left_chat_member': User(33, 'kicked', False)
}, {
'new_chat_title': 'new title'
],
)
},
{
'photo': [PhotoSize('photo_id', 'unique_id', 50, 50)],
'caption': 'photo_file'
},
{
'sticker': Sticker('sticker_id', 'unique_id', 50, 50, True)
},
{
'video': Video('video_id', 'unique_id', 12, 12, 12),
'caption': 'video_file'
},
{
'voice': Voice('voice_id', 'unique_id', 5)
},
{
'video_note': VideoNote('video_note_id', 'unique_id', 20, 12)
},
{
'new_chat_members': [User(55, 'new_user', False)]
},
{
'contact': Contact('phone_numner', 'contact_name')
},
{
'location': Location(-23.691288, 46.788279)
},
{
'venue':
def _get_voice(self):
import uuid
from random import randint
return Voice(str(uuid.uuid4()), randint(1, 120))
def read_file(update, context):
if update.message.reply_to_message:
if update.message.reply_to_message.animation:
file_id = str(update.message.reply_to_message.animation.file_id)
unique_id = str(
update.message.reply_to_message.animation.file_unique_id)
width = str(update.message.reply_to_message.animation.width)
height = str(update.message.reply_to_message.animation.height)
duration = str(update.message.reply_to_message.animation.duration)
# context.bot.send_message(update.effective_user.id,text=f"file_id = '{file_id}', \nfile_unique_id = '{unique_id}', \nwidth = {width}, \nheight = {height}, \nduration = {duration}")
animation = Animation(file_id, unique_id, width, height, duration)
update.message.reply_animation(
animation,
caption=
f"This is an animation. \n\n\nfile_id = '{file_id}', \n\nfile_unique_id = '{unique_id}', \n\nwidth = {width}, \n\nheight = {height}, \n\nduration = {duration}"
)
if update.message.reply_to_message.photo:
file_id = str(update.message.reply_to_message.photo[2]['file_id'])
unique_id = str(
update.message.reply_to_message.photo[2]['file_unique_id'])
width = str(update.message.reply_to_message.photo[2]['width'])
height = str(update.message.reply_to_message.photo[2]['height'])
file_size = str(
update.message.reply_to_message.photo[2]['file_size'])
photo = PhotoSize(file_id, unique_id, width, height, file_size)
update.message.reply_photo(
photo,
caption=
f"This is a photo. \n\n\nfile_id = '{file_id}', \n\nfile_unique_id = '{unique_id}', \n\nwidth = {width},\n\nheight = {height},\n\nfile_size = {file_size}\n\n"
)
if update.message.reply_to_message.voice:
file_id = str(update.message.reply_to_message.voice.file_id)
unique_id = str(
update.message.reply_to_message.voice.file_unique_id)
duration = str(update.message.reply_to_message.voice.duration)
mime_type = str(update.message.reply_to_message.voice.mime_type)
file_size = str(update.message.reply_to_message.voice.file_size)
voice = Voice(file_id, unique_id, duration, mime_type, file_size)
update.message.reply_voice(
voice,
caption=
f"This is a voice message. \n\n\nfile_id = '{file_id}', \n\nfile_unique_id = '{unique_id}', \n\nduration = {duration},\n\nmime_type = {mime_type},\n\nfile_size = {file_size}\n\n"
)
if update.message.reply_to_message.audio:
file_id = str(update.message.reply_to_message.audio.file_id)
unique_id = str(
update.message.reply_to_message.audio.file_unique_id)
duration = str(update.message.reply_to_message.audio.duration)
performer = str(update.message.reply_to_message.audio.performer)
title = str(update.message.reply_to_message.audio.title)
mime_type = str(update.message.reply_to_message.audio.mime_type)
file_size = str(update.message.reply_to_message.audio.file_size)
audio = Audio(file_id, unique_id, duration, performer, title,
mime_type, file_size)
update.message.reply_audio(
audio,
caption=
f"This is an audio. \n\n\nfile_id = '{file_id}', \n\nfile_unique_id = '{unique_id}', \n\nduration = {duration},\n\nperformer = {performer} \n\ntitle = {title}, \n\nmime_type = {mime_type},\n\nfile_size = {file_size}"
)
if update.message.reply_to_message.document:
file_id = str(update.message.reply_to_message.document.file_id)
unique_id = str(
update.message.reply_to_message.document.file_unique_id)
document = Document(file_id, unique_id)
update.message.reply_document(
document,
caption=
f"This is a document. \n\n\nfile_id = '{file_id}',\n\nfile_unique_id = '{unique_id}'"
)
if update.message.reply_to_message.video:
file_id = str(update.message.reply_to_message.video.file_id)
unique_id = str(
update.message.reply_to_message.video.file_unique_id)
width = str(update.message.reply_to_message.video.width)
height = str(update.message.reply_to_message.video.height)
duration = str(update.message.reply_to_message.video.duration)
video = Video(file_id, unique_id, width, height, duration)
update.message.reply_video(
video,
caption=
f"This is a video. \n\n\nfile_id = '{file_id}',\n\nfile_unique_id = '{unique_id}', \n\nwidth = {width}, \n\nheight = {height},\n\nduration = {duration}"
)
if update.message.reply_to_message.sticker:
file_id = str(update.message.reply_to_message.sticker.file_id)
unique_id = str(
update.message.reply_to_message.sticker.file_unique_id)
width = str(update.message.reply_to_message.sticker.width)
height = str(update.message.reply_to_message.sticker.height)
is_animated = str(
update.message.reply_to_message.sticker.is_animated)
sticker = Sticker(file_id, unique_id, width, height, is_animated)
update.message.reply_sticker(sticker, caption='')
context.bot.send_message(
update.effective_chat.id,
text=
f"This is a sticker. \n\n\nfile_id = '{file_id}',\n\nfile_unique_id = '{unique_id}', \n\nwidth = {width}, \n\nheight = {height},\n\nis_animated = {is_animated}"
)
if update.message.reply_to_message.game:
title = str(update.message.reply_to_message.game.title)
description = str(update.message.reply_to_message.game.description)
photo = str(update.message.reply_to_message.game.photo)
text = str(update.message.reply_to_message.game.text)
text_entities = str(
update.message.reply_to_message.game.text_entities)
animation = str(update.message.reply_to_message.game.animation)
game = Game(title, description, photo, text, text_entities,
animation)
update.message.reply_game(
game,
caption=
f'This is a game.\n\n\ntitle = {title}, \n\ndescription = {description}, \n\nphoto = {photo}, \n\ntext = {text}, \n\ntext_entities = {text_entities}, \n\nanimation = {animation}'
)
if update.message.reply_to_message.video_note:
file_id = str(update.message.reply_to_message.video_note.file_id)
unique_id = str(
update.message.reply_to_message.video_note.file_unique_id)
length = str(update.message.reply_to_message.video_note.length)
duration = str(update.message.reply_to_message.video_note.duration)
videoNote = VideoNote(file_id, unique_id, length, duration)
update.message.reply_video_note(videoNote, caption='')
context.bot.send_message(
update.effective_chat.id,
text=
f"This is a video message.\n\n\nfile_id = '{file_id}',\n\nfile_unique_id = '{unique_id}',\n\nlength = {length}, \n\nduration = {duration}"
)
if update.message.reply_to_message.text:
ttext = str(update.message.reply_to_message.text)
update.message.reply_text(f'{ttext}\n\nThis is a text.')
else:
u = str(update)
u = dumps(eval(u), indent=2)
update.message.reply_text(u)
def _handle_attachments(
self,
audio,
contact,
document,
location,
photo,
sticker,
user,
venue,
video,
voice,
caption,
):
attachments = [
x
for x in [
photo,
venue,
location,
contact,
voice,
video,
sticker,
document,
audio,
]
if x
]
if caption and not attachments:
raise BadMessageException("Can't have a caption without attachment")
if len(attachments) > 1:
raise BadMessageException("can't add more than one attachment")
if photo:
if isinstance(photo, list):
if all([isinstance(x, PhotoSize) for x in photo]):
pass
else:
raise BadMessageException(
"photo must either be True or list(telegram.PhotoSize)"
)
elif isinstance(photo, bool):
photo = self._get_photosize()
else:
raise BadMessageException(
"photo must either be True or list(telegram.PhotoSize)"
)
if location:
if isinstance(location, Location):
pass
elif isinstance(location, dict):
location = Location(**location)
elif isinstance(location, bool):
location = self._get_location()
else:
raise BadMessageException(
"location must either be True or telegram.Location"
)
if venue:
if isinstance(venue, Venue):
pass
elif isinstance(venue, bool):
venue = self._get_venue()
elif isinstance(venue, dict):
venue["location"] = Location(**venue)
venue = Venue(**venue)
else:
raise BadMessageException("venue must either be True or telegram.Venue")
if contact:
if isinstance(contact, Contact):
pass
elif isinstance(contact, dict):
contact = Contact(**contact)
elif isinstance(contact, bool):
contact = self._get_contact(user)
else:
raise BadMessageException(
"contact must either be True or telegram.Contact"
)
if voice:
if isinstance(voice, Voice):
pass
elif isinstance(voice, bool):
voice = self._get_voice()
elif isinstance(voice, dict):
voice = Voice(**voice)
else:
raise BadMessageException("voice must either be True or telegram.Voice")
if video:
if isinstance(video, Video):
pass
elif isinstance(video, bool):
video = self._get_video()
elif isinstance(video, dict):
video = self._get_video(data=video)
else:
raise BadMessageException("video must either be True or telegram.Video")
if sticker:
if isinstance(sticker, Sticker):
pass
elif isinstance(sticker, bool):
sticker = self._get_sticker()
elif isinstance(sticker, dict):
sticker = self._get_sticker(sticker)
else:
raise BadMessageException(
"sticker must either be True or telegram.Sticker"
)
if document:
if isinstance(document, Document):
pass
elif isinstance(document, dict):
document = Document(**document)
elif isinstance(document, bool):
document = self._get_document()
else:
raise BadMessageException(
"document must either be True or telegram.Document"
)
if audio:
if isinstance(audio, Audio):
pass
elif isinstance(audio, bool):
audio = self._get_audio()
elif isinstance(audio, dict):
audio = Audio(**audio)
else:
raise BadMessageException("audio must either be True or telegram.Audio")
return audio, contact, document, location, photo, sticker, venue, video, voice
