__init__.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. import sys,os,shutil
  2. def get_overlay_dir(repo_root):
  3. return os.path.join(repo_root,'test','overlay','tree')
  4. def overlay_setup(repo_root):
  5. def process_srcdir(d):
  6. srcdir = os.path.join(repo_root,*d.split('.'))
  7. destdir = os.path.join(overlay_dir,*d.split('.'))
  8. os.makedirs(destdir)
  9. for fn in os.listdir(srcdir):
  10. if (
  11. fn.endswith('.py') or
  12. d == 'mmgen.data' or
  13. d == 'mmgen' and fn.startswith('secp256k1')
  14. ):
  15. if d == 'mmgen' and fn in fakemods:
  16. make_link(
  17. os.path.join(fakemod_dir,fn),
  18. os.path.join(destdir,fn) )
  19. # link_fn = fn.removesuffix('.py') + '_orig.py' # Python 3.9
  20. link_fn = fn[:-3] + '_orig.py'
  21. else:
  22. link_fn = fn
  23. make_link(
  24. os.path.join(srcdir,fn),
  25. os.path.join(destdir,link_fn) )
  26. overlay_dir = get_overlay_dir(repo_root)
  27. fakemod_dir = os.path.join(repo_root,'test','overlay','fakemods')
  28. fakemods = os.listdir(fakemod_dir)
  29. make_link = os.symlink if sys.platform == 'linux' else shutil.copy2
  30. if not os.path.exists(os.path.join(overlay_dir,'mmgen','main.py')):
  31. sys.stderr.write('Setting up overlay tree\n')
  32. shutil.rmtree(overlay_dir,ignore_errors=True)
  33. for d in (
  34. 'mmgen',
  35. 'mmgen.base_proto',
  36. 'mmgen.base_proto.bitcoin',
  37. 'mmgen.base_proto.bitcoin.tx',
  38. 'mmgen.base_proto.ethereum',
  39. 'mmgen.base_proto.ethereum.pyethereum',
  40. 'mmgen.base_proto.ethereum.rlp',
  41. 'mmgen.base_proto.ethereum.rlp.sedes',
  42. 'mmgen.base_proto.ethereum.tx',
  43. 'mmgen.base_proto.monero',
  44. 'mmgen.data',
  45. 'mmgen.proto',
  46. 'mmgen.share',
  47. 'mmgen.tool',
  48. 'mmgen.tx' ):
  49. process_srcdir(d)
  50. return overlay_dir