python-esios 0.1.3__tar.gz → 0.1.4__tar.gz

{python_esios-0.1.3 → python_esios-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-esios
-Version: 0.1.3
+Version: 0.1.4
 Summary: A Python wrapper for the ESIOS API
 Home-page: https://github.com/datons/python-esios
 Author: Jesús López

{python_esios-0.1.3 → python_esios-0.1.4}/esios/endpoints/indicators/__init__.py

@@ -87,6 +87,10 @@ class Indicators(APIClient):
         response = self._api_call('GET', endpoint, params=params)
         return IndicatorData(response.json())
 
+class IndicatorData:
+    def __init__(self, data):
+        self.data = data
+
 class IndicatorData:
     def __init__(self, data):
         self.data = data
@@ -109,18 +113,19 @@ class IndicatorData:
         
         if values:
             df = pd.DataFrame(values)
-            df = df.set_index('datetime')
-            df.index = pd.to_datetime(df.index, utc=True)
-            
-            columns = df.columns
-            mask = columns.str.contains('time')
+            if 'datetime' in df.columns:
+                df['datetime'] = pd.to_datetime(df['datetime'], utc=True)
+                df = df.set_index('datetime')
+                df.index = df.index.tz_convert('Europe/Madrid')
             
-            df.drop(columns=columns[mask], inplace=True)
+            # Drop all columns that contain the word 'time' in them
+            df = df[[col for col in df.columns if not 'time' in col]]
             
-            if column_name != 'value':
-                df = df.rename(columns={'value': data[column_name]})
+            # Rename 'value' column if a different column_name is requested
+            if column_name in data and column_name != 'value':
+                df.rename(columns={'value': data[column_name]}, inplace=True)
             
-            return df.tz_convert('Europe/Madrid')
+            return df
         else:
             return pd.DataFrame()
 
@@ -133,9 +138,7 @@ class IndicatorData:
         dict
             A dictionary containing the metadata of the indicator.
         """
+        metadata = self.data.get('indicator', {}).copy()  # Use a copy to avoid mutating original data
+        metadata.pop('values', None)  # Remove values key to extract only metadata
         
-        # extract all inside indicator key, except values
-        metadata = self.data.get('indicator', {})
-        metadata.pop('values', None)
-        
-        return metadata
+        return metadata

{python_esios-0.1.3 → python_esios-0.1.4}/python_esios.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-esios
-Version: 0.1.3
+Version: 0.1.4
 Summary: A Python wrapper for the ESIOS API
 Home-page: https://github.com/datons/python-esios
 Author: Jesús López

{python_esios-0.1.3 → python_esios-0.1.4}/setup.py

@@ -7,7 +7,7 @@ long_description = (this_directory / "README.md").read_text(encoding='utf-8')
 
 setup(
     name='python-esios',
-    version='0.1.3',
+    version='0.1.4',
     packages=find_packages(),
     install_requires=[
         'requests',