parite.py

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

"""
============================================
  Analyse Woman / Man balance in Politics
============================================
"""
import argparse
import re
import logging as lg

import analysis.csv as c_an
import analysis.xml as x_an

lg.basicConfig(level=lg.DEBUG)

def parse_arguments():
    """ Get arguments from the command line
    to get different results according to our needs.
    """
    parser = argparse.ArgumentParser()
    parser.add_argument("-d", "--datafile", help="""CSV file containing pieces of
        information about the members of parliament""")
    parser.add_argument("-p", "--byparty", action='store_true', help="""displays
        a graph for each political party""")
    parser.add_argument("-i", "--info", action='store_true', help="""information about
        the file""")
    parser.add_argument("-n", "--displaynames", action='store_true', help="""displays
        the names of all the mps""")
    parser.add_argument("-s", "--searchname", help="""search for a mp name""")
    parser.add_argument("-I", "--index", help="""displays information about the Ith mp""")
    parser.add_argument("-a","--byage", help="""displays a graph for the MPs splitted
    #        between those who are over and those who are under the value of --byage""")
    parser.add_argument("-g", "--groupfirst", help="""displays a graph groupping all the 'g'
        biggest political parties""")
    return parser.parse_args()

if __name__ == '__main__':
    ARGS = parse_arguments()
    try:
        DATAFILE = ARGS.datafile
        if DATAFILE is None:
            raise Warning('You must indicate a datafile!')
    except Warning as exception:
        lg.warning(exception)
    else:
        e = re.search('^.+\.(\D{3})$', ARGS.datafile)
        EXTENSION = e.group(1)
        if EXTENSION == 'xml':
            x_an.launch_analysis(DATAFILE)
        elif EXTENSION == 'csv':
            c_an.launch_analysis(DATAFILE, ARGS.byparty, ARGS.info, ARGS.displaynames,
                           ARGS.searchname, ARGS.index, ARGS.groupfirst, ARGS.byage)
    finally:
        lg.info('#################### Analysis is over ######################')