also check caption entities

This commit is contained in:
Tomáš Mládek 2018-02-06 19:30:58 +01:00 committed by Tomáš Mládek
parent b61933977d
commit 9530e328df

View file

@ -87,7 +87,7 @@ def download_raw(url, filename, subdir, date):
def get_first_hashtag(message): def get_first_hashtag(message):
hashtags = list(filter(lambda e: e.type == 'hashtag', message.entities)) hashtags = list(filter(lambda e: e.type == 'hashtag', message.entities + message.caption_entities))
if len(hashtags) == 0: if len(hashtags) == 0:
return None return None
hashtag = message.parse_entity(hashtags[0]) hashtag = message.parse_entity(hashtags[0])