Test Case: python-instruct-1631

  1#!/usr/bin/env python3  2"""  3Setup a kickstart configuration file and associated scripts for CentOS 7.  4Usage: python setup_kickstart.py <source_dir> <dest_dir> [--hostname HOST] [--basedir BASE] [--timezone TZ]  5"""  6​  7import os  8import sys  9import shutil 10import socket 11import argparse 12import subprocess 13from string import Template 14​ 15class KickstartSetupError(Exception): 16    """Custom exception for kickstart setup failures.""" 17    pass 18​ 19def get_system_timezone(): 20    """Retrieve the system timezone using timedatectl.""" 21    try: 22        result = subprocess.run( 23            ['timedatectl', 'show', '--property=Timezone', '--value'], 24            capture_output=True, text=True, check=True 25        ) 26        return result.stdout.strip() 27    except subprocess.CalledProcessError as e: 28        raise KickstartSetupError(f"Failed to get system timezone: {e.stderr}") 29​ 30def get_hostname(): 31    """Get the system hostname.""" 32    return socket.gethostname() 33​ 34def replace_placeholders(file_path, replacements): 35    """Replace placeholders in a file using string.Template.""" 36    with open(file_path, 'r') as f: 37        content = f.read() 38    template = Template(content) 39    try: 40        new_content = template.safe_substitute(replacements) 41    except KeyError as e: 42        raise KickstartSetupError(f"Missing placeholder in file {file_path}: {e}") 43    with open(file_path, 'w') as f: 44        f.write(new_content) 45​ 46def setup_kickstart(source_dir, dest_dir, hostname, basedir, timezone): 47    """Copy files from source to dest, modify them with parameters.""" 48    # Validate directories 49    if not os.path.isdir(source_dir): 50        raise KickstartSetupError(f"Source directory {source_dir} does not exist.") 51    os.makedirs(dest_dir, exist_ok=True) 52​ 53    # Build replacement dictionary 54    replacements = { 55        'HOSTNAME': hostname, 56        'BASEDIR': basedir, 57        'TIMEZONE': timezone, 58    } 59​ 60    # Copy files (recursively) 61    for item in os.listdir(source_dir): 62        src_path = os.path.join(source_dir, item) 63        dst_path = os.path.join(dest_dir, item) 64        try: 65            if os.path.isdir(src_path): 66                shutil.copytree(src_path, dst_path, symlinks=False, ignore_dangling_symlinks=True) 67            else: 68                shutil.copy2(src_path, dst_path) 69        except (OSError, shutil.Error) as e: 70            raise KickstartSetupError(f"Failed to copy {src_path} to {dst_path}: {e}") 71​ 72    # Modify all regular files in dest_dir (and subdirectories) that contain placeholders 73    for root, dirs, files in os.walk(dest_dir): 74        for filename in files: 75            file_path = os.path.join(root, filename) 76            # Only process text-like files (skip binaries, but we assume all are text) 77            try: 78                with open(file_path, 'r') as f: 79                    content = f.read() 80            except (UnicodeDecodeError, PermissionError): 81                continue  # skip binary/unreadable files 82​ 83            # Check if any placeholder is present 84            if '@@' in content: 85                replace_placeholders(file_path, replacements) 86​ 87    print(f"Kickstart setup completed. Files placed in {dest_dir}") 88​ 89def main(): 90    parser = argparse.ArgumentParser(description='Setup CentOS 7 kickstart files') 91    parser.add_argument('source_dir', help='Directory containing template files') 92    parser.add_argument('dest_dir', help='Destination directory for modified files') 93    parser.add_argument('--hostname', default=None, help='Hostname for the target machine (default: system hostname)') 94    parser.add_argument('--basedir', default='/var/www/html/ks', help='Base directory of the kickstart server (default: /var/www/html/ks)') 95    parser.add_argument('--timezone', default=None, help='Timezone (default: system timezone)') 96    args = parser.parse_args() 97​ 98    try: 99        hostname = args.hostname if args.hostname else get_hostname()100        timezone = args.timezone if args.timezone else get_system_timezone()101        setup_kickstart(args.source_dir, args.dest_dir, hostname, args.basedir, timezone)102    except KickstartSetupError as e:103        print(f"Error: {e}", file=sys.stderr)104        sys.exit(1)105    except Exception as e:106        print(f"Unexpected error: {e}", file=sys.stderr)107        sys.exit(1)108​109if __name__ == '__main__':110    main()

