From 321484b845e18471950585302737f9a1a9119c40 Mon Sep 17 00:00:00 2001 From: Daniel Thompson Date: Mon, 9 Mar 2020 21:34:01 +0000 Subject: [PATCH] tools: wasptool: Sneaky hack to avoid copying docstrings to the device --- tools/wasptool | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/tools/wasptool b/tools/wasptool index 099e802..72e88bc 100755 --- a/tools/wasptool +++ b/tools/wasptool @@ -27,16 +27,29 @@ def unsync(c): c.sendline('wasp.run()') def paste(c, f, verbose=False): + docstring = False for ln in f.readlines(): ln = ln.rstrip() - if not ln.lstrip().startswith('#'): - c.sendline(ln) - c.expect('=== ') + + # This is a bit loose (definitely not PEP-257 compliant) but + # is enough for most code. + if ln.lstrip().startswith('"""'): + docstring = True + if docstring: + if ln.rstrip().endswith('"""'): + docstring = False + continue + + if ln.lstrip().startswith('#'): + continue + + + c.sendline(ln) + c.expect('=== ') if not verbose: print('.', end='', flush=True) - def handle_eval(c, cmd): verbose = bool(c.logfile)