Initial Wiki Pages functionality

gak · gak · commit 3c6c4cdec3f9 · 2014-02-07T14:35:25.000+11:00
diff --git a/AUTHORS.md b/AUTHORS.md
@@ -6,3 +6,4 @@ Authors
 - [Simon Johnson](https://github.com/cybersimon)
 - [FredericBlain](https://github.com/FredericBlain)
 - [Greg Turner](https://github.com/cogat)
+- [Gerald Kaszuba](https://github.com/gak)
diff --git a/assembla/api.py b/assembla/api.py
@@ -222,7 +222,8 @@ def _delete_json(self, instance, space=None, rel_path=None, extra_params=None):
                 )
             )
 
-    def _put_json(self, instance, space=None, rel_path=None, extra_params=None):
+    def _put_json(self, instance, space=None, rel_path=None, extra_params=None,
+                  id_field='number'):
         """
         Base level method for adding new data to the API
         """
@@ -245,7 +246,7 @@ def _put_json(self, instance, space=None, rel_path=None, extra_params=None):
             settings.API_ROOT_PATH,
             settings.API_VERSION,
             rel_path or model.rel_path,
-            instance['number'],
+            instance[id_field],
             urllib.urlencode(extra_params),
         )
 
@@ -345,6 +346,18 @@ def users(self, extra_params=None):
             space=self,
             rel_path=self._build_rel_path('users'),
             extra_params=extra_params,
+            )
+
+    @assembla_filter
+    def wiki_pages(self, extra_params=None):
+        """
+        All Wiki Pages with access to this Space
+        """
+        return self.api._get_json(
+            WikiPage,
+            space=self,
+            rel_path=self._build_rel_path('wiki_pages'),
+            extra_params=extra_params,
         )
 
     def _build_rel_path(self, to_append=None):
@@ -459,3 +472,29 @@ def tickets(self, extra_params=None):
                 space.tickets(extra_params=extra_params)
             )
         return tickets
+
+
+class WikiPage(AssemblaObject):
+    def write(self):
+        """
+        Create or update a Wiki Page on Assembla
+        """
+        if not hasattr(self, 'space'):
+            raise AttributeError("A ticket must have a 'space' attribute before you can write it to Assembla.")
+
+        self.api = self.space.api
+
+        if self.get('id'):  # We are modifying an existing wiki page
+            return self.api._put_json(
+                self,
+                space=self.space,
+                rel_path=self.space._build_rel_path('wiki_pages'),
+                id_field='id'
+            )
+        else:  # Creating a new wiki page
+            return self.api._post_json(
+                self,
+                space=self.space,
+                rel_path=self.space._build_rel_path('wiki_pages'),
+            )
+
