Merge pull request #148 from getnikola/ipynb-tags

Add ipynb posts support to tags plugin
getnikola · Jun 3, 2016 · f205c97 · f205c97
2 parents 172af8d + 84b82b4
commit f205c97
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 6 deletions.
diff --git a/v6/tags/tags.plugin b/v6/tags/tags.plugin
@@ -5,6 +5,6 @@ Tests = test_command_tags
 
 [Documentation]
 Author = Puneeth Chaganti
-Version = 0.6
+Version = 0.7
 Website = http://plugins.getnikola.com/#tags
 Description = Manage the tags for your site.
diff --git a/v6/tags/tags.py b/v6/tags/tags.py
@@ -27,13 +27,17 @@
 from __future__ import unicode_literals, print_function
 import codecs
 from collections import defaultdict
+import io
 import math
 from os.path import relpath
 import re
 from textwrap import dedent
 
 from nikola.plugin_categories import Command
-from nikola.utils import bytes_str, LOGGER, sys_decode, unicode_str
+from nikola.utils import bytes_str, LOGGER, req_missing, sys_decode, unicode_str
+from nikola.plugins.compile.ipynb import flag as ipy_flag
+if ipy_flag:
+    from nikola.plugins.compile.ipynb import current_nbformat, ipy_modern, nbformat
 
 
 def add_tags(site, tags, filepaths, dry_run=False):
@@ -69,7 +73,7 @@ def add_tags(site, tags, filepaths, dry_run=False):
             )
 
         elif new_tags != old_tags:
-            _replace_tags_line(post, new_tags)
+            _replace_tags(site, post, new_tags)
 
     return all_new_tags
 
@@ -134,7 +138,7 @@ def merge_tags(site, tags, filepaths, dry_run=False):
             )
 
         elif new_tags != old_tags:
-            _replace_tags_line(post, new_tags)
+            _replace_tags(site, post, new_tags)
 
     return all_new_tags
 
@@ -175,7 +179,7 @@ def remove_tags(site, tags, filepaths, dry_run=False):
             )
 
         elif new_tags != old_tags:
-            _replace_tags_line(post, new_tags)
+            _replace_tags(site, post, new_tags)
 
     return all_new_tags
 
@@ -236,7 +240,7 @@ def sort_tags(site, filepaths, dry_run=False):
             )
 
         elif new_tags != old_tags:
-            _replace_tags_line(post, new_tags)
+            _replace_tags(site, post, new_tags)
 
     return all_new_tags
 
@@ -657,6 +661,32 @@ def _remove_tags(tags, removals):
     return tags
 
 
+def _replace_tags(site, post, tags):
+    """Chooses the appropriate replace method based on post type."""
+    compiler = site.get_compiler(post.source_path)
+    if compiler.name == 'ipynb':
+        _replace_ipynb_tags(post, tags)
+
+    else:
+        _replace_tags_line(post, tags)
+
+
+def _replace_ipynb_tags(post, tags):
+    """Replaces tags in the notebook metadata with the given tags."""
+    if ipy_flag is None:
+        req_missing(['ipython[notebook]>=2.0.0'], 'build this site (compile ipynb)')
+
+    nb = nbformat.read(post.source_path, current_nbformat)
+    metadata = nb['metadata']['nikola']
+    metadata['tags'] = ','.join(tags)
+
+    with io.open(post.source_path, "w+", encoding="utf8") as fd:
+        if ipy_modern:
+            nbformat.write(nb, fd, 4)
+        else:
+            nbformat.write(nb, fd, 'ipynb')
+
+
 def _replace_tags_line(post, tags):
     """ Replaces the line that lists the tags, with given tags. """