client.py

import requests
import convertapi
import simplejson

from io import BytesIO
from .exceptions import *

class Client:
	def get(self, path, params = {}, timeout = None):
		timeout = timeout or convertapi.timeout
		r = requests.get(self.url(path), params = params, headers = self.headers(), timeout = timeout)
		return self.handle_response(r)

	def post(self, path, payload, timeout = None):
		timeout = timeout or convertapi.timeout
		r = requests.post(self.url(path), data = payload, headers = self.headers(), timeout = timeout)
		return self.handle_response(r)

	def upload(self, io, filename):
		url = convertapi.base_uri + 'upload'
		encoded_filename = requests.utils.quote(filename)

		headers = self.headers()
		headers.update({
			'Content-Disposition': "attachment; filename*=UTF-8''" + encoded_filename,
		})

		r = requests.post(url, data = io, headers = headers, timeout = convertapi.upload_timeout)
		return self.handle_response(r)

	def download(self, url, path):
		r = requests.get(url, stream = True, timeout = convertapi.download_timeout)

		with open(path, 'wb') as f:
			for chunk in r.iter_content(chunk_size = 1024):
				if chunk:
					f.write(chunk)

		return path

	def download_io(self, url):
		response = requests.get(url, timeout = convertapi.download_timeout)
		return BytesIO(response.content)

	def handle_response(self, r):
		try:
			r.raise_for_status()
		except requests.RequestException as e:
			try:
				raise ApiError(r.json())
			except ValueError:
				raise e

		if r.content == b'':
			raise AsyncConversionInProgress

		return r.json()

	def url(self, path):
		return "%s%s?Secret=%s" % (convertapi.base_uri, path, convertapi.api_secret)

	def headers(self):
		return {
			'User-Agent': convertapi.user_agent,
		}