88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
# ICANN API for the Centralized Zones Data Service - developed by acidvegas (https://git.acid.vegas/czds)
|
|
# czds/__main__.py
|
|
|
|
import argparse
|
|
import asyncio
|
|
import getpass
|
|
import logging
|
|
import os
|
|
|
|
from .client import CZDS
|
|
|
|
|
|
async def main():
|
|
'''Entry point for the command line interface'''
|
|
|
|
# Create argument parser
|
|
parser = argparse.ArgumentParser(description='ICANN API for the Centralized Zones Data Service')
|
|
|
|
# Authentication
|
|
parser.add_argument('-u', '--username', default=os.getenv('CZDS_USER'), help='ICANN Username')
|
|
parser.add_argument('-p', '--password', default=os.getenv('CZDS_PASS'), help='ICANN Password')
|
|
parser.add_argument('-o', '--output', default=os.getcwd(), help='Output directory')
|
|
|
|
# Zone download options
|
|
zone_group = parser.add_argument_group('Zone download options')
|
|
zone_group.add_argument('-z', '--zones', action='store_true', help='Download zone files')
|
|
zone_group.add_argument('-c', '--concurrency', type=int, default=3, help='Number of concurrent downloads')
|
|
|
|
# Report options
|
|
report_group = parser.add_argument_group('Report options')
|
|
report_group.add_argument('-r', '--report', action='store_true', help='Download the zone stats report')
|
|
report_group.add_argument('-s', '--scrub', action='store_true', help='Scrub the username from the report')
|
|
report_group.add_argument('-f', '--format', choices=['csv', 'json'], default='csv', help='Report output format')
|
|
|
|
# Parse arguments
|
|
args = parser.parse_args()
|
|
|
|
# Configure logging
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
|
|
# Get username and password
|
|
username = args.username or input('ICANN Username: ')
|
|
password = args.password or getpass.getpass('ICANN Password: ')
|
|
|
|
# Create output directory
|
|
output_directory = os.path.join(args.output, 'zones')
|
|
os.makedirs(output_directory, exist_ok=True)
|
|
|
|
logging.info('Authenticating with ICANN API...')
|
|
|
|
# Create the CZDS client
|
|
async with CZDS(username, password) as client:
|
|
# Download zone stats report if requested
|
|
if args.report:
|
|
logging.info('Fetching zone stats report...')
|
|
try:
|
|
# Create the report directory
|
|
output = os.path.join(output_directory, '.report.csv')
|
|
|
|
# Download the report
|
|
await client.get_report(output, scrub=args.scrub, format=args.format)
|
|
|
|
logging.info(f'Zone stats report saved to {output}')
|
|
|
|
return
|
|
except Exception as e:
|
|
raise Exception(f'Failed to download zone stats report: {e}')
|
|
|
|
# Download zone files if requested
|
|
if args.zones:
|
|
logging.info('Downloading zone files...')
|
|
try:
|
|
# Download the zone files
|
|
await client.download_zones(output_directory, args.concurrency)
|
|
except Exception as e:
|
|
raise Exception(f'Failed to download zone files: {e}')
|
|
|
|
|
|
def cli_entry():
|
|
'''Synchronous entry point for console script'''
|
|
|
|
return asyncio.run(main())
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main()) |