Make 'class' option to wrap directive optional.

sk29110 · Dec 30, 2013 · cc998cc · cc998cc
1 parent 0965e08
commit cc998cc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/wrap.py b/modules/wrap.py
@@ -57,7 +57,9 @@ def run(self):
         element_id = self.options.get('id', None)
         if element_id is not None:
             node['ids'].append(element_id)
-        node['classes'].extend(self.options['class'])
+        cls = self.options.get('class')
+        if cls:
+            node['classes'].extend(cls)
         self.add_name(node)
         self.state.nested_parse(self.content, self.content_offset, node)
         return [node]