import argparse import enum import json import os import re import shutil import subprocess import sys parser = argparse.ArgumentParser() parser.add_argument('df_folder', help='DF base folder') parser.add_argument('--headless', action='store_true', help='Run without opening DF window (requires non-Windows)') parser.add_argument('--keep-status', action='store_true', help='Do not delete final status file') parser.add_argument('--no-quit', action='store_true', help='Do not quit DF when done') parser.add_argument('--test-dir', '--test-folder', help='Base test folder (default: df_folder/test)') parser.add_argument('-t', '--test', dest='tests', nargs='+', help='Test(s) to run (Lua patterns accepted)') args = parser.parse_args() if (not sys.stdin.isatty() or not sys.stdout.isatty() or not sys.stderr.isatty()) and not args.headless: print('WARN: no TTY detected, enabling headless mode') args.headless = True if args.test_dir is not None: args.test_dir = os.path.normpath(os.path.join(os.getcwd(), args.test_dir)) if not os.path.isdir(args.test_dir): print('ERROR: invalid test folder: %r' % args.test_dir) MAX_TRIES = 5 dfhack = 'Dwarf Fortress.exe' if sys.platform == 'win32' else './dfhack' test_status_file = 'test_status.json' class TestStatus(enum.Enum): PENDING = 'pending' PASSED = 'passed' FAILED = 'failed' def get_test_status(): if os.path.isfile(test_status_file): with open(test_status_file) as f: return {k: TestStatus(v) for k, v in json.load(f).items()} def change_setting(content, setting, value): return '[' + setting + ':' + value + ']\n' + re.sub( r'\[' + setting + r':.+?\]', '(overridden)', content, flags=re.IGNORECASE) os.chdir(args.df_folder) if os.path.exists(test_status_file): os.remove(test_status_file) print('Backing up init.txt to init.txt.orig') init_txt_path = 'data/init/init.txt' shutil.copyfile(init_txt_path, init_txt_path + '.orig') with open(init_txt_path) as f: init_contents = f.read() init_contents = change_setting(init_contents, 'INTRO', 'NO') init_contents = change_setting(init_contents, 'SOUND', 'NO') init_contents = change_setting(init_contents, 'WINDOWED', 'YES') init_contents = change_setting(init_contents, 'WINDOWEDX', '80') init_contents = change_setting(init_contents, 'WINDOWEDY', '25') init_contents = change_setting(init_contents, 'FPS', 'YES') if args.headless: init_contents = change_setting(init_contents, 'PRINT_MODE', 'TEXT') test_init_file = 'dfhackzzz_test.init' # Core sorts these alphabetically with open(test_init_file, 'w') as f: f.write(''' devel/dump-rpc dfhack-rpc.txt :lua dfhack.internal.addScriptPath(dfhack.getHackPath()) test/main "lua scr.breakdown_level=df.interface_breakdown_types.%s" ''' % ('NONE' if args.no_quit else 'QUIT')) test_config_file = 'test_config.json' with open(test_config_file, 'w') as f: json.dump({ 'test_dir': args.test_dir, 'tests': args.tests, }, f) try: with open(init_txt_path, 'w') as f: f.write(init_contents) tries = 0 while True: status = get_test_status() if status is not None: if all(s != TestStatus.PENDING for s in status.values()): print('Done!') sys.exit(int(any(s != TestStatus.PASSED for s in status.values()))) elif tries > 0: print('ERROR: Could not read status file') sys.exit(2) tries += 1 print('Starting DF: #%i' % (tries)) if tries > MAX_TRIES: print('ERROR: Too many tries - aborting') sys.exit(1) if args.headless: os.environ['DFHACK_HEADLESS'] = '1' os.environ['DFHACK_DISABLE_CONSOLE'] = '1' process = subprocess.Popen([dfhack], stdin=subprocess.PIPE if args.headless else sys.stdin, stdout=subprocess.PIPE, stderr=subprocess.PIPE) _, err = process.communicate() if err: print('WARN: DF produced stderr: ' + repr(err[:5000])) if process.returncode != 0: print('ERROR: DF exited with ' + repr(process.returncode)) finally: print('\nRestoring original init.txt') shutil.copyfile(init_txt_path + '.orig', init_txt_path) if os.path.isfile(test_init_file): os.remove(test_init_file) if not args.keep_status and os.path.isfile(test_status_file): os.remove(test_status_file) print('Cleanup done')