scblocks.py

#!/usr/bin/python3

"""Read BlocksData.xml and make block information searchable."""

import sys
import xml.etree.ElementTree as ETree
from collections import namedtuple


ToolPower = namedtuple('ToolPower', 'quarry shovel hack weapon longevity')
Block = namedtuple('Block', 'id name power resilience blocks_fluid aimable '
                            'light_attenuation light_emission max_stacking '
                            'nutrition')


def read_block_data(filename):
    """Build Block namedtuples from information in BlocksData.xml."""
    for child in ETree.parse(filename).getroot():
        if child.tag != 'Block' or len(child):
            raise ValueError('Root element may only contain <Block /> tags.')
        yield Block(id=int(child.get('BlockId')),
                    name=child.get('Name'),
                    power=ToolPower._make(map(float, map(child.get, (
                        'QuarryPower', 'ShovelPower', 'HackPower',
                        'WeaponPower', 'AverageToolLongevity')))),
                    resilience=float(child.get('DigResilience')),
                    blocks_fluid=child.get('IsFluidBlocker') == 'True',
                    aimable=child.get('IsAimable') == 'True',
                    light_attenuation=int(child.get('LightAttenuation')),
                    light_emission=int(child.get('EmittedLightAmount')),
                    max_stacking=int(child.get('MaxStacking')),
                    nutrition=float(child.get('NutritionalValue')))


def handle_args(custom_args=None):
    """Parse and return the script's command-line arguments."""
    Args = namedtuple('Args', 'filename block_name')
    try:
        if custom_args is not None:
            filename, block_name = custom_args
        else:
            _, filename, block_name = sys.argv
    except ValueError:
        print('Usage: blocks.py FILENAME BLOCKNAME', file=sys.stderr)
        raise
    return Args(filename, block_name)


def main():
    """The script's main entry point."""
    args = handle_args()
    blocks = read_block_data(args.filename)
    matched_blocks = [blk for blk in blocks
                      if args.block_name.lower() in blk.name.lower()]
    for block in matched_blocks:
        print(block.name)
        for k, v in block._asdict().items():
            print(' '*3, k, '=', v)


if __name__ == '__main__':
    try:
        sys.exit(main())
    except KeyboardInterrupt:
        sys.exit(130)
    except BrokenPipeError:
        sys.exit(0)     # We were piped into something that crashed.