src/data_generator.py - master - lorkinor/sys_analysis_data_generator

src/data_generator.py
93 строки3 KB
Vlad Borodaev
fixed diapason declaration format and bug in array usage
30 мар 2019, 01:30
30 мар 2019, 01:303f7c0a4
import sys
import csv
import random

DEFAULT_FILENAME = 'data.txt'
PARAMS_DELIMITER = ":"
VALUES_DELIMITER = ","

COUNT_KEY = 'ITERATIONS_COUNT'


def run():
    try:
        file_name = sys.argv[1]
    except IndexError:
        file_name = DEFAULT_FILENAME
    finally:
        process_file(file_name)


def process_file(file_name=DEFAULT_FILENAME):
    with open(file_name, "r") as file:
        if file.mode == "r":
            content = file.readlines()
            params = {}
            for line in content:
                params_setter(line.strip(), params)
            result = generate_data(params)

            generate_csv(result, params.keys())


def params_setter(params_line: str, result):
    params = params_line.split(PARAMS_DELIMITER)
    value = params[1].split(VALUES_DELIMITER)

    if isinstance(value, list) and len(value) == 1:
        value = value[0]
    else:
        value = list(map(strip_param, value))
    result[params[0].strip().upper()] = value


def generate_data(params):
    iterations = int(params[COUNT_KEY])
    del params[COUNT_KEY]

    data_string_array = []

    for x in range(0, iterations):
        result = []

        for key, value in params.items():
            # first parameter is iteration count
            if key == COUNT_KEY:
                del params[key]
            elif isinstance(value, list) and len(value) > 1:
                # take a number from a diapason
                if value[-1] == "DI": # int numbers needed
                    start = int(value[0])
                    stop = int(value[1])
                    result.append(random.randrange(start, stop))
                elif value[-1] == "DF": # float numbers needed
                    start = float(value[0])
                    stop = float(value[1])
                    result.append(round(random.uniform(start, stop), 4))
                # take a number from an array
                elif value[-1] == "A":
                    result.append(random.choice(value[:-1]))
            # take just a single number
            elif isinstance(value, list) and len(value) == 1:
                result += value     #value is a list
        data_string_array.append(result)
    return data_string_array


def generate_csv(generated_data, header_keys):
    with open('output.csv', "w", newline="") as csv_file:
        writer = csv.writer(csv_file)
        writer.writerow(header_keys)
        line_count = 1

        for line in generated_data:
            writer.writerow(line)
            line_count += 1
        print("processed {0} lines".format(line_count))


def strip_param(param: str): return param.strip()


if __name__ == "__main__":
    run()