cli.py

from pathlib import Path

from stratification import (
    create_readable_sample_file,
    init_categories_people,
    read_in_cats,
    run_stratification,
    write_selected_people_to_file,
    Settings,
)

# INPUT FILES etc:
root_io_dir = Path(
    # "/Users/bsh/brett/sortition/foundation/projects-events/Stratification-Services/HelathyDemocracy/"
    #"/Users/bsh/brett/sortition/foundation/projects-events/Stratification-Services/process/python/example-input-output/"
    "/Users/bsh/brett/sortition/foundation/projects-events/Stratification-Services/UK climate CA/data"
    # "/home/hamish/dev/sortition/brett-20190513/"
)

# first row MUST be fields named: category, name, min, max
# to randomly select one person, modify this file (and change total below to 1 !)
category_file_path = root_io_dir / "categories.csv"

# this file MUST have at least columns = id_column (below) AND those described
# in category file (other columns are ignored)
people_file_path = root_io_dir / "Respondents for Ariel.csv"

# OUTPUT FILES:
people_selected_file_path = root_io_dir / "example_people_22-selected.csv"
people_remaining_file_path = root_io_dir / "example_people_22-remaining.csv"

#######

# the number of people in each category must be (more or less) the total number of people to be selected
number_people_wanted = 110

######

# for testing - WARNING: will overwrite people_file_path  file!
create_sample_file = False
number_people_example_file = 300


def main():
    output_lines = []
    min_max_people = {}
    settings, message = Settings.load_from_file()
    if message:
        print(message)

    with open(category_file_path, "r") as category_file:
        categories, min_max_people = read_in_cats(category_file)
    if create_sample_file:
        with open(people_file_path, "w") as people_file:
            create_readable_sample_file(categories, people_file, number_people_example_file, settings)

    with open(people_file_path, "r") as people_file:
        people, columns_data, output_lines = init_categories_people(people_file, categories, settings)

    success, tries, people_selected, new_output_lines = run_stratification(categories, people, columns_data, number_people_wanted, min_max_people, settings)
    output_lines += new_output_lines

    if success:
        # write selected people to a file
        with open(people_selected_file_path, mode="w") as selected_file, open(people_remaining_file_path, mode="w") as remaining_file:
            # output_lines += write_selected_people_to_file(people, people_selected, id_column, categories, columns_to_keep, columns_data, check_same_address, check_same_address_columns, selected_file, remaining_file)
            output_lines += write_selected_people_to_file(people, people_selected, categories, columns_data, selected_file, remaining_file, settings)
    print("\n".join(output_lines) + "\n")


if __name__ == "__main__":
    main()