def from_egg(cls, egg_path):
zid = compat.ZipFile(egg_path)
try:
data = json.loads(zid.read("EGG-INFO/ipkg.info"))
return cls.from_json_dict(data)
finally:
zid.close()
def create_zarchive(node_pkg, archive_root, archive_node):
zid = compat.ZipFile(archive_node.abspath(), "w", compat.ZIP_DEFLATED)
try:
for filename, alias in node_pkg.iter_source_files():
zid.write(filename, op.join(archive_root, alias))
finally:
zid.close()
def build_egg(build_manifest,
build_node,
source_root,
output_dir=None,
output_file=None):
meta = PackageMetadata.from_build_manifest(build_manifest)
egg_info = EggInfo.from_build_manifest(build_manifest, build_node)
# FIXME: fix egg name
if output_dir is None:
if output_file is None:
egg = egg_filename(os.path.join("dist", meta.fullname))
else:
egg = os.path.join("dist", output_file)
else:
if output_file is None:
egg = egg_filename(os.path.join(output_dir, meta.fullname))
else:
egg = os.path.join(output_dir, output_file)
bento.utils.path.ensure_dir(egg)
egg_scheme = {
"prefix": source_root.abspath(),
"eprefix": source_root.abspath(),
"sitedir": source_root.abspath()
}
zid = compat.ZipFile(egg, "w", compat.ZIP_DEFLATED)
try:
for filename, cnt in egg_info.iter_meta(build_node):
zid.writestr(os.path.join("EGG-INFO", filename), cnt)
for kind, source, target in build_manifest.iter_built_files(
source_root, egg_scheme):
if not kind in ["executables"]:
zid.write(source.abspath(), target.path_from(source_root))
if kind == "pythonfiles":
try:
bytecode = bcompile(source.abspath())
except PyCompileError:
e = extract_exception()
warnings.warn("Error byte-compiling %r" % source.abspath())
else:
zid.writestr("%sc" % target.path_from(source_root),
bcompile(source.abspath()))
finally:
zid.close()
return
def create_wininst(ipkg,
src_root_node,
build_node,
egg_info=None,
wininst=None,
output_dir=None):
meta = PackageMetadata.from_ipkg(ipkg)
if egg_info is None:
egg_info = EggInfo.from_ipkg(ipkg, build_node)
# XXX: do this correctly, maybe use same as distutils ?
if wininst is None:
wininst = os.path.join(output_dir, wininst_filename(meta))
else:
wininst = os.path.join(output_dir, wininst)
bento.utils.path.ensure_dir(wininst)
egg_info_dir = os.path.join("PURELIB", egg_info_dirname(meta.fullname))
fid, arcname = tempfile.mkstemp(prefix="zip")
zid = compat.ZipFile(arcname, "w", compat.ZIP_DEFLATED)
try:
for filename, cnt in egg_info.iter_meta(build_node):
zid.writestr(os.path.join(egg_info_dir, filename), cnt)
wininst_paths = compat.defaultdict(lambda: r"DATA\share\$pkgname")
wininst_paths.update({
"bindir": "SCRIPTS",
"sitedir": "PURELIB",
"gendatadir": "$sitedir"
})
d = {}
for k in ipkg._path_variables:
d[k] = wininst_paths[k]
ipkg.update_paths(d)
file_sections = ipkg.resolve_paths(src_root_node)
def write_content(source, target, kind):
zid.write(source.abspath(), target.abspath())
for kind, source, target in iter_files(file_sections):
write_content(source, target, kind)
finally:
zid.close()
os.close(fid)
create_exe(ipkg, arcname, wininst)
def build_egg(ipkg, ctx, source_root, output_dir=None, output_file=None):
meta = PackageMetadata.from_ipkg(ipkg)
egg_info = EggInfo.from_ipkg(ipkg, ctx.build_node)
# FIXME: fix egg name
if output_dir is None:
if output_file is None:
egg = egg_filename(os.path.join("dist", meta.fullname))
else:
egg = os.path.join("dist", output_file)
else:
if output_file is None:
egg = egg_filename(os.path.join(output_dir, meta.fullname))
else:
egg = os.path.join(output_dir, output_file)
bento.utils.path.ensure_dir(egg)
zid = compat.ZipFile(egg, "w", compat.ZIP_DEFLATED)
try:
ipkg.update_paths({
"prefix": source_root.abspath(),
"eprefix": source_root.abspath(),
"sitedir": source_root.abspath()
})
for filename, cnt in egg_info.iter_meta(ctx.build_node):
zid.writestr(os.path.join("EGG-INFO", filename), cnt)
file_sections = ipkg.resolve_paths(source_root)
for kind, source, target in iter_files(file_sections):
if not kind in ["executables"]:
zid.write(source.abspath(), target.path_from(source_root))
pprint("PINK", "Byte-compiling ...")
for kind, source, target in iter_files(file_sections):
if kind in ["pythonfiles"]:
try:
bytecode = bcompile(source.abspath())
except PyCompileError:
e = extract_exception()
warnings.warn("Error byte-compiling %r" % source.abspath())
else:
zid.writestr("%sc" % target.path_from(source_root),
bcompile(source.abspath()))
finally:
zid.close()
return
def build_wheel(build_manifest,
build_node,
source_root,
output_dir=None,
output_file=None):
meta = PackageMetadata.from_ipkg(build_manifest)
egg_info = WheelInfo.from_ipkg(build_manifest, build_node)
assert not '_' in meta.version
# FIXME: fix egg name
if output_dir is None:
if output_file is None:
egg = wheel_filename(os.path.join("dist", meta.fullname))
else:
egg = os.path.join("dist", output_file)
else:
if output_file is None:
egg = wheel_filename(os.path.join(output_dir, meta.fullname))
else:
egg = os.path.join(output_dir, output_file)
bento.utils.path.ensure_dir(egg)
egg_scheme = {
"prefix": source_root.abspath(),
"eprefix": source_root.abspath(),
"sitedir": source_root.abspath()
}
record = []
zid = compat.ZipFile(egg, "w", compat.ZIP_DEFLATED)
try:
for kind, source, target in build_manifest.iter_built_files(
source_root, egg_scheme):
if not kind in ["executables"]:
abspath = source.abspath()
target_path = target.path_from(source_root).replace(
os.path.sep, '/')
zid.write(abspath, target_path)
hash, length = hash_and_length(abspath)
digest = "sha256=" + urlsafe_b64encode(hash)
record.append((target_path, digest, length))
for filename, cnt in egg_info.iter_meta(build_node):
name = '/'.join((meta.fullname + ".dist-info", filename))
digest = "sha256=" + urlsafe_b64encode(
hashlib.sha256(cnt).digest())
zid.writestr(name, cnt)
record.append((name, digest, len(cnt)))
name = '/'.join((meta.fullname + ".dist-info", "WHEEL"))
wheelfile = b'Wheel-Version: 0.1\nGenerator: bento\nRoot-Is-Purelib: true\n\n'
zid.writestr(name, wheelfile)
digest = "sha256=" + urlsafe_b64encode(
hashlib.sha256(wheelfile).digest())
record.append((name, digest, len(wheelfile)))
name = '/'.join((meta.fullname + ".dist-info", "RECORD"))
sio = cStringIO.StringIO()
writer = csv.writer(sio)
for row in record:
writer.writerow(row)
writer.writerow((name, '', ''))
zid.writestr(name, sio.getvalue())
finally:
zid.close()
return
