Added capability to specify docstring width

Author: themarpe

pybind11_mkdoc/__main__.py

@@ -6,6 +6,7 @@
     mkdoc_out = None
     mkdoc_help = False
     mkdoc_args = []
+    docstring_width = None
 
     i = 1
     while i < len(sys.argv):
@@ -18,6 +19,11 @@
             i += 1 # Skip next
         elif arg.startswith('-o'):
             mkdoc_out = arg[2:]
+        elif arg == '-w':
+            docstring_width = int(sys.argv[i + 1])
+            i += 1 # Skip next
+        elif arg.startswith('-w'):
+            docstring_width = int(arg[2:])
         elif arg == '-I':
             # Concatenate include directive and path
             mkdoc_args.append(arg + sys.argv[i + 1])
@@ -38,10 +44,12 @@
 
   -o <filename>     Write to the specified filename (default: use stdout)
 
+  -w <width>        Specify docstring width before wrapping
+
   -I <path>         Specify an include directory
 
   -Dkey=value       Specify a compiler definition
 
 (Other compiler flags that Clang understands can also be supplied)""")
     else:
-        mkdoc(mkdoc_args, mkdoc_out)
+        mkdoc(mkdoc_args, docstring_width, mkdoc_out)

pybind11_mkdoc/mkdoc_lib.py

@@ -62,6 +62,7 @@
 job_count = cpu_count()
 job_semaphore = Semaphore(job_count)
 errors_detected = False
+docstring_width = int(70)
 
 
 class NoFilenamesError(ValueError):
@@ -165,7 +166,7 @@ def process_comment(comment):
     wrapper.expand_tabs = True
     wrapper.replace_whitespace = True
     wrapper.drop_whitespace = True
-    wrapper.width = 70
+    wrapper.width = docstring_width
     wrapper.initial_indent = wrapper.subsequent_indent = ''
 
     result = ''
@@ -376,7 +377,10 @@ def write_header(comments, out_file=sys.stdout):
 ''', file=out_file)
 
 
-def mkdoc(args, output=None):
+def mkdoc(args, width, output=None):
+    if width != None:
+        global docstring_width
+        docstring_width = int(width)
     comments = extract_all(args)
     if errors_detected:
         return