okfn-brasil/serenata-de-amor

"""
Hi, I am Rosie from Operação Serenata de Amor! 🤖

I'm a proof-of-concept for the usage of artificial intelligence for social
control of public administration.

Usage:
  rosie.py run (chamber_of_deputies|federal_senate) [--output=<directory>]
  rosie.py test [chamber_of_deputies|federal_senate|core]

Options:
  --help                Show this screen
  --output=<directory>  Output directory [default: /tmp/serenata-data]
"""
import os
import unittest

from docopt import docopt

import rosie
import rosie.chamber_of_deputies
import rosie.federal_senate


def get_module(arguments):
    modules = ('chamber_of_deputies', 'federal_senate', 'core')
    for module in modules:
        if arguments[module]:
            return module


def run(module, directory):
    module = getattr(rosie, module)
    module.main(directory)


def test(module=None):
    loader = unittest.TestLoader()
    tests_path = 'rosie'

    if module:
        tests_path = os.path.join(tests_path, module)

    tests = loader.discover(tests_path)
    testRunner = unittest.runner.TextTestRunner()
    result = testRunner.run(tests)
    if not result.wasSuccessful():
        exit(1)


def main():
    arguments = docopt(__doc__)
    module = get_module(arguments)

    if arguments['test']:
        test(module)

    if arguments['run']:
        module = module if module != 'core' else None
        run(module, arguments['--output'])


if __name__ == '__main__':
    main()