mirror of
https://github.com/paulhagstrom/dsk2po.git
synced 2024-09-12 00:54:38 +00:00
36 lines
985 B
Python
36 lines
985 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# scramble po back into dsk
|
||
|
|
# Chris Torrence, Oct 2020
|
||
|
|
import sys, getopt, re
|
||
|
|
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
|
||
|
|
|
||
|
|
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(35):
|
||
|
|
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())
|