2 # Adapted from custom_dsstore.py by the Jalview team 2024
3 # Copyright (c) 2013-2017 The Bitcoin Core developers
4 # Distributed under the MIT software license, see the accompanying
5 # file COPYING or http://www.opensource.org/licenses/mit-license.php.
6 from ds_store import DSStore
7 from pprint import pprint
12 parser = argparse.ArgumentParser(
13 prog = "jalview_custom_dsstore.py",
14 description = "Take an existing DS_Store file and change the volume name and add path to background image"
16 parser.add_argument("-i", "--input", help="The existing styled DS_Store file", dest="input")
17 parser.add_argument("-o", "--output", help="The output DS_Store file", dest="output")
18 parser.add_argument("-v", "--volumename", help="The name of the output DS_Store volume", dest="volumename")
19 parser.add_argument("-b", "--background", help="The background image filename to use", dest="background")
20 parser.add_argument("-c", "--config", help="YAML configuration for window and icon sizes and positions", dest="config")
21 parser.add_argument("-d", "--dump", help="Display contents of the input DS_Store to stdout", action="store_true", dest="dump")
22 parser.add_argument("-q", "--quiet", help="Don't print messages to stdout", action="store_true", dest="quiet")
23 parser.add_argument("-f", "--filenames", help="List of comma-separated filenames to show information for --dump", dest="filenames", default="")
25 args = parser.parse_args()
35 if (args.dump and not args.input):
36 exit("When --dump used, an --input must be given")
39 with DSStore.open(args.input, 'r+') as d:
40 for key in ["bwsp", "icvp","Iloc"]:
42 myprint(f"d['.']['{key}']=")
43 mypprint(str(d['.'][key]))
45 myprint(f"No info for d['.']['{key}']")
47 alias = mac_alias.Alias.from_bytes( d['.']['icvp']['backgroundImageAlias'] )
48 myprint("backgroundImageAlias=")
54 mypprint(f"d['{data}']=",d[data])
56 myprint(f"No info for d['{data}']")
57 # for filename in args.filenames.split(","):
59 # for f in d.find(filename):
60 # myprint(f"d['{f}']=", d[f])
62 # myprint(f"# No filename '{filename}' found in DS_Store '{args.input}'")
65 if (args.output and not (args.input or args.config)):
66 exit("Need --input FILENAME or --config FILENAME to produce an --output")
69 exit("Provide --output FILENAME to output DS_Store")
71 if (not args.volumename):
72 exit("Provide a volume name with --volumename NAME")
74 if (not args.background):
75 exit("Provide a background image filename (just the file, no path) with --background FILENAME")
77 package_name_ns = args.volumename
78 configfilename = args.config
79 configfile = open(configfilename, 'r')
81 config = json.load(configfile)
82 inputds = DSStore.open(args.input)
83 outputds = DSStore.open(args.output, 'w+')
88 for key in "ShowStatusBar SidebarWidthTenElevenOrLater ShowToolbar ShowTabView ContainerShowSidebar WindowBounds ShowSidebar ShowPathbar".split():
89 bwsp[key] = inputds['.']['bwsp'][key]
90 myprint(f"Setting bwsp['{key}'] to '"+str(inputds['.']['bwsp'][key])+"'")
92 outputds['.']['bwsp'] = bwsp
96 alias = mac_alias.Alias.from_bytes(inputds['.']['icvp']['backgroundImageAlias'])
97 alias.volume.name = args.volumename
98 alias.volume.posix_path = '/Volumes/' + args.volumename
99 alias.volume.disk_image_alias.target.filename = args.volumename + '.dmg'
100 alias.volume.disk_image_alias.target.carbon_path = 'Macintosh HD:Users:\x00Jalview:\x00' + args.volumename + ':\x00' + args.volumename + '.dmg'
101 alias.volume.disk_image_alias.target.posix_path = '/Users/Jalview/' + args.volumename + '.dmg'
102 alias.target.carbon_path = args.volumename + ':.background:\x00' + args.background
104 icvp['backgroundImageAlias'] = alias.to_bytes()
106 for key in "backgroundColorRed backgroundColorBlue backgroundColorGreen gridSpacing gridOffsetX gridOffsetY showItemInfo viewOptionsVersion arrangeBy textSize labelOnBottom backgroundType showIconPreview iconSize".split():
107 icvp[key] = inputds['.']['icvp'][key]
108 myprint(f"Setting icvp['{key}'] to '"+str(inputds['.']['icvp'][key])+"'")
110 outputds['.']['icvp'] = icvp
112 # copy filenames properties? not working, using JSON config file
113 #for filename in args.filenames.split(","):
114 # outputds[filename]['Iloc'] = inputds[filename]['Iloc']
116 outputds['.']['vSrn'] = ('long', 1)
118 for fileinfo in config['files']:
119 outputds[fileinfo['name']]['Iloc'] = (fileinfo['xpos'], fileinfo['ypos'])