docker-assemble 0.1.0__py3-none-any.whl

docker_assemble/docker_utils.py

@@ -0,0 +1,5 @@
+import docker
+
+def list_docker_images():
+    client = docker.from_env()
+    return client.images.list()

docker_assemble/image_exporter.py

@@ -0,0 +1,68 @@
+import docker
+import tarfile
+import tempfile
+import os
+import shutil
+from pathlib import Path
+import logging
+
+def extract_image(image_name: str, output_dir: str):
+    client = docker.from_env()
+
+    try:
+        image = client.images.get(image_name)
+        logging.info(f"Image '{image_name}' found locally.")
+    except docker.errors.ImageNotFound:
+        logging.info(f"Image '{image_name}' not found locally. Pulling...")
+        image = client.images.pull(image_name)
+
+    container = client.containers.run(image=image_name, command="sleep infinity", detach=True)
+    logging.debug(f"Created temporary container: {container.id[:12]}")
+
+    try:
+        stream, _ = container.get_archive("/")
+        tmp_tar_path = tempfile.mktemp(suffix=".tar")
+        with open(tmp_tar_path, "wb") as f:
+            for chunk in stream:
+                f.write(chunk)
+
+        logging.debug(f"Filesystem archive saved to: {tmp_tar_path}")
+
+        output_path = Path(output_dir).resolve()
+        output_path.mkdir(parents=True, exist_ok=True)
+
+        extract_tar_safely(tmp_tar_path, output_path)
+
+        logging.info(f"Image filesystem extracted to: {output_path}")
+
+    finally:
+        container.remove(force=True)
+        if os.path.exists(tmp_tar_path):
+            os.remove(tmp_tar_path)
+        logging.debug("Cleaned up temporary container and tar file.")
+
+
+def extract_tar_safely(tar_path: str, output_path: Path):
+    # def is_safe_path(base: Path, target: Path) -> bool:
+    #     try:
+    #         return target.resolve().is_relative_to(base.resolve())
+    #     except AttributeError:
+    #         # For Python < 3.9 fallback
+    #         return str(target.resolve()).startswith(str(base.resolve()))
+
+    with tarfile.open(tar_path, "r") as tar:
+        for member in tar.getmembers():
+            member.name = member.name.lstrip("/")
+            member_path = output_path / member.name
+
+            # if not is_safe_path(output_path, member_path):
+            #     logging.warning(f"Blocked unsafe path: {member.name}, output_path: {output_path}, member_path: {member_path}")
+            #     continue
+
+            tar.extract(member, path=output_path)
+            logging.debug(f"Extracted: {member.name}")
+
+    logging.info(f"Extraction completed to: {output_path}")
+
+
+

docker_assemble/main.py

@@ -0,0 +1,18 @@
+import argparse
+import logging
+import os
+from docker_assemble.image_exporter import extract_image
+
+def run():
+    parser = argparse.ArgumentParser(description="Docker Assemble CLI")
+    parser.add_argument("-d", action="store_true", help="Disassemble an image")
+    parser.add_argument("--debug", action="store_true", help="Enable debug mode")
+    parser.add_argument("image", help="Docker image name")
+    parser.add_argument("output_dir", nargs="?", default=".", help="Optional output directory")
+
+    args = parser.parse_args()
+
+    logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO)
+
+    logging.debug(f"Extracting image: {args.image} to directory: {args.output_dir}")
+    extract_image(image_name=args.image, output_dir=args.output_dir)

docker_assemble-0.1.0.dist-info/METADATA

@@ -0,0 +1,25 @@
+Metadata-Version: 2.4
+Name: docker-assemble
+Version: 0.1.0
+Summary: A CLI tool to extract and analyze Docker images
+Author: Sina Kashipazha
+License: Apache-2.0
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: docker>=6.0.0
+
+# docker-assemble
+
+A CLI tool to extract and analyze Docker images for research & optimization purposes.
+
+## Installation
+
+```bash
+pip install docker-assemble
+```
+
+## Usage
+
+```bash
+docker-assemble -d ubuntu:20.04 output_dir
+```

docker_assemble-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+docker_assemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+docker_assemble/docker_utils.py,sha256=kxEFEGUR1qO1bWwdJzLFRhwF6ejr1VWt2FflzwtCaAY,104
+docker_assemble/image_exporter.py,sha256=vCylMBJS5FiVIfX7djxmo_-cCKB7dO4pTITUn2jaNQE,2288
+docker_assemble/main.py,sha256=YN8hIKX59NIZWXQchZbGgpex2bvb8_ECaarImUmDeZw,764
+docker_assemble-0.1.0.dist-info/METADATA,sha256=HqThiGS74UHvoqR35bFE2egDIAd7bnxiiUBFLRfTVZM,482
+docker_assemble-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+docker_assemble-0.1.0.dist-info/entry_points.txt,sha256=eCHj2yf-IrJI3KlkWGYFlZxA5gwS4KP2nynEKKPiTq4,61
+docker_assemble-0.1.0.dist-info/top_level.txt,sha256=SBZ5oqc6t1oj0QSyM7fDrXarkFbEdAPOMVaF9B0fAew,16
+docker_assemble-0.1.0.dist-info/RECORD,,

docker_assemble-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

docker_assemble-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+docker-assemble = docker_assemble.main:run

docker_assemble-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+docker_assemble