From 0f6c76556b535975979b94723de49b1378880b1d Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Wed, 9 Apr 2025 12:55:51 +0200 Subject: refactor imports --- parse_data.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'parse_data.py') diff --git a/parse_data.py b/parse_data.py index ac75bd5..ac4ab99 100644 --- a/parse_data.py +++ b/parse_data.py @@ -1,6 +1,6 @@ -import csv -import subprocess -import io +from csv import DictReader +from subprocess import PIPE, Popen +from io import StringIO class Test: def __init__(self, number: int, description: str): @@ -48,22 +48,22 @@ def find_category_in_data(name: str, data: list[Category]) -> int: def read_input(path: str, is_csv: bool): if is_csv: with open(path, 'r') as fd: - return io.StringIO(fd.read()) + return StringIO(fd.read()) else: - process = subprocess.Popen([path], stdout=subprocess.PIPE, stderr=subprocess.PIPE) + process = Popen([path], stdout=PIPE, stderr=PIPE) stdout, stderr = process.communicate() if stderr: print("Error:", stderr.decode()) stdout_str = stdout.decode() - return io.StringIO(stdout_str) + return StringIO(stdout_str) def parse_data(path: str, is_csv: bool): data = [] # List of Category objects stdout_file = read_input(path, is_csv) - csv_reader = csv.DictReader(stdout_file) # Rows are dictionaries + csv_reader = DictReader(stdout_file) # Rows are dictionaries for row in csv_reader: log_type, timestamp, category_name, number, test_description, fail_message = row.values() -- cgit 1.4.1