Merge branch 'fgouteroux-add_fork_support'

Gauvain Pocentek · Gauvain Pocentek · commit 37912c1ccd39 · 2016-01-08T21:58:55.000+01:00
diff --git a/gitlab/__init__.py b/gitlab/__init__.py
@@ -451,6 +451,13 @@ def UserProject(self, id=None, **kwargs):
                       DeprecationWarning)
         return UserProject._get_list_or_object(self, id, **kwargs)
 
+    def ProjectFork(self, id=None, **kwargs):
+        """Fork a project for a user.
+
+        id must be a dict.
+        """
+        return ProjectFork._get_list_or_object(self, id, **kwargs)
+
     def _list_projects(self, url, **kwargs):
         r = self._raw_get(url, **kwargs)
         raise_error_from_response(r, GitlabListError)
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -545,6 +545,15 @@ class ProjectEventManager(BaseManager):
     obj_cls = ProjectEvent
 
 
+class ProjectFork(GitlabObject):
+    _url = '/projects/fork/%(project_id)s'
+    canUpdate = False
+    canDelete = False
+    canList = False
+    canGet = False
+    requiredUrlAttrs = ['project_id']
+
+
 class ProjectHook(GitlabObject):
     _url = '/projects/%(project_id)s/hooks'
     requiredUrlAttrs = ['project_id']
@@ -956,6 +965,25 @@ def delete_file(self, path, branch, message, **kwargs):
         r = self.gitlab._raw_delete(url, **kwargs)
         raise_error_from_response(r, GitlabDeleteError)
 
+    def create_fork_relation(self, forked_from_id):
+        """Create a forked from/to relation between existing projects.
+
+        Args:
+            forked_from_id (int): The ID of the project that was forked from
+
+        Raises:
+            GitlabCreateError: Operation failed
+            GitlabConnectionError: Connection to GitLab-server failed
+        """
+        url = "/projects/%s/fork/%s" % (self.id, forked_from_id)
+        r = self.gitlab._raw_post(url)
+        raise_error_from_response(r, GitlabCreateError, 201)
+
+    def delete_fork_relation(self):
+        url = "/projects/%s/fork" % self.id
+        r = self.gitlab._raw_delete(url)
+        raise_error_from_response(r, GitlabDeleteError)
+
 
 class TeamMember(GitlabObject):
     _url = '/user_teams/%(team_id)s/members'
