python-esios 0.1.0__tar.gz → 0.1.2__tar.gz

{python-esios-0.1.0/python_esios.egg-info → python_esios-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-esios
-Version: 0.1.0
+Version: 0.1.2
 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.0 → python_esios-0.1.2}/esios/api_client.py

@@ -1,4 +1,5 @@
 import requests
+import os
 
 class APIClient:
     base_url = 'https://api.esios.ree.es'
@@ -8,7 +9,11 @@ class APIClient:
         'Host': 'api.esios.ree.es',
     }
 
-    def __init__(self, api_key):
+    def __init__(self, api_key=None):
+        if api_key is None:
+            api_key = os.getenv('ESIOS_API_KEY')  # Try to get the API key from environment variable
+            if api_key is None:
+                raise ValueError("API key must be provided either through constructor or as an environment variable 'ESIOS_API_KEY'")
         self.api_key = api_key
         self.headers['x-api-key'] = self.api_key
 
@@ -16,4 +21,4 @@ class APIClient:
         url = self.base_url + endpoint
         response = requests.request(method, url, headers=self.headers, params=params, json=data)
         response.raise_for_status()
-        return response
+        return response

{python-esios-0.1.0 → python_esios-0.1.2}/esios/endpoints/indicators/__init__.py

@@ -1,7 +1,6 @@
 import pandas as pd
 from esios.api_client import APIClient
 import html
-from bs4 import BeautifulSoup
 
 class Indicators(APIClient):
     def __init__(self, api_key):
@@ -29,7 +28,6 @@ class Indicators(APIClient):
         if df:
             df = pd.DataFrame(data['indicators'])
             df['description'] = df['description'].apply(html.unescape)
-            df
             return df.set_index('id')
         else:
             return data

{python-esios-0.1.0 → python_esios-0.1.2/python_esios.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-esios
-Version: 0.1.0
+Version: 0.1.2
 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.0 → python_esios-0.1.2}/python_esios.egg-info/SOURCES.txt

@@ -4,12 +4,7 @@ setup.py
 esios/__init__.py
 esios/api_client.py
 esios/endpoints/__init__.py
-esios/endpoints/archives/__init__.py
-esios/endpoints/archives/archives.py
 esios/endpoints/indicators/__init__.py
-esios/tests/__init__.py
-esios/tests/test_api_client.py
-esios/tests/test_indicators.py
 python_esios.egg-info/PKG-INFO
 python_esios.egg-info/SOURCES.txt
 python_esios.egg-info/dependency_links.txt

{python-esios-0.1.0 → python_esios-0.1.2}/setup.py

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

python-esios-0.1.0/esios/endpoints/archives/archives.py

@@ -1,32 +0,0 @@
-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)

python-esios-0.1.0/esios/tests/test_api_client.py

@@ -1,15 +0,0 @@
-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()

python-esios-0.1.0/esios/tests/test_indicators.py

@@ -1,20 +0,0 @@
-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()