| 1 |
def extract_data(data, link, key, encoding): file = link.split('.') if len(file) >= 2: file_extension = file[-1].lower() if file_extension != 'csv': raise ValueError('File must have extension CSV') with open(link, 'r', encoding=encoding) as f: data[key] = [row[0] for row in csv.reader(f)] if __name__ == '__main__': parser = create_parser() namespace = parser.parse_args() json_data = read_json_data(namespace.database, namespace.encoding) data = extract_all_data(json_data, namespace.encoding) if namespace.format != '': for _ in range(namespace.count): print(format_data_generator.generate_format_data( data, namespace.gender, namespace.format, namespace.weights)) else: for _ in range(namespace.count): print(generate_human(data, namespace.gender, namespace.weights)) |
Комментарии