forked from semios/semios-packages
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import json
|
|
import os
|
|
import shutil
|
|
import sys
|
|
|
|
if sys.version_info >= (3, 14):
|
|
_archive_format = "zstdtar"
|
|
_archive_suffix = ".tar.zst"
|
|
else:
|
|
_archive_format = "gztar"
|
|
_archive_suffix = ".tar.gz"
|
|
|
|
|
|
class PkgComposer:
|
|
def __init__(self, pkgfile: str):
|
|
self.pkgfile = pkgfile
|
|
|
|
def setworkdir(self, workdir: str):
|
|
self.workdir = workdir
|
|
shutil.rmtree(workdir, ignore_errors=True)
|
|
os.mkdir(workdir)
|
|
os.mkdir(f"{workdir}/bundle")
|
|
|
|
def write_manifest(self, manifest: dict):
|
|
with open(self.workdir + "/PkgManifest.json", "w") as f:
|
|
json.dump(manifest, f)
|
|
|
|
def write_links(self, links: list[dict]):
|
|
with open(self.workdir + "/_links", "w") as f:
|
|
for link in links:
|
|
json.dump(link, f)
|
|
f.write("\n")
|
|
|
|
def compose(self):
|
|
shutil.make_archive(self.pkgfile, _archive_format, self.workdir)
|
|
shutil.move(self.pkgfile + _archive_suffix, self.pkgfile)
|
|
|
|
def makedir(self, name: str):
|
|
os.mkdir(f"{self.workdir}/bundle/{name}")
|
|
|
|
def makedirs(self, name: str):
|
|
os.makedirs(f"{self.workdir}/bundle/{name}", exist_ok=True)
|
|
|
|
def addfile(self, src: str, dst: str):
|
|
shutil.copy2(src, f"{self.workdir}/bundle/{dst}", follow_symlinks=False)
|
|
|
|
def add_dir(self, src: str, dst: str, merge: bool = False):
|
|
shutil.copytree(src, f"{self.workdir}/bundle/{dst}", dirs_exist_ok=merge)
|