From 8d8e97afa09d98e314b148cc1c797af2f9cb2a18 Mon Sep 17 00:00:00 2001 From: eth3lbert Date: Sun, 14 Apr 2024 03:09:56 +0800 Subject: [PATCH] Optimize: Replace StringIO with TextWrapIO --- AutoBuild/nrdlist.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/AutoBuild/nrdlist.py b/AutoBuild/nrdlist.py index 634a065..4dc4fb8 100644 --- a/AutoBuild/nrdlist.py +++ b/AutoBuild/nrdlist.py @@ -5,7 +5,7 @@ import logging import os import pathlib from base64 import b64encode -from io import BytesIO, StringIO +from io import BytesIO, TextIOWrapper from typing import Dict, List, Set from zipfile import ZipFile, BadZipfile @@ -101,9 +101,8 @@ class Phase3: return False with gzip.GzipFile(fileobj=BytesIO(r.content), mode="rb") as f: - raw_data = BytesIO(f.read()).getvalue().decode() + data_file = TextIOWrapper(BytesIO(f.read())) - data_file = StringIO(raw_data) reader = csv.DictReader(data_file) for row in reader: if row["create_date"]: