hiro/python-docstring-mode
1
0
Fork 0
mirror of https://github.com/vale981/python-docstring-mode synced 2025-03-04 17:11:41 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

add linewise option; prototypical support for vim

Browse source
This commit is contained in:
Glyph 2015-09-07 10:09:13 -04:00
parent 27badf76ca
commit 86757d0b7b
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
docstring_wrap.py
View file

@ -534,6 +534,7 @@ if __name__ == '__main__':
parser.add_argument("--offset")
parser.add_argument("--indent")
parser.add_argument("--width")
parser.add_argument("--linewise", action='store_true')
namespace = parser.parse_args(sys.argv[1:])
from io import StringIO
@ -541,6 +542,8 @@ if __name__ == '__main__':
io = StringIO()
indata = sys.stdin.read().decode("utf-8")
inlines = indata.split("\n")
if namespace.linewise:
inlines.insert(0, "")
initialBlank, indentCount = indentHeuristic(inlines, io)
point = 0
width = 79
@ -564,5 +567,7 @@ if __name__ == '__main__':
sys.stdout.write(" ")
output = io.getvalue()
if namespace.linewise:
output = "\n".join(output.split("\n")[1:-1])
sys.stdout.write(output.encode("utf-8"))
sys.stdout.flush()