import sys

def main():
    nowait = 'nowait' not in sys.argv
    args = [arg for arg in sys.argv[1:] if arg != 'nowait']
    if len(args) == 0:
        args = ['calculate', 'generate', 'send']

    try:
        from ausbutler.config import load_config
        from ausbutler.goniec import Goniec
        from ausbutler.interface import Interface

        i = Interface(load_config('butler'))

        if 'calculate' in args:
            i.calculate_all()

        if 'generate' in args:
            files = i.generate_all()
            if 'send' in args:
                client = Goniec(load_config('goniec'))
                client.send(files)
    except Exception as e:
        print 'ERROR: %s: %s' % (type(e).__name__, str(e))

    if nowait:
        raw_input('Press any key to continue...')

if __name__ == '__main__':
    main()