#!/usr/bin/python # -*- coding: utf-8 -*- # Copyright: (c) 2020, Ansible Project # Copyright: (c) 2020, VMware, Inc. All Rights Reserved. # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) from __future__ import absolute_import, division, print_function __metaclass__ = type DOCUMENTATION = r''' --- module: iso_create short_description: Generate ISO file with specified files or folders description: - This module is used to generate ISO file with specified path of files. author: - Diane Wang (@Tomorrow9) <dianew@vmware.com> requirements: - "pycdlib" - "python >= 2.7" version_added: '0.2.0' options: src_files: description: - This is a list of absolute paths of source files or folders which will be contained in the new generated ISO file. - Will fail if specified file or folder in C(src_files) does not exist on local machine. - 'Note: With all ISO9660 levels from 1 to 3, all file names are restricted to uppercase letters, numbers and underscores (_). File names are limited to 31 characters, directory nesting is limited to 8 levels, and path names are limited to 255 characters.' type: list required: yes elements: path dest_iso: description: - The absolute path with file name of the new generated ISO file on local machine. - Will create intermediate folders when they does not exist. type: path required: yes interchange_level: description: - The ISO9660 interchange level to use, it dictates the rules on the names of files. - Levels and valid values C(1), C(2), C(3), C(4) are supported. - The default value is level C(1), which is the most conservative, level C(3) is recommended. - ISO9660 file names at interchange level C(1) cannot have more than 8 characters or 3 characters in the extension. type: int default: 1 choices: [1, 2, 3, 4] vol_ident: description: - The volume identification string to use on the new generated ISO image. type: str rock_ridge: description: - Whether to make this ISO have the Rock Ridge extensions or not. - Valid values are C(1.09), C(1.10) or C(1.12), means adding the specified Rock Ridge version to the ISO. - If unsure, set C(1.09) to ensure maximum compatibility. - If not specified, then not add Rock Ridge extension to the ISO. type: str choices: ['1.09', '1.10', '1.12'] joliet: description: - Support levels and valid values are C(1), C(2), or C(3). - Level C(3) is by far the most common. - If not specified, then no Joliet support is added. type: int choices: [1, 2, 3] udf: description: - Whether to add UDF support to this ISO. - If set to C(True), then version 2.60 of the UDF spec is used. - If not specified or set to C(False), then no UDF support is added. type: bool default: False ''' EXAMPLES = r''' - name: Create an ISO file community.general.iso_create: src_files: - /root/testfile.yml - /root/testfolder dest_iso: /tmp/test.iso interchange_level: 3 - name: Create an ISO file with Rock Ridge extension community.general.iso_create: src_files: - /root/testfile.yml - /root/testfolder dest_iso: /tmp/test.iso rock_ridge: 1.09 - name: Create an ISO file with Joliet support community.general.iso_create: src_files: - ./windows_config/Autounattend.xml dest_iso: ./test.iso interchange_level: 3 joliet: 3 vol_ident: WIN_AUTOINSTALL ''' RETURN = r''' source_file: description: Configured source files or directories list. returned: on success type: list elements: path sample: ["/path/to/file.txt", "/path/to/folder"] created_iso: description: Created iso file path. returned: on success type: str sample: "/path/to/test.iso" interchange_level: description: Configured interchange level. returned: on success type: int sample: 3 vol_ident: description: Configured volume identification string. returned: on success type: str sample: "OEMDRV" joliet: description: Configured Joliet support level. returned: on success type: int sample: 3 rock_ridge: description: Configured Rock Ridge version. returned: on success type: str sample: "1.09" udf: description: Configured UDF support. returned: on success type: bool sample: False ''' import os import traceback PYCDLIB_IMP_ERR = None try: import pycdlib HAS_PYCDLIB = True except ImportError: PYCDLIB_IMP_ERR = traceback.format_exc() HAS_PYCDLIB = False from ansible.module_utils.basic import AnsibleModule, missing_required_lib from ansible.module_utils._text import to_native def add_file(module, iso_file=None, src_file=None, file_path=None, rock_ridge=None, use_joliet=None, use_udf=None): rr_name = None joliet_path = None udf_path = None # In standard ISO interchange level 1, file names have a maximum of 8 characters, followed by a required dot, # followed by a maximum 3 character extension, followed by a semicolon and a version file_name = os.path.basename(file_path) if '.' not in file_name: file_in_iso_path = file_path.upper() + '.;1' else: file_in_iso_path = file_path.upper() + ';1' if rock_ridge: rr_name = file_name if use_joliet: joliet_path = file_path if use_udf: udf_path = file_path try: iso_file.add_file(src_file, iso_path=file_in_iso_path, rr_name=rr_name, joliet_path=joliet_path, udf_path=udf_path) except Exception as err: module.fail_json(msg="Failed to add file %s to ISO file due to %s" % (src_file, to_native(err))) def add_directory(module, iso_file=None, dir_path=None, rock_ridge=None, use_joliet=None, use_udf=None): rr_name = None joliet_path = None udf_path = None iso_dir_path = dir_path.upper() if rock_ridge: rr_name = os.path.basename(dir_path) if use_joliet: joliet_path = iso_dir_path if use_udf: udf_path = iso_dir_path try: iso_file.add_directory(iso_path=iso_dir_path, rr_name=rr_name, joliet_path=joliet_path, udf_path=udf_path) except Exception as err: module.fail_json(msg="Failed to directory %s to ISO file due to %s" % (dir_path, to_native(err))) def main(): argument_spec = dict( src_files=dict(type='list', required=True, elements='path'), dest_iso=dict(type='path', required=True), interchange_level=dict(type='int', choices=[1, 2, 3, 4], default=1), vol_ident=dict(type='str'), rock_ridge=dict(type='str', choices=['1.09', '1.10', '1.12']), joliet=dict(type='int', choices=[1, 2, 3]), udf=dict(type='bool', default=False), ) module = AnsibleModule( argument_spec=argument_spec, supports_check_mode=True, ) if not HAS_PYCDLIB: module.fail_json(missing_required_lib('pycdlib'), exception=PYCDLIB_IMP_ERR) src_file_list = module.params.get('src_files') if src_file_list and len(src_file_list) == 0: module.fail_json(msg='Please specify source file and/or directory list using src_files parameter.') for src_file in src_file_list: if not os.path.exists(src_file): module.fail_json(msg="Specified source file/directory path does not exist on local machine, %s" % src_file) dest_iso = module.params.get('dest_iso') if dest_iso and len(dest_iso) == 0: module.fail_json(msg='Please specify the absolute path of the new created ISO file using dest_iso parameter.') dest_iso_dir = os.path.dirname(dest_iso) if dest_iso_dir and not os.path.exists(dest_iso_dir): # will create intermediate dir for new ISO file try: os.makedirs(dest_iso_dir) except OSError as err: module.fail_json(msg='Exception caught when creating folder %s, with error %s' % (dest_iso_dir, to_native(err))) volume_id = module.params.get('vol_ident') if volume_id is None: volume_id = '' inter_level = module.params.get('interchange_level') rock_ridge = module.params.get('rock_ridge') use_joliet = module.params.get('joliet') use_udf = None if module.params['udf']: use_udf = '2.60' result = dict( changed=False, source_file=src_file_list, created_iso=dest_iso, interchange_level=inter_level, vol_ident=volume_id, rock_ridge=rock_ridge, joliet=use_joliet, udf=use_udf ) if not module.check_mode: iso_file = pycdlib.PyCdlib() iso_file.new(interchange_level=inter_level, vol_ident=volume_id, rock_ridge=rock_ridge, joliet=use_joliet, udf=use_udf) for src_file in src_file_list: # if specify a dir then go through the dir to add files and dirs if os.path.isdir(src_file): dir_list = [] file_list = [] src_file = src_file.rstrip('/') dir_name = os.path.basename(src_file) add_directory(module, iso_file=iso_file, dir_path='/' + dir_name, rock_ridge=rock_ridge, use_joliet=use_joliet, use_udf=use_udf) # get dir list and file list for path, dirs, files in os.walk(src_file): for filename in files: file_list.append(os.path.join(path, filename)) for dir in dirs: dir_list.append(os.path.join(path, dir)) for new_dir in dir_list: add_directory(module, iso_file=iso_file, dir_path=new_dir.split(os.path.dirname(src_file))[1], rock_ridge=rock_ridge, use_joliet=use_joliet, use_udf=use_udf) for new_file in file_list: add_file(module, iso_file=iso_file, src_file=new_file, file_path=new_file.split(os.path.dirname(src_file))[1], rock_ridge=rock_ridge, use_joliet=use_joliet, use_udf=use_udf) # if specify a file then add this file directly to the '/' path in ISO else: add_file(module, iso_file=iso_file, src_file=src_file, file_path='/' + os.path.basename(src_file), rock_ridge=rock_ridge, use_joliet=use_joliet, use_udf=use_udf) iso_file.write(dest_iso) iso_file.close() result['changed'] = True module.exit_json(**result) if __name__ == '__main__': main()