manager.py

#!/usr/bin/env python
# coding: utf-8

__author__ = 'Sophie Kirschner'
__license__ = 'zlib/libpng'
__email__ = 'sophiek@pineapplemachine.com'
__version__ = '1.1.4'



import raws
import pydwarf

import re
import shutil
import importlib
import argparse
import json



# Actually run the program
def __main__(args=None):
    conf = pydwarf.config.load(args=args.__dict__)
    pydwarf.log.debug('Proceeding with configuration: %s.' % conf)
    
    # Report versions
    pydwarf.log.info('Running PyDwarf manager version %s.' % __version__)
    pydwarf.log.debug('With pydwarf version %s.' % pydwarf.__version__)
    pydwarf.log.debug('With raws version %s.' % raws.__version__)
    pydwarf.log.debug('With Dwarf Fortress version %s.' % conf.version)
    pydwarf.log.debug('With DFHack version %s.' % conf.hackversion)
    
    # Handle flags that completely change behavior
    specialtext = None
    
    if args.list:
        scripts = pydwarf.urist.list()
        specialtext = '\n'.join(str(script) for script in scripts)
    elif args.meta is not None:
        specialtext = pydwarf.urist.doclist(args.meta, format=args.metaformat)
    
    if specialtext is not None:
        pydwarf.log.info('\n\n%s\n' % specialtext)
        if args.writedoc:
            with open(args.writedoc, 'wb') as writedoc: writedoc.write(specialtext)
        exit(0)
    
    # Create a new session and run it
    pydwarf.log.info('Configuring session using raws input directory %s.' % conf.input)
    session = pydwarf.session(raws, conf)
    session.run()
    
    # All done!
    pydwarf.log.info('All done!')



def parseargs():
    parser = argparse.ArgumentParser()
    parser.add_argument('-ver', '--version', help='indicate Dwarf Fortress version', type=str)
    parser.add_argument('-i', '--input', help='dwarf fortress input directory', type=str)
    parser.add_argument('-o', '--output', help='dwarf fortress output directory', type=str)
    parser.add_argument('-bak', '--backup', help='dwarf fortress backup directory', type=str)
    parser.add_argument('-p', '--paths', help='which paths relative to input to store in memory and allow access to', nargs='+', type=str)
    parser.add_argument('-s', '--scripts', help='run scripts by name or namespace', nargs='+', type=str)
    parser.add_argument('-pk', '--packages', help='import packages containing PyDwarf scripts', nargs='+', type=str)
    parser.add_argument('-c', '--config', help='indicate json files, yaml files, or python modules to use for settings', nargs='+', type=str)
    parser.add_argument('-v', '--verbose', help='set stdout logging level to DEBUG', action='store_true')
    parser.add_argument('-hver', '--hackversion', help='indicate DFHack version', type=str)
    parser.add_argument('-l', '--log', help='output log file to path', type=str)
    parser.add_argument('-ls', '--list', help='list available scripts', action='store_true')
    parser.add_argument('-js', '--jscripts', help='specify scripts given a json array', type=str)
    parser.add_argument('-m', '--meta', help='show metadata for scripts', nargs='*', type=str)
    parser.add_argument('-mf', '--metaformat', help='how to format shown metadata', type=str)
    parser.add_argument('-wd', '--writedoc', help='write data given by --list or --meta to a file path in addition to the log', type=str)
    args = parser.parse_args()
    
    if args.jscripts is not None: args.scripts = json.loads(args.jscripts) if args.scripts is None else (args.scripts + json.loads(args.jscripts))
    
    return args



if __name__ == "__main__":
    __main__(parseargs())