#!/usr/bin/env python3 # # MMGen Wallet, a terminal-based cryptocurrency wallet # Copyright (C)2013-2024 The MMGen Project # Licensed under the GNU General Public License, Version 3: # https://www.gnu.org/licenses # Public project repositories: # https://github.com/mmgen/mmgen-wallet # https://gitlab.com/mmgen/mmgen-wallet """ test.overlay.__init__: Initialize the MMGen test suite overlay tree """ import sys, os, shutil def get_overlay_tree_dir(repo_root): return os.path.join(repo_root, 'test', 'overlay', 'tree') def overlay_setup(repo_root): def process_srcdir(pkgname, d): relpath = d.split('.') srcdir = os.path.join(repo_root, *relpath) destdir = os.path.join(overlay_tree_dir, *relpath) fakemod_dir = os.path.join(fakemod_root, pkgname, *(relpath[1:])) os.makedirs(destdir) for fn in os.listdir(srcdir): if ( fn.endswith('.py') or d == f'{pkgname}.data' or (d == 'mmgen.proto.secp256k1' and fn.startswith('secp256k1')) ): if fn.endswith('.py') and os.path.exists(os.path.join(fakemod_dir, fn)): make_link( os.path.join(fakemod_dir, fn), os.path.join(destdir, fn)) link_fn = fn.removesuffix('.py') + '_orig.py' else: link_fn = fn make_link( os.path.join(srcdir, fn), os.path.join(destdir, link_fn)) overlay_tree_dir = get_overlay_tree_dir(repo_root) fakemod_root = os.path.join(repo_root, 'test', 'overlay', 'fakemods') common_path = os.path.join(os.path.sep, 'test', 'overlay', 'fakemods') pkgdata = (( os.path.realpath(e.path).removesuffix(os.path.join(common_path, e.name)), e.name ) for e in os.scandir(fakemod_root) if e.is_dir()) for repodir, pkgname in pkgdata: if not os.path.exists(os.path.join(overlay_tree_dir, pkgname)): sys.stderr.write(f'Setting up overlay tree: {pkgname}\n') make_link = os.symlink if sys.platform in ('linux', 'darwin') else shutil.copy2 shutil.rmtree(os.path.join(overlay_tree_dir, pkgname), ignore_errors=True) import configparser cfg = configparser.ConfigParser() cfg.read(os.path.join(repodir, 'setup.cfg')) for d in cfg.get('options', 'packages').split(): process_srcdir(pkgname, d) sys.path.insert(0, overlay_tree_dir) return overlay_tree_dir