diff --git a/dsk2po.py b/dsk2po.py index ea0ce34..6238add 100755 --- a/dsk2po.py +++ b/dsk2po.py @@ -1,8 +1,7 @@ #!/usr/bin/env python3 # unscramble dsk into po # Paul Hagstrom, Dec 2015 -import sys -import getopt +import sys, getopt, re def main(argv=None): print("dsk2po - convert dsk files to po files") @@ -23,7 +22,7 @@ def main(argv=None): for track in range(35): trackbuffer = dskfile.read(4096) potracks.append(dsk2po(trackbuffer)) - pofilename = dskfilename + '.po' + pofilename = re.sub('\.dsk$', '', dskfilename, flags=re.IGNORECASE) + ".po" print('Writing po image to {}'.format(pofilename)) with open(pofilename, mode="wb") as pofile: for potrack in potracks: @@ -54,4 +53,4 @@ def dsk2po(trackbuffer): return potrack if __name__ == "__main__": - sys.exit(main()) \ No newline at end of file + sys.exit(main())