From d50ee6639ccdc5c9dcadb61247c3ea12c0149baa Mon Sep 17 00:00:00 2001 From: Rob McMullen Date: Mon, 1 May 2017 08:41:23 -0700 Subject: [PATCH] Added slice support for binary file writing --- atrcopy/__init__.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/atrcopy/__init__.py b/atrcopy/__init__.py index 05c0f4d..a41da22 100644 --- a/atrcopy/__init__.py +++ b/atrcopy/__init__.py @@ -164,8 +164,24 @@ def assemble(image, source_files, data_files): raise AtrError("Data files must include a load address specified with the @ char") name, addr = name.rsplit("@", 1) first = text_to_int(addr) + subset = slice(0, -1) + if "[" in name and "]" in name: + name, slicetext = name.rsplit("[", 1) + if ":" in slicetext: + start, end = slicetext.split(":", 1) + try: + start = int(start) + except: + start = 0 + if end.endswith("]"): + end = end[:-1] + try: + end = int(end) + except: + end = None + subset = slice(start, end) with open(name, 'rb') as fh: - data = fh.read() + data = fh.read()[subset] s = segments.add_segment(data, first) print s.name if options.verbose: