provide a ProjectSnippet.Content() method

Gauvain Pocentek · Gauvain Pocentek · commit 72e097d8b2d3 · 2013-05-18T16:36:44.000+02:00
diff --git a/gitlab.py b/gitlab.py
@@ -129,6 +129,19 @@ def setCredentials(self, email, password):
         self.email = email
         self.password = password
 
+    def rawGet(self, path, with_token=False):
+        url = '%s%s' % (self._url, path)
+        if with_token:
+            url += "?private_token=%s" % self.private_token
+
+        try:
+            r = requests.get(url)
+        except:
+            raise GitlabConnectionError(
+                "Can't connect to GitLab server (%s)" % self._url)
+
+        return r
+
     def rawPost(self, path, data):
         url = '%s%s' % (self._url, path)
         try:
@@ -658,6 +671,16 @@ class ProjectSnippet(GitlabObject):
     requiredCreateAttrs = ['project_id', 'title', 'file_name', 'code']
     optionalCreateAttrs = ['lifetime']
 
+    def Content(self):
+        url = "/projects/%(project_id)s/snippets/%(snippet_id)s/raw" % \
+            {'project_id': self.project_id, 'snippet_id': self.id}
+        r = self.gitlab.rawGet(url, True)
+
+        if r.status_code == 200:
+            return r.content
+        else:
+            raise GitlabGetError
+
     def Note(self, id=None, **kwargs):
         return self._getListOrObject(ProjectSnippetNote, id,
                                      project_id=self.project_id,
