refactor imports

1 files changed, 7 insertions, 7 deletions

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()