prettyfy

Author: spicquenot

dataikuapi/dss/project.py

@@ -86,7 +86,7 @@ def duplicate(self, target_project_key,
                   export_git_repository=True,
                   export_insights_data=True,
                   remapping={},
-                  target_project_folder_id=None):
+                  target_project_folder=None):
         """
         Duplicate the project
 
@@ -98,6 +98,8 @@ def duplicate(self, target_project_key,
         :param bool export_git_repository:
         :param bool export_insights_data:
         :param dict remapping: dict of connections to be remapped for the new project
+        :param target_project_folder: the project folder where to put the duplicated project
+        :type target_project_folder: A :class:`dataikuapi.dss.projectfolder.DSSProjectFolder
         :returns: A dict containing the original and duplicated project's keys
         :rtype: :class:`ProjectDuplicateResult`
         """
@@ -110,9 +112,10 @@ def duplicate(self, target_project_key,
             "exportSavedModels": export_saved_models,
             "exportGitRepository": export_git_repository,
             "exportInsightsData": export_insights_data,
-            "remapping": remapping,
-            "targetProjectFolderId": target_project_folder_id
+            "remapping": remapping
         }
+        if target_project_folder is not None:
+            obj["targetProjectFolderId"] = target_project_folder.project_folder_id
 
         ref = self.client._perform_json("POST", "/projects/%s/duplicate/" % self.project_key, body = obj)
         return ref