Merge pull request #53 from dataiku/pbailly/ch45366/improve-wiki-python-public-api

Use the UUID of the page when saving a wiki article. [ch45366]

Author: apichery

dataikuapi/dss/wiki.py

@@ -30,15 +30,15 @@ def get_settings(self):
         """
         return DSSWikiSettings(self.client, self.project_key, self.client._perform_json("GET", "/projects/%s/wiki/" % (self.project_key)))
 
-    def get_article(self, article_id):
+    def get_article(self, article_id_or_name):
         """
         Get a wiki article
 
-        :param str article_id: the article ID
+        :param str article_id_or_name: reference to the article, it can be its ID or its name
         :returns: a handle to manage the Article
         :rtype: :class:`dataikuapi.dss.wiki.DSSWikiArticle`
         """
-        return DSSWikiArticle(self.client, self.project_key, article_id)
+        return DSSWikiArticle(self.client, self.project_key, article_id_or_name)
 
     def __flatten_taxonomy__(self, taxonomy):
         """
@@ -206,11 +206,14 @@ class DSSWikiArticle(object):
     """
     A handle to manage an article
     """
-    def __init__(self, client, project_key, article_id):
+    def __init__(self, client, project_key, article_id_or_name):
         """Do not call directly, use :meth:`dataikuapi.dss.wiki.DSSWiki.get_article`"""
         self.client = client
         self.project_key = project_key
-        self.article_id = article_id
+
+        # Retrieve the real article id
+        article_data = self.client._perform_json("GET", "/projects/%s/wiki/%s" % (project_key, article_id_or_name))
+        self.article_id = article_data["article"]['id']
         # encode in UTF-8 if its python2 and unicode
         if sys.version_info < (3,0) and isinstance(self.article_id, unicode):
             self.article_id = self.article_id.encode('utf-8')
@@ -327,6 +330,6 @@ def set_name(self, name):
 
     def save(self):
         """
-        Save the current article data to the backend
+        Save the current article data to the backend.
         """
         self.article_data = self.client._perform_json("PUT", "/projects/%s/wiki/%s" % (self.project_key, dku_quote_fn(self.article_id)), body=self.article_data)