python-esios 0.1.0__py3-none-any.whl

esios/api_client.py

@@ -0,0 +1,19 @@
+import requests
+
+class APIClient:
+    base_url = 'https://api.esios.ree.es'
+    headers = {
+        'Accept': "application/json; application/vnd.esios-api-v1+json",
+        'Content-Type': "application/json",
+        'Host': 'api.esios.ree.es',
+    }
+
+    def __init__(self, api_key):
+        self.api_key = api_key
+        self.headers['x-api-key'] = self.api_key
+
+    def _api_call(self, method, endpoint, params=None, data=None):
+        url = self.base_url + endpoint
+        response = requests.request(method, url, headers=self.headers, params=params, json=data)
+        response.raise_for_status()
+        return response

esios/endpoints/__init__.py

@@ -0,0 +1,3 @@
+from .indicators import Indicators
+
+__all__ = ['Indicators']

esios/endpoints/archives/archives.py

@@ -0,0 +1,32 @@
+import os
+import pandas as pd
+from esios.api_client import APIClient
+from esios.utils import preprocessing
+
+class Archives(APIClient):
+    def __init__(self, api_key):
+        super().__init__(api_key)
+
+    def list(self, params=None):
+        endpoint = "/archives"
+        response = self._api_call('GET', endpoint)
+        return response.json()
+
+    def get(self, archive_id, params=None, write_home=None):
+        endpoint = f"/archives/{archive_id}/download"
+        response = self._api_call('GET', endpoint, params=params)
+
+        if write_home:
+            path = os.path.join(write_home, f'{archive_id}.zip')
+            with open(path, 'wb') as f:
+                f.write(response.content)
+            preprocessing.unzip_files_and_remove(path)
+
+        return ArchiveData(response.content)
+
+class ArchiveData:
+    def __init__(self, data):
+        self.data = data
+
+    def to_pandas(self):
+        return pd.DataFrame(self.data)

esios/endpoints/indicators/__init__.py

@@ -0,0 +1,143 @@
+import pandas as pd
+from esios.api_client import APIClient
+import html
+from bs4 import BeautifulSoup
+
+class Indicators(APIClient):
+    def __init__(self, api_key):
+        super().__init__(api_key)
+    
+    def list(self, df=False):
+        """
+        Fetches a list of indicators.
+
+        Parameters
+        ----------
+        params : dict, optional
+            URL parameters to be sent with the request.
+
+        Returns
+        -------
+        list
+            A list of indicators.
+        """
+        endpoint = "/indicators"
+        response = self._api_call('GET', endpoint)
+        
+        data = response.json()
+        
+        if df:
+            df = pd.DataFrame(data['indicators'])
+            df['description'] = df['description'].apply(html.unescape)
+            df
+            return df.set_index('id')
+        else:
+            return data
+
+    def get(self, indicator_id, start_date=None, end_date=None, geo_ids=None, locale=None, time_agg=None, geo_agg=None, time_trunc=None, geo_trunc=None):
+        """
+        Fetches disaggregated indicator data optionally filtered by a date range and geo_ids,
+        grouped by geo_id and month, using specified aggregation settings. All parameters except
+        the indicator_id are optional.
+
+        Parameters
+        ----------
+        indicator_id : int
+            The ID of the indicator to fetch data for.
+        start_date : str, optional
+            The start date of the data range in ISO 8601 format (YYYY-MM-DD).
+        end_date : str, optional
+            The end date of the data range in ISO 8601 format (YYYY-MM-DD).
+        geo_ids : list of int, optional
+            A list of geographic IDs to filter the data by.
+        locale : str, optional, default 'es'
+            Language for translations. Defaults to Spanish ('es'). Can be 'es' for Spanish or 'en' for English.
+        time_agg : str, optional, default 'average'
+            Specifies how to aggregate the data over time. Typical values are 'average', 'sum', etc.
+        geo_agg : str, optional, default 'average'
+            Specifies how to aggregate the data geographically. Typical values are 'average', 'sum', etc.
+        time_trunc : str, optional
+            Specifies how to truncate the data time series. Typical values are 'day', 'month', 'year', etc.
+        geo_trunc : str, optional
+            Specifies how to group data at the geolocalization level. Typical values are 'province', 'region', etc.
+
+        Returns
+        -------
+        IndicatorData
+            An instance of IndicatorData containing the fetched data.
+        """
+        
+        params = {}
+        if start_date:
+            params['start_date'] = start_date
+        if end_date:
+            params['end_date'] = end_date + 'T23:59:59'
+        if geo_ids:
+            params['geo_ids'] = ','.join(map(str, geo_ids))
+        if locale:
+            params['locale'] = locale
+        if time_agg:
+            params['time_agg'] = time_agg
+        if geo_agg:
+            params['geo_agg'] = geo_agg
+        if time_trunc:
+            params['time_trunc'] = time_trunc
+        if geo_trunc:
+            params['geo_trunc'] = geo_trunc
+
+        endpoint = f"/indicators/{indicator_id}"
+        response = self._api_call('GET', endpoint, params=params)
+        return IndicatorData(response.json())
+
+class IndicatorData:
+    def __init__(self, data):
+        self.data = data
+
+    def to_dataframe(self, column_name='value'):
+        """
+        Converts the indicator values data to a Pandas DataFrame.
+
+        Parameters
+        -------
+        column_name : {'value', 'id', 'short_name'}, default 'value'
+            
+        Returns
+        -------
+        pandas.DataFrame
+            A DataFrame containing the indicator values.
+        """
+        data = self.data.get('indicator', {})
+        values = data.get('values', [])
+        
+        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')
+            
+            df.drop(columns=columns[mask], inplace=True)
+            
+            if column_name != 'value':
+                df = df.rename(columns={'value': data[column_name]})
+            
+            return df.tz_convert('Europe/Madrid')
+        else:
+            return pd.DataFrame()
+
+    def get_metadata(self):
+        """
+        Extracts metadata from the indicator data.
+
+        Returns
+        -------
+        dict
+            A dictionary containing the metadata of the indicator.
+        """
+        
+        # extract all inside indicator key, except values
+        metadata = self.data.get('indicator', {})
+        metadata.pop('values', None)
+        
+        return metadata

