Initial commit

examples/create_custom_categories.py

@@ -0,0 +1,115 @@
+#!/usr/bin/env python3
+"""
+Example script demonstrating how to create custom URL files and download
+specific sets of distributions.
+This example shows how to create different categories of Linux distributions.
+"""
+
+import os
+
+from download_isos import ISODownloader
+
+
+def create_lightweight_distros():
+    """Create a URL file for lightweight Linux distributions."""
+    lightweight_urls = [
+        "https://geo.mirror.pkgbuild.com/iso/2025.07.01/archlinux-x86_64.iso",
+        "https://mirrors.gigenet.com/endeavouros/iso/EndeavourOS_Mercury-Neo-2025.03.19.iso",
+        "https://download.manjaro.org/gnome/25.0.4/manjaro-gnome-25.0.4-250623-linux612.iso",
+    ]
+
+    with open("examples/lightweight_distros.txt", "w") as f:
+        for url in lightweight_urls:
+            f.write(url + "\n")
+
+    print("✅ Created lightweight_distros.txt")
+
+
+def create_enterprise_distros():
+    """Create a URL file for enterprise Linux distributions."""
+    enterprise_urls = [
+        "https://repo.almalinux.org/almalinux/10/isos/x86_64/AlmaLinux-10.0-x86_64-dvd.iso",
+        "https://download.rockylinux.org/pub/rocky/10/isos/x86_64/Rocky-10.0-x86_64-dvd1.iso",
+        "https://yum.oracle.com/ISOS/OracleLinux/OL10/u0/x86_64/OracleLinux-R10-U0-x86_64-dvd.iso",
+    ]
+
+    with open("examples/enterprise_distros.txt", "w") as f:
+        for url in enterprise_urls:
+            f.write(url + "\n")
+
+    print("✅ Created enterprise_distros.txt")
+
+
+def download_category(url_file, category_name):
+    """Download ISOs from a specific category URL file."""
+    print(f"\n🐧 Downloading {category_name} distributions...")
+
+    # Read URLs from the category file
+    try:
+        with open(url_file) as f:
+            urls = [line.strip() for line in f if line.strip()]
+    except FileNotFoundError:
+        print(f"❌ Error: {url_file} not found!")
+        return
+
+    if not urls:
+        print(f"❌ No URLs found in {url_file}")
+        return
+
+    print(f"📋 Found {len(urls)} URLs to download")
+
+    # Create output directory based on category
+    output_dir = f"./examples_output/{category_name.lower().replace(' ', '_')}"
+
+    # Create downloader instance
+    downloader = ISODownloader(output_dir=output_dir, max_workers=3, retry_attempts=3)
+
+    # Download the files
+    successful, failed, results = downloader.download_all(urls)
+
+    print(f"\n✅ Successfully downloaded: {successful} files")
+    if failed > 0:
+        print(f"❌ Failed downloads: {failed}")
+        for result in results:
+            print(f"  - {result}")
+
+    print(f"\n📂 Files saved to: {os.path.abspath(output_dir)}")
+
+
+def main():
+    """Main function to demonstrate custom URL file creation and usage."""
+    print("🔧 Creating custom URL files for different distribution categories...")
+
+    # Create the examples directory if it doesn't exist
+    os.makedirs("examples", exist_ok=True)
+
+    # Create category-specific URL files
+    create_lightweight_distros()
+    create_enterprise_distros()
+
+    print("\n📋 Available categories:")
+    print("  1. Lightweight distributions (Arch-based)")
+    print("  2. Enterprise distributions (RHEL-based)")
+
+    # Example: Download lightweight distributions
+    print("\n" + "=" * 50)
+    download_category("examples/lightweight_distros.txt", "Lightweight")
+
+    print("\n" + "=" * 50)
+    print("💡 To download enterprise distributions, run:")
+    print("   python examples/create_custom_categories.py --enterprise")
+
+
+if __name__ == "__main__":
+    import sys
+
+    if "--enterprise" in sys.argv:
+        download_category("examples/enterprise_distros.txt", "Enterprise")
+    elif "--help" in sys.argv or "-h" in sys.argv:
+        print("Usage:")
+        print("  python examples/create_custom_categories.py")
+        print("      # Create files and download lightweight")
+        print("  python examples/create_custom_categories.py --enterprise")
+        print("      # Download enterprise distributions")
+    else:
+        main()