mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2024-10-23 17:36:26 +02:00
136 lines
3.9 KiB
Python
136 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
|
|
import mimetools
|
|
import mimetypes
|
|
import os
|
|
import re
|
|
import urllib2
|
|
|
|
from .error import TelegramError
|
|
|
|
DEFAULT_MIME_TYPE = 'application/octet-stream'
|
|
USER_AGENT = 'Python Telegram Bot' \
|
|
' (https://github.com/leandrotoledo/python-telegram-bot)'
|
|
|
|
|
|
class InputFile(object):
|
|
def __init__(self,
|
|
data):
|
|
self.data = data
|
|
self.boundary = mimetools.choose_boundary()
|
|
|
|
if 'audio' in data:
|
|
self.input_name = 'audio'
|
|
self.input_file = data.pop('audio')
|
|
if 'document' in data:
|
|
self.input_name = 'document'
|
|
self.input_file = data.pop('document')
|
|
if 'photo' in data:
|
|
self.input_name = 'photo'
|
|
self.input_file = data.pop('photo')
|
|
if 'video' in data:
|
|
self.input_name = 'video'
|
|
self.input_file = data.pop('video')
|
|
|
|
if isinstance(self.input_file, file):
|
|
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 \
|
|
DEFAULT_MIME_TYPE
|
|
|
|
if 'http' in self.input_file:
|
|
self.input_file_content = urllib2.urlopen(self.input_file).read()
|
|
self.mimetype = InputFile.is_image(self.input_file_content)
|
|
self.filename = self.mimetype.replace('/', '.')
|
|
|
|
@property
|
|
def headers(self):
|
|
return {'User-agent': USER_AGENT,
|
|
'Content-type': self.content_type}
|
|
|
|
@property
|
|
def content_type(self):
|
|
return 'multipart/form-data; boundary=%s' % self.boundary
|
|
|
|
def to_form(self):
|
|
form = []
|
|
form_boundary = '--' + self.boundary
|
|
|
|
# Add data fields
|
|
for name, value in self.data.iteritems():
|
|
form.extend([
|
|
form_boundary,
|
|
str('Content-Disposition: form-data; name="%s"' % name),
|
|
'',
|
|
str(value)
|
|
])
|
|
|
|
# Add input_file to upload
|
|
form.extend([
|
|
form_boundary,
|
|
str('Content-Disposition: form-data; name="%s"; filename="%s"' % (
|
|
self.input_name, self.filename
|
|
)),
|
|
'Content-Type: %s' % self.mimetype,
|
|
'',
|
|
self.input_file_content
|
|
])
|
|
|
|
form.append('--' + self.boundary + '--')
|
|
form.append('')
|
|
|
|
return '\r\n'.join(form)
|
|
|
|
@staticmethod
|
|
def is_image(stream):
|
|
"""Check if the content file is an image by analyzing its headers.
|
|
|
|
Args:
|
|
stream:
|
|
A str representing the content of a file.
|
|
|
|
Returns:
|
|
The str mimetype of an image.
|
|
"""
|
|
try:
|
|
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'
|
|
except IndexError as e:
|
|
raise TelegramError(str(e))
|
|
|
|
raise TelegramError({'message': 'Could not parse file content'})
|
|
|
|
@staticmethod
|
|
def is_inputfile(data):
|
|
"""Check if the request is a file request
|
|
Args:
|
|
data:
|
|
A dict of (str, unicode) key/value pairs
|
|
|
|
Returns:
|
|
bool
|
|
"""
|
|
if data:
|
|
file_types = ['audio', 'document', 'photo', 'video']
|
|
file_type = [i for i in data.keys() if i in file_types]
|
|
|
|
if file_type:
|
|
file_content = data[file_type[0]]
|
|
|
|
if file_type[0] == 'photo':
|
|
return isinstance(file_content, file) or \
|
|
str(file_content).startswith('http')
|
|
|
|
return isinstance(file_content, file)
|
|
|
|
return False
|