import json

from collections import defaultdict

import click

from ennstatus import create_app

from ...status.functions import split_all_servers_to_types

@click.group(short_help='Get statistics')

def stats():

    pass

@stats.command('count')

@click.option('--by-type', 'by_type', is_flag=True, default=False)

@click.pass_obj

def count(obj, by_type):

    app = create_app()

    with app.app_context():

        app.logger.removeHandler(app.logger.handlers[1])

        servers = split_all_servers_to_types()

    if not by_type:

        click.echo(

            'We have %s servers in total!' % (

                click.style(

                    str(sum(len(x) for x in servers.values())),

                    fg='blue'

                )

            )

        )

@stats.command('countries')

@click.option('--by-type', 'by_type', is_flag=True, default=False)

@click.pass_obj

def countries(obj, by_type):

    app = create_app()

    with app.app_context():

        app.logger.removeHandler(app.logger.handlers[1])

        servers = split_all_servers_to_types()

    if not by_type:

        countries = defaultdict(int)

        for key, value in servers.items():

            for server in value:

                countries[server.country] += 1

        for key, value in sorted(countries.items(), key=lambda x: x[1]):

            click.echo(

                '%s: %s' % (

                    click.style(key, fg='green'),

                    click.style(str(value), fg='blue')

                )