diff options
Diffstat (limited to 'lab/extract_code.py')
| -rw-r--r-- | lab/extract_code.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/lab/extract_code.py b/lab/extract_code.py index ec077af7..c5a9df81 100644 --- a/lab/extract_code.py +++ b/lab/extract_code.py @@ -50,7 +50,10 @@ then you can do:: import sys import textwrap -fname, lineno = sys.argv[1:] +if len(sys.argv) == 2: + fname, lineno = sys.argv[1].split(":") +else: + fname, lineno = sys.argv[1:] lineno = int(lineno) with open(fname) as code_file: |