postgresql/src/tools/install_files

76 lines
2.2 KiB
Python

#!/usr/bin/env python3
# Helper to install files that are not part of the default meson install
# target.
#
# This includes files that should only get installed into the temporary
# installation for tests and documentation.
import argparse
import os
import shutil
import sys
from pathlib import PurePath
parser = argparse.ArgumentParser()
parser.add_argument('--destdir', type=str,
default=os.environ.get('DESTDIR', None))
parser.add_argument('--prefix', type=str)
parser.add_argument('--install', type=str, nargs='+',
action='append', default=[])
parser.add_argument('--install-dirs', type=str, nargs='+',
action='append', default=[])
parser.add_argument('--install-dir-contents', type=str, nargs='+',
action='append', default=[])
args = parser.parse_args()
def error_exit(msg: str):
print(msg, file=sys.stderr)
exit(1)
def create_target_dir(prefix: str, destdir: str, targetdir: str):
if not os.path.isabs(targetdir):
targetdir = os.path.join(prefix, targetdir)
if destdir is not None:
# copy of meson's logic for joining destdir and install paths
targetdir = str(PurePath(destdir, *PurePath(targetdir).parts[1:]))
os.makedirs(targetdir, exist_ok=True)
return targetdir
def copy_files(targetdir: str, src_list: list):
for src in src_list:
shutil.copy2(src, targetdir)
def copy_dirs(targetdir: str, src_list: list, contents: bool):
for src in src_list:
if not os.path.isdir(src):
error_exit('{0} is not a directory'.format(src))
if contents:
target = targetdir
else:
target = os.path.join(targetdir, os.path.split(src)[1])
shutil.copytree(src, target, dirs_exist_ok=True)
for installs in args.install:
targetdir = create_target_dir(args.prefix, args.destdir, installs[0])
copy_files(targetdir, installs[1:])
for installs in args.install_dirs:
targetdir = create_target_dir(args.prefix, args.destdir, installs[0])
copy_dirs(targetdir, installs[1:], contents=False)
for installs in args.install_dir_contents:
targetdir = create_target_dir(args.prefix, args.destdir, installs[0])
copy_dirs(targetdir, installs[1:], contents=True)