mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
#!/usr/bin/env python
|
|
"""Create a role archive which overwrites an arbitrary file."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import pathlib
|
|
import tarfile
|
|
import tempfile
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('archive', type=pathlib.Path, help='archive to create')
|
|
parser.add_argument('content', type=pathlib.Path, help='content to write')
|
|
parser.add_argument('target', type=pathlib.Path, help='file to overwrite')
|
|
|
|
args = parser.parse_args()
|
|
|
|
create_archive(args.archive, args.content, args.target)
|
|
|
|
|
|
def generate_files_from_path(path):
|
|
if os.path.isdir(path):
|
|
for subpath in os.listdir(path):
|
|
_path = os.path.join(path, subpath)
|
|
yield from generate_files_from_path(_path)
|
|
elif os.path.isfile(path):
|
|
yield pathlib.Path(path)
|
|
|
|
|
|
def create_archive(archive_path: pathlib.Path, content_path: pathlib.Path, target_path: pathlib.Path) -> None:
|
|
with (
|
|
tarfile.open(name=archive_path, mode='w') as role_archive,
|
|
tempfile.TemporaryDirectory() as temp_dir_name,
|
|
):
|
|
temp_dir_path = pathlib.Path(temp_dir_name)
|
|
|
|
meta_main_path = temp_dir_path / 'meta' / 'main.yml'
|
|
meta_main_path.parent.mkdir()
|
|
meta_main_path.write_text('')
|
|
|
|
symlink_path = temp_dir_path / 'symlink'
|
|
symlink_path.symlink_to(target_path)
|
|
|
|
role_archive.add(meta_main_path)
|
|
role_archive.add(symlink_path)
|
|
|
|
for path in generate_files_from_path(content_path):
|
|
if path == content_path:
|
|
arcname = str(symlink_path)
|
|
else:
|
|
arcname = os.path.join(temp_dir_path, path)
|
|
|
|
content_tarinfo = role_archive.gettarinfo(path, arcname)
|
|
with path.open('rb') as file_content:
|
|
role_archive.addfile(content_tarinfo, file_content)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|