Allow creation of objects using the "hidden API"

Gauvain Pocentek · Gauvain Pocentek · commit 1d55e67b7335 · 2013-05-18T14:17:20.000+02:00
diff --git a/gitlab.py b/gitlab.py
@@ -374,6 +374,11 @@ def _getListOrObject(self, cls, id, **kwargs):
                 raise GitlabGetError
 
             return cls.list(self.gitlab, **kwargs)
+        elif isinstance(id, dict):
+            if not cls.canCreate:
+                raise GitlabCreateError
+
+            return cls(self.gitlab, id, **kwargs)
         else:
             if not cls.canGet:
                 raise GitlabGetError
@@ -536,28 +541,28 @@ class ProjectCommit(GitlabObject):
 class ProjectKey(GitlabObject):
     _url = '/projects/%(project_id)d/keys'
     canUpdate = False
-    requiredCreateAttrs = ['title', 'key']
+    requiredCreateAttrs = ['project_id', 'title', 'key']
 
 
 class ProjectHook(GitlabObject):
     _url = '/projects/%(project_id)d/hooks'
-    requiredCreateAttrs = ['url']
+    requiredCreateAttrs = ['project_id', 'url']
 
 
 class ProjectIssueNote(GitlabObject):
     _url = '/projects/%(project_id)d/issues/%(issue_id)d/notes'
     _constructorTypes = {'author': 'User'}
     canUpdate = False
     canDelete = False
-    requiredCreateAttrs = ['body']
+    requiredCreateAttrs = ['project_id', 'body']
 
 
 class ProjectIssue(GitlabObject):
     _url = '/projects/%(project_id)s/issues/'
     _constructorTypes = {'author': 'User', 'assignee': 'User',
                          'milestone': 'ProjectMilestone'}
     canDelete = False
-    requiredCreateAttrs = ['title']
+    requiredCreateAttrs = ['project_id', 'title']
     optionalCreateAttrs = ['description', 'assignee_id', 'milestone_id',
                            'labels']
 
@@ -571,15 +576,15 @@ def Note(self, id=None, **kwargs):
 class ProjectMember(GitlabObject):
     _url = '/projects/%(project_id)d/members'
     _returnClass = User
-    requiredCreateAttrs = ['user_id', 'access_level']
+    requiredCreateAttrs = ['project_id', 'user_id', 'access_level']
 
 
 class ProjectNote(GitlabObject):
     _url = '/projects/%(project_id)d/notes'
     _constructorTypes = {'author': 'User'}
     canUpdate = False
     canDelete = False
-    requiredCreateAttrs = ['body']
+    requiredCreateAttrs = ['project_id', 'body']
 
 
 class ProjectTag(GitlabObject):
@@ -603,7 +608,7 @@ class ProjectMergeRequest(GitlabObject):
     _url = '/projects/%(project_id)d/merge_request'
     _constructorTypes = {'author': 'User', 'assignee': 'User'}
     canDelete = False
-    requiredCreateAttrs = ['source_branch', 'target_branch', 'title']
+    requiredCreateAttrs = ['project_id', 'source_branch', 'target_branch', 'title']
     optionalCreateAttrs = ['assignee_id']
 
     def Note(self, id=None, **kwargs):
@@ -616,7 +621,7 @@ def Note(self, id=None, **kwargs):
 class ProjectMilestone(GitlabObject):
     _url = '/projects/%(project_id)s/milestones'
     canDelete = False
-    requiredCreateAttrs = ['title']
+    requiredCreateAttrs = ['project_id', 'title']
     optionalCreateAttrs = ['description', 'due_date']
 
 
@@ -625,13 +630,13 @@ class ProjectSnippetNote(GitlabObject):
     _constructorTypes = {'author': 'User'}
     canUpdate = False
     canDelete = False
-    requiredCreateAttrs = ['body']
+    requiredCreateAttrs = ['project_id', 'snippet_id', 'body']
 
 
 class ProjectSnippet(GitlabObject):
     _url = '/projects/%(project_id)d/snippets'
     _constructorTypes = {'author': 'User'}
-    requiredCreateAttrs = ['title', 'file_name', 'code']
+    requiredCreateAttrs = ['project_id', 'title', 'file_name', 'code']
     optionalCreateAttrs = ['lifetime']
 
     def Note(self, id=None, **kwargs):
