From 8c3e123e0a232b5e8dfb9f405711f2bcf78ff33d Mon Sep 17 00:00:00 2001 From: David Medina Date: Sun, 19 Feb 2012 19:41:45 +0100 Subject: Support to Mimetypes in some services --- pygithub3/services/base.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'pygithub3/services/base.py') diff --git a/pygithub3/services/base.py b/pygithub3/services/base.py index 1f5d07e..21b22eb 100644 --- a/pygithub3/services/base.py +++ b/pygithub3/services/base.py @@ -87,3 +87,30 @@ class Service(object): def _get_result(self, request, **kwargs): return Result(self._client, request, **kwargs) + + +class MimeTypeMixin(object): + + VERSION = 'beta' + + def __set_mimetype(self, mimetype): + self.mimetype = 'application/vnd.github.%s.%s+json' % ( + self.VERSION, mimetype) + + def set_raw_mimetype(self): + self.__set_mimetype('raw') + + def set_text_mimetype(self): + self.__set_mimetype('text') + + def set_html_mimetype(self): + self.__set_mimetype('html') + + def set_full_mimetype(self): + self.__set_mimetype('full') + + def _get_mimetype_as_header(self): + try: + return {'headers': {'Accept': self.mimetype}} + except AttributeError: + return {} -- cgit v1.2.3-59-g8ed1b