treat flag options as special

Instead of requiring the user to provide a value for a flag option, use the presence of the option on the command line as a "true" value, just as is done in rst. Signed-off-by: Doug Hellmann <[email protected]>
prometheanfire · Sep 6, 2020 · 69c2d99 · 69c2d99
1 parent a46c224
commit 69c2d99
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/doc/source/cli.rst b/doc/source/cli.rst
@@ -24,9 +24,9 @@ Command Line
 
 .. code-block:: bat
 
-   datatemplate -o multiple-documents:1  doc/source/_templates/sample-multiple.tmpl doc/source/sample-multiple.yaml
+   datatemplate -o multiple-documents  doc/source/_templates/sample-multiple.tmpl doc/source/sample-multiple.yaml
 
 Output
 ======
 
-.. runcmd:: datatemplate -o multiple-documents:1  doc/source/_templates/sample-multiple.tmpl doc/source/sample-multiple.yaml
+.. runcmd:: datatemplate -o multiple-documents  doc/source/_templates/sample-multiple.tmpl doc/source/sample-multiple.yaml
diff --git a/sphinxcontrib/datatemplates/cli.py b/sphinxcontrib/datatemplates/cli.py
@@ -38,11 +38,16 @@ def main():
         with io.open(args.config_file, 'r', encoding='utf-8-sig') as f:
             config_body = f.read()
         exec(config_body, config_globals)
-    # add options
-    config_globals.update({
-        k.replace("-", "_"): v
-        for k, _, v in (s.partition(':') for s in args.option)
-    })
+    # Add options. If there is a colon, use it to separate the option
+    # name from its value. If there is no colon, treat the option as a
+    # flag.
+    for opt in args.option:
+        if ':' in opt:
+            k, _, v = opt.partition(':')
+        else:
+            k = opt
+            v = True
+        config_globals[k.replace('-', '_')] = v
     # add special options
     config_globals.update({
         "source":