116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
#!/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()
|