Merge pull request #30 from simranmattu14/pg_updates

Added get_value method to easily get values from csv files

Author: kngai

woudc_extcsv/__init__.py

@@ -382,7 +382,6 @@ def init_table(self, table_name, fields, line_num):
         :param line_num: line number of the table's header (its name)
         :returns: final name for the new table
         """
-
         if table_name not in self._table_count:
             self._table_count[table_name] = 1
         else:
@@ -402,6 +401,39 @@ def init_table(self, table_name, fields, line_num):
 
         return table_name
 
+    def get_value(self, table, field=None, index=None):
+        """
+        Get value from table field.
+
+        :param table: table name
+        :param field: field name
+        :returns: value in the extcsv file.
+        """
+        if table not in self.extcsv:
+            raise KeyError(f"Table '{table}' not found in extcsv.")
+
+        if field is not None and field not in self.extcsv[table]:
+            raise KeyError(f"Field '{field}' not found in table '{table}'.")
+
+        headers = [
+            'CONTENT', 'DATA_GENERATION', 'PLATFORM',
+            'INSTRUMENT', 'LOCATION', 'TIMESTAMP'
+        ]
+
+        if index == 1:
+            return self.extcsv[table][field][0]
+
+        if table in headers and field is None:
+            data = self.extcsv[table]
+            for field in list(data.keys()):
+                if field != 'comments' and isinstance(data[field], list):
+                    data[field] = data[field][0]
+            return data
+        elif table not in headers and field is None:
+            return self.extcsv[table]
+
+        return self.extcsv[table][field]
+
     def add_field_to_table(self, table_name, fields, index=1):
         """
         Record an empty column(s) in Extended CSV table named <table_name>