cpython/Mac/scripts/PackLibDir.py

90 lines
2.3 KiB
Python

#
# Turn a pyc file into a resource file containing it in 'PYC ' resource form
import addpack
addpack.addpack('Tools')
addpack.addpack('bgen')
addpack.addpack('res')
from Res import *
import Res
from Resources import *
import os
import macfs
import sys
READ = 1
WRITE = 2
smAllScripts = -3
error = 'mkpycresourcefile.error'
def Pstring(str):
if len(str) > 255:
raise ValueError, 'String too large'
return chr(len(str))+str
def createoutput(dst):
"""Create output file. Return handle and first id to use."""
FSpCreateResFile(dst, 'PYTH', 'rsrc', smAllScripts)
output = FSpOpenResFile(dst, WRITE)
UseResFile(output)
num = 128
return output, num
def writemodule(name, id, data):
"""Write pyc code to a PYC resource with given name and id."""
# XXXX Check that it doesn't exist
res = Resource(data)
res.AddResource('PYC ', id, name)
res.WriteResource()
res.ReleaseResource()
def mkpycresourcefile(src, dst):
"""Copy pyc file/dir src to resource file dst."""
if not os.path.isdir(src) and src[-4:] <> '.pyc':
raise error, 'I can only handle .pyc files or directories'
handle, oid = createoutput(dst)
if os.path.isdir(src):
id = handlesubdir(handle, oid, src)
else:
id = handleonepycfile(handle, oid, src)
print 'Wrote',id-oid,'PYC resources to', dst
CloseResFile(handle)
def handleonepycfile(handle, id, file):
"""Copy one pyc file to the open resource file"""
d, name = os.path.split(file)
name = name[:-4]
print ' module', name
writemodule(name, id, open(file, 'rb').read())
return id+1
def handlesubdir(handle, id, srcdir):
"""Recursively scan a directory for pyc files and copy to resources"""
print 'Directory', srcdir
src = os.listdir(srcdir)
for file in src:
file = os.path.join(srcdir, file)
if os.path.isdir(file):
id = handlesubdir(handle, id, file)
elif file[-4:] == '.pyc':
id = handleonepycfile(handle, id, file)
return id
if __name__ == '__main__':
args = sys.argv[1:]
if not args:
ifss, ok = macfs.GetDirectory('Select root of tree to pack:')
if not ok:
sys.exit(0)
args = [ifss.as_pathname()]
for ifn in args:
ofss, ok = macfs.StandardPutFile('Output for '+os.path.split(ifn)[1])
if not ok:
sys.exit(0)
mkpycresourcefile(ifn, ofss.as_pathname())
sys.exit(1) # So we can see something...