|
|
@ -37,7 +37,7 @@ def drop_imports(code): |
|
|
|
def split_line(line): |
|
|
|
# splits an assignment statement into varname and command strings |
|
|
|
# in: "parser = ArgumentParser(description='Example Argparse Program')" |
|
|
|
# out: "parser", "= parser = ArgumentParser(description='Example Argparse Program" |
|
|
|
# out: "parser", "ArgumentParser(description='Example Argparse Program" |
|
|
|
# take/dropwhile used to avoid splitting on multiple '=' signs |
|
|
|
not_equal_sign = lambda x: x != '=' |
|
|
|
varname = ''.join(takewhile(not_equal_sign, line)).strip() |
|
|
|