implement get_tags API

Author: kmaehashi

README.rst

@@ -51,6 +51,10 @@ Python API provides access to @wiki features.
 
   api = AtWikiAPI(AtWikiURI('http://www65.atwiki.jp/python-client/'))
 
+  # Show list of tags.
+  for page in api.get_tags():
+    print(page['name'])
+
   # Show list of pages.
   for page in api.get_list():
     print(page['name'])

atwiki/core.py

@@ -54,6 +54,25 @@ def get_list(self, tag=None):
       index += 1
       time.sleep(self._sleep)
 
+  def get_tags(self):
+    index = 0
+    while True:
+      count = 0
+      soup = self._request(self._uri.tag('', index))
+      links = soup.find('div', attrs={'class': 'cmd_tag'}).select('a.tag')
+      for link in links:
+        tag_name = link.text
+        tag_weight = 0
+        for clazz in link.attrs['class']:
+          if clazz.startswith('weight'):
+            tag_weight = int(clazz[6:])
+            break
+        count += 1
+        yield {'name': tag_name, 'weight': tag_weight}
+      if count == 0: break
+      index += 1
+      time.sleep(self._sleep)
+
   def get_source(self, page_id, generation=0):
     soup = self._request(self._uri.backup_source(page_id, generation))
     pre = soup.find('pre', attrs={'class': 'cmd_backup'})

atwiki/test/test_core.py

@@ -23,6 +23,12 @@ def test_get_list_tag(self):
     self.assertEqual(results[0]['id'], 18)
     self.assertEqual(results[0]['name'], 'Test_atwiki.test.test_core:AtWikiAPITest:test_get_list_tag')
 
+  def test_get_tags(self):
+    results = list(self._api.get_tags())
+    self.assertEqual(len(results), 1)
+    self.assertEqual(results[0]['name'], 'tag01')
+    self.assertEqual(results[0]['weight'], 3)
+
   def test_get_source(self):
     self.assertEqual(self._api.get_source(14, 0),
                      'テスト1\nテスト2\n\nテスト3\nテスト4\n\n\nテスト5')