MimeType mixin

1 files changed, 32 insertions, 0 deletions

diff --git a/github3/handlers/base.py b/github3/handlers/base.py
index a85ce20..7bc55d2 100644
--- a/github3/handlers/base.py
+++ b/github3/handlers/base.py
@@ -4,6 +4,38 @@
 from github3.core import Paginate
 from github3.converters import Modelizer
 
+
+class MimeTypeMixin(object):
+
+    VERSION = 'beta'
+
+    def __init__(self):
+        self.mimetypes = set()
+
+    def _parse_mime_type(self, type):
+        return 'application/vnd.github.%s.%s+json' % (
+            self.VERSION, type)
+
+    def add_raw(self):
+        self.mimetypes.add(self._parse_mime_type('raw'))
+        return self
+
+    def add_text(self):
+        self.mimetypes.add(self._parse_mime_type('text'))
+        return self
+
+    def add_html(self):
+        self.mimetypes.add(self._parse_mime_type('html'))
+        return self
+
+    def add_full(self):
+        self.mimetypes.add(self._parse_mime_type('full'))
+        return self
+
+    def mime_header(self):
+        return {'Accept': ', '.join(self.mimetypes)}
+
+
 class Handler(object):
     """ Handler base. Requests to API and modelize responses """