esios/tests/test_api_client.py

@@ -0,0 +1,15 @@
+import unittest
+from esios.api_client import APIClient
+
+class TestAPIClient(unittest.TestCase):
+
+    def setUp(self):
+        self.api_client = APIClient(api_key="test_key")
+
+    def test_api_call(self):
+        # This test is just a placeholder
+        # You would mock requests and test _api_call
+        pass
+
+if __name__ == '__main__':
+    unittest.main()

esios/tests/test_indicators.py

@@ -0,0 +1,20 @@
+import unittest
+from esios.endpoints.indicators import Indicators
+
+class TestIndicators(unittest.TestCase):
+
+    def setUp(self):
+        self.indicators_client = Indicators(api_key="test_key")
+
+    def test_list(self):
+        # This test is just a placeholder
+        # You would mock requests and test the list method
+        pass
+
+    def test_get(self):
+        # This test is just a placeholder
+        # You would mock requests and test the get method
+        pass
+
+if __name__ == '__main__':
+    unittest.main()

python_esios-0.1.0.dist-info/METADATA

@@ -0,0 +1,35 @@
+Metadata-Version: 2.1
+Name: python-esios
+Version: 0.1.0
+Summary: A Python wrapper for the ESIOS API
+Home-page: https://github.com/datons/python-esios
+Author: Jesús López
+Author-email: jesus.lopez@datons.ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests
+Requires-Dist: pandas
+
+# ESIOS
+
+ESIOS is a Python library that wraps the ESIOS API and returns preprocessed data as Pandas DataFrames.
+
+## Overview
+
+This library provides a simple interface to interact with the ESIOS API, allowing you to fetch and process indicator data.
+
+## Installation
+
+You can install the library using pip:
+
+```bash
+pip install python-esios
+```
+
+## Usage
+
+Visit the [instructions here](https://github.com/datons/python-esios/blob/main/esios/endpoints/indicators/README.ipynb) and learn how to use the library.

python_esios-0.1.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+esios/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+esios/api_client.py,sha256=YZFcXeAmaiebGu2CfSO0qxUidZWn2QiPOktzS-6v3Nk,629
+esios/endpoints/__init__.py,sha256=yNcxKzZGpa_ftPZdk9AhgjzQZjmpfuhNA14Fprp6vLo,60
+esios/endpoints/archives/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+esios/endpoints/archives/archives.py,sha256=2ynY_hE_YXZyPy_O22fJfcLWRHtDDIZ5sitRfV7C8So,943
+esios/endpoints/indicators/__init__.py,sha256=tvXs32eIO2zWZYw_SXJyZwxeKeJQa2rqzD2OeRqe2QA,4832
+esios/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+esios/tests/test_api_client.py,sha256=rotc2dEmKrdm_JMX76H_hmBtihlJp5tS4hGIWPUz458,360
+esios/tests/test_indicators.py,sha256=jymdpuDYh0QCicBYsaCJCipm847xeEA2FjyjSRxS43c,520
+python_esios-0.1.0.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_esios-0.1.0.dist-info/METADATA,sha256=CbIdTKQt9Hvso_pt-dGfXPtLCEd5pNoNYmycx14sHOk,995
+python_esios-0.1.0.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+python_esios-0.1.0.dist-info/top_level.txt,sha256=73ob1QLIeSFvhLPrdMM7880xfv9FfWLkq2ZlxRRA9vs,6
+python_esios-0.1.0.dist-info/RECORD,,

python_esios-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.41.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

python_esios-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+esios