43 lines
1.2 KiB
Python
Executable File
43 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# scramble po back into dsk
|
|
# Chris Torrence, Oct 2020
|
|
import sys, getopt, re, os
|
|
from dsk2po import dsk2po
|
|
def main(argv=None):
|
|
print("po2dsk - convert po files to dsk files")
|
|
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], '')
|
|
except getopt.GetoptError as err:
|
|
print(str(err))
|
|
usage()
|
|
return 1
|
|
try:
|
|
filenameIn = args[0]
|
|
except:
|
|
print('You need to provide the name of a PO file to begin.')
|
|
return 1
|
|
|
|
# Handle arbitrary number of tracks (normally should be 35)
|
|
fileSize = os.path.getsize(filenameIn)
|
|
ntracks = fileSize // 4096
|
|
if ntracks != 35:
|
|
print("Warning: PO file has non-standard {} tracks".format(ntracks))
|
|
|
|
tracks = []
|
|
|
|
# Note that the same algorithm can be used to convert in either direction
|
|
with open(filenameIn, mode="rb") as fileIn:
|
|
for track in range(ntracks):
|
|
trackbuffer = fileIn.read(4096)
|
|
tracks.append(dsk2po(trackbuffer))
|
|
dskfilename = re.sub('\.po$', '', filenameIn, flags=re.IGNORECASE) + ".dsk"
|
|
print('Writing dsk image to {}'.format(dskfilename))
|
|
with open(dskfilename, mode="wb") as file:
|
|
for track in tracks:
|
|
file.write(track)
|
|
return 1
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|