Updated init function of csv reader

lanl · Nov 15, 2024 · 2047081 · 2047081
1 parent c9e0845
commit 2047081
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/dsi/plugins/file_reader.py b/dsi/plugins/file_reader.py
@@ -42,9 +42,14 @@ class Csv(FileReader):
     # Default value is False.
     strict_mode = False
 
-    def __init__(self, filenames, **kwargs):
+    def __init__(self, filenames, db_name = None, **kwargs):
         super().__init__(filenames, **kwargs)
-        self.csv_data = {}
+        self.csv_data = OrderedDict()
+        if isinstance(filenames, str):
+            self.filenames = [filenames]
+        else:
+            self.filenames = filenames
+        self.db_name = db_name
 
     # def pack_header(self) -> None:
     #     """ Set schema based on the CSV columns """