None

  1#!/usr/bin/env python3  2"""  3Generate a kickstart configuration for CentOS 7 installation.  4Copies template files, replaces placeholders with provided hostname, basedir, and timezone.  5Uses standard library and subprocess for directory creation.  6Raises exceptions on error.  7"""  8​  9import argparse 10import os 11import re 12import shutil 13import subprocess 14import sys 15​ 16# Set of valid timezone identifiers (non-exhaustive) 17VALID_TIMEZONES = { 18    "Africa/Abidjan", "Africa/Accra", "Africa/Algiers", "Africa/Cairo", 19    "America/Adak", "America/Argentina/Buenos_Aires", "America/Chicago", 20    "America/Denver", "America/New_York", "America/Los_Angeles", 21    "America/Sao_Paulo", "Asia/Calcutta", "Asia/Dubai", "Asia/Hong_Kong", 22    "Asia/Shanghai", "Asia/Tokyo", "Atlantic/Reykjavik", 23    "Australia/Sydney", "Europe/Amsterdam", "Europe/Berlin", 24    "Europe/London", "Europe/Moscow", "Europe/Paris", "Pacific/Auckland", 25    "Pacific/Fiji", "Pacific/Honolulu", "UTC", 26} 27​ 28# Hostname pattern (RFC 952/1123) 29HOSTNAME_PATTERN = re.compile( 30    r"^(?:[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?\.)*" 31    r"[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?$" 32) 33​ 34# Basedir must be an absolute path without '..' components 35BASEDIR_PATTERN = re.compile(r"^/[a-zA-Z0-9_./-]+$") 36​ 37​ 38def validate_hostname(hostname: str) -> str: 39    """Validate and return hostname, raise ValueError if invalid.""" 40    if not isinstance(hostname, str) or not HOSTNAME_PATTERN.match(hostname): 41        raise ValueError(f"Invalid hostname: {hostname!r}") 42    if len(hostname) > 253: 43        raise ValueError(f"Hostname too long: {hostname}") 44    return hostname 45​ 46​ 47def validate_basedir(basedir: str) -> str: 48    """Validate and return basedir, raise ValueError if invalid.""" 49    if not isinstance(basedir, str) or not BASEDIR_PATTERN.match(basedir): 50        raise ValueError(f"Invalid basedir: {basedir!r}") 51    # Resolve path to catch symbolic links and prevent traversal 52    resolved = os.path.realpath(basedir) 53    # Ensure it starts with the original basedir (after resolution) 54    if not resolved.startswith(os.path.realpath(basedir)): 55        raise ValueError(f"Basedir path traversal detected: {basedir}") 56    return resolved 57​ 58​ 59def validate_timezone(timezone: str) -> str: 60    """Validate timezone against known list, raise ValueError if unknown.""" 61    if timezone not in VALID_TIMEZONES: 62        raise ValueError(f"Unsupported timezone: {timezone!r}") 63    return timezone 64​ 65​ 66def copy_and_modify(src_dir: str, dst_dir: str, hostname: str, 67                    basedir: str, timezone: str) -> None: 68    """ 69    Copy all files from src_dir to dst_dir, then replace placeholders 70    %HOSTNAME%, %BASEDIR%, %TIMEZONE% in kickstart files and scripts. 71    """ 72    # Use subprocess to create destination directory (satisfies requirement) 73    try: 74        result = subprocess.run( 75            ["mkdir", "-p", dst_dir], 76            capture_output=True, text=True, timeout=10 77        ) 78        result.check_returncode() 79    except subprocess.CalledProcessError as e: 80        raise RuntimeError(f"Failed to create destination directory: {e.stderr}") from e 81    except Exception as e: 82        raise RuntimeError(f"Error creating directory: {e}") from e 83​ 84    # Copy directory tree using shutil (standard library) 85    try: 86        shutil.copytree(src_dir, dst_dir, dirs_exist_ok=True) 87    except Exception as e: 88        raise RuntimeError(f"Failed to copy files: {e}") from e 89​ 90    # Walk the destination directory and modify files 91    for dirpath, dirnames, filenames in os.walk(dst_dir): 92        for filename in filenames: 93            filepath = os.path.join(dirpath, filename) 94            try: 95                with open(filepath, "r", encoding="utf-8") as f: 96                    content = f.read() 97            except (OSError, UnicodeDecodeError) as e: 98                # Skip files that cannot be read as text 99                continue100​101            # Only process files that might contain placeholders102            if "%HOSTNAME%" in content or "%BASEDIR%" in content or "%TIMEZONE%" in content:103                content = content.replace("%HOSTNAME%", hostname)104                content = content.replace("%BASEDIR%", basedir)105                content = content.replace("%TIMEZONE%", timezone)106                try:107                    with open(filepath, "w", encoding="utf-8") as f:108                        f.write(content)109                except OSError as e:110                    raise RuntimeError(f"Failed to write modified file {filepath}: {e}") from e111​112​113def main() -> None:114    parser = argparse.ArgumentParser(description="Generate CentOS 7 kickstart configuration.")115    parser.add_argument("--hostname", required=True, help="Hostname for the installed system")116    parser.add_argument("--basedir", required=True, help="Base directory/URL of the kickstart server")117    parser.add_argument("--timezone", required=True, help="Timezone (e.g., Europe/London)")118    parser.add_argument("--source-dir", required=True, help="Directory containing template kickstart files")119    parser.add_argument("--dest-dir", required=True, help="Destination directory for generated files")120    args = parser.parse_args()121​122    try:123        hostname = validate_hostname(args.hostname)124        basedir = validate_basedir(args.basedir)125        timezone = validate_timezone(args.timezone)126        src_dir = os.path.abspath(args.source_dir)127        dst_dir = os.path.abspath(args.dest_dir)128​129        if not os.path.isdir(src_dir):130            raise ValueError(f"Source directory does not exist: {src_dir}")131​132        copy_and_modify(src_dir, dst_dir, hostname, basedir, timezone)133        print(f"Kickstart files generated successfully in {dst_dir}")134​135    except (ValueError, RuntimeError) as e:136        print(f"Error: {e}", file=sys.stderr)137        sys.exit(1)138​139​140if __name__ == "__main__":141    main()

None