_bytes_to_unicode for line data iteration on PackedCchFile

gisce · Apr 2, 2024 · 4fc4cb8 · 4fc4cb8
1 parent b4a1fb6
commit 4fc4cb8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cchloader/file.py b/cchloader/file.py
@@ -104,10 +104,16 @@ def __iter__(self):
     def __next__(self):
         return self.next()
 
+    @staticmethod
+    def _bytes_to_unicode(data):
+        return {x: y.decode() if isinstance(y, bytes) else y for x, y in data.items()}
+
     def next(self):
         for line in self.fd:
             try:
+                # data, errors = self.parser.parse_line(line)
                 data, errors = self.parser.parse_line(line)
+                data = self._bytes_to_unicode(data)
 #                if errors:
 #                    self.stats.errors.append(
 #                        (self.stats.line_number, errors)