OpenSource/python-telegram-bot
OpenSource/python-telegram-bot
1
0
Fork 0
mirror of https://github.com/python-telegram-bot/python-telegram-bot.git synced 2024-11-21 22:56:38 +01:00
Code Issues Projects Releases Packages Wiki Activity

hack to detect image from stream

Browse source
This commit is contained in:
leandrotoledo 2015-07-13 18:53:30 -03:00
parent 3df9d2b429
commit b83a659bc2
1 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
telegram/inputfile.py
View file

@ -4,6 +4,7 @@
import mimetools
import mimetypes
import os
import re
import urllib2
@ -27,14 +28,16 @@ class InputFile(object):
self.input_file = data.pop('video')
if isinstance(self.input_file, file):
self.filename = os.path.basename(self.input_file.name)
self.input_file_content = self.input_file.read()
self.filename = os.path.basename(self.input_file.name)
self.mimetype = mimetypes.guess_type(self.filename)[0] or \
'application/octet-stream'
if 'http' in self.input_file:
self.filename = os.path.basename(self.input_file)
self.input_file_content = urllib2.urlopen(self.input_file).read()
self.mimetype = InputFile.image_type(self.input_file_content)
self.filename = self.mimetype.replace('/', '.')
self.mimetype = mimetypes.guess_type(self.filename)[0] or \
'application/octet-stream'
@property
def headers(self):
@ -73,3 +76,17 @@ class InputFile(object):
form.append('')
return '\r\n'.join(form)
@staticmethod
def image_type(stream):
header = stream[:10]
if re.match(r'GIF8', header):
return 'image/gif'
if re.match(r'\x89PNG', header):
return 'image/png'
if re.match(r'\xff\xd8\xff\xe0\x00\x10JFIF', header) or \
re.match(r'\xff\xd8\xff\xe1(.*){2}Exif', header):
return 'image/jpeg'