pygazpar 1.2.5__py39-none-any.whl → 1.2.7__py39-none-any.whl

pygazpar/datasource.py

@@ -59,9 +59,9 @@ class WebDataSource(IDataSource):
     # ------------------------------------------------------
     def load(self, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
 
-        auth_token = self._login(self.__username, self.__password)
+        self._login(self.__username, self.__password)  # We ignore the return value.
 
-        res = self._loadFromSession(auth_token, pceIdentifier, startDate, endDate, frequencies)
+        res = self._loadFromSession(pceIdentifier, startDate, endDate, frequencies)
 
         Logger.debug("The data update terminates normally")
 
@@ -88,23 +88,23 @@ class WebDataSource(IDataSource):
 
         jar = http.cookiejar.CookieJar()
 
-        session = Session()
-        session.headers.update({"Content-Type": "application/json"})
-        session.headers.update({"X-Requested-With": "XMLHttpRequest"})
+        self._session = Session()
+        self._session.headers.update({"Content-Type": "application/json"})
+        self._session.headers.update({"X-Requested-With": "XMLHttpRequest"})
 
         params = json.loads(AUTH_TOKEN_PARAMS.format(session_token))
 
-        response = session.get(AUTH_TOKEN_URL, params=params, allow_redirects=True, cookies=jar)  # type: ignore
+        response = self._session.get(AUTH_TOKEN_URL, params=params, allow_redirects=True, cookies=jar)  # type: ignore
 
         if response.status_code != 200:
             raise Exception(f"An error occurred while getting the auth token. Status code: {response.status_code} - {response.text}")
 
-        auth_token = session.cookies.get("auth_token", domain="monespace.grdf.fr")
+        auth_token = self._session.cookies.get("auth_token", domain="monespace.grdf.fr")
 
         return auth_token  # type: ignore
 
     @abstractmethod
-    def _loadFromSession(self, auth_token: str, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
+    def _loadFromSession(self, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
         pass
 
 
@@ -133,7 +133,7 @@ class ExcelWebDataSource(WebDataSource):
         self.__tmpDirectory = tmpDirectory
 
     # ------------------------------------------------------
-    def _loadFromSession(self, auth_token: str, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
+    def _loadFromSession(self, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
 
         res = {}
 
@@ -166,16 +166,8 @@ class ExcelWebDataSource(WebDataSource):
             retry = 10
             while retry > 0:
 
-                # Create a session.
-                session = Session()
-                session.headers.update({"Host": "monespace.grdf.fr"})
-                session.headers.update({"Domain": "grdf.fr"})
-                session.headers.update({"X-Requested-With": "XMLHttpRequest"})
-                session.headers.update({"Accept": "application/json"})
-                session.cookies.set("auth_token", auth_token, domain="monespace.grdf.fr")
-
                 try:
-                    self.__downloadFile(session, downloadUrl, self.__tmpDirectory)
+                    self.__downloadFile(self._session, downloadUrl, self.__tmpDirectory)
                     break
                 except Exception as e:
 
@@ -267,7 +259,7 @@ class JsonWebDataSource(WebDataSource):
 
         super().__init__(username, password)
 
-    def _loadFromSession(self, auth_token: str, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
+    def _loadFromSession(self, pceIdentifier: str, startDate: date, endDate: date, frequencies: Optional[List[Frequency]] = None) -> MeterReadingsByFrequency:
 
         res = {}
 
@@ -286,16 +278,8 @@ class JsonWebDataSource(WebDataSource):
         retry = 10
         while retry > 0:
 
-            # Create a session.
-            session = Session()
-            session.headers.update({"Host": "monespace.grdf.fr"})
-            session.headers.update({"Domain": "grdf.fr"})
-            session.headers.update({"X-Requested-With": "XMLHttpRequest"})
-            session.headers.update({"Accept": "application/json"})
-            session.cookies.set("auth_token", auth_token, domain="monespace.grdf.fr")
-
             try:
-                response = session.get(downloadUrl)
+                response = self._session.get(downloadUrl)
 
                 if "text/html" in response.headers.get("Content-Type"):  # type: ignore
                     raise Exception("An error occurred while loading data. Please check your credentials.")
@@ -321,7 +305,7 @@ class JsonWebDataSource(WebDataSource):
         temperaturesUrl = JsonWebDataSource.TEMPERATURES_URL.format(pceIdentifier, endDate.strftime(JsonWebDataSource.INPUT_DATE_FORMAT), days)
 
         # Get weather data.
-        temperatures = session.get(temperaturesUrl).text
+        temperatures = self._session.get(temperaturesUrl).text
 
         # Transform all the data into the target structure.
         daily = JsonParser.parse(data, temperatures, pceIdentifier)

pygazpar/version.py

@@ -1 +1 @@
-__version__ = "1.2.5"
+__version__ = "1.2.7"

{pygazpar-1.2.5.dist-info → pygazpar-1.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pygazpar
-Version: 1.2.5
+Version: 1.2.7
 Summary: Retrieve gas consumption from GrDF web site (French Gas Company)
 Home-page: https://github.com/ssenart/pygazpar
 Author: Stephane Senart
@@ -218,6 +218,16 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.2.7](https://github.com/ssenart/PyGazpar/compare/1.2.7...1.2.6) - 2025-01-06
+
+### Fixed
+- [#79](https://github.com/ssenart/PyGazpar/issues/79): Fix some unittests that wrongly failed because of the new year.
+
+## [1.2.6](https://github.com/ssenart/PyGazpar/compare/1.2.6...1.2.5) - 2025-01-03
+
+### Fixed
+- [#77](https://github.com/ssenart/PyGazpar/issues/77): Some error may occur while requesting data from GrDF API.
+
 ## [1.2.5](https://github.com/ssenart/PyGazpar/compare/1.2.5...1.2.4) - 2024-12-21
 
 ### Fixed

{pygazpar-1.2.5.dist-info → pygazpar-1.2.7.dist-info}/RECORD

@@ -1,11 +1,11 @@
 pygazpar/__init__.py,sha256=qshO_XZbDA2Wrt80ABDs0MoScqJytClAuIJjAnILglk,309
 pygazpar/__main__.py,sha256=Pt3PInX7QiWcs0aBKZN90NTaU8KFnrQiZ5Hsow1eR5U,3177
 pygazpar/client.py,sha256=JdVm0jZbeibwtTumcRbUSFadfXnCUClPMjL95_J6p5Y,2595
-pygazpar/datasource.py,sha256=N9OVdJr4BwtA0W8DvKe4PEE4kLKbSEH2wxAYGNk-iIU,21558
+pygazpar/datasource.py,sha256=D6VBy4ngQDECQfPVi9ZGhKuuOxNDUHjK-9gTDJlATPI,20699
 pygazpar/enum.py,sha256=3ZCk4SziXF6pxgP3MuQ1qxYfqB3X5DOV8Rtd0GHsK9w,898
 pygazpar/excelparser.py,sha256=QNFIErXNgMjGGjsGjdnRxAbNpfVagdYmZQtXRub3_Xc,5931
 pygazpar/jsonparser.py,sha256=OrRdMZNBi7rI4dRGoRW7gjyFwJFk-IvtkRZ_t1XVFrI,1859
-pygazpar/version.py,sha256=jBmZf3HLbiQlWiolOsAA6J5-BbxXD2bqFqEqDH3lfqo,22
+pygazpar/version.py,sha256=49prCLbE3fFzLfxem5rd2dr1iV4_L-bN0N4J7jxU5yA,22
 pygazpar/resources/daily_data_sample.json,sha256=YJovtrNUMs257magTfyxiewLmecySFypcelbGFUUeT8,199583
 pygazpar/resources/hourly_data_sample.json,sha256=N1F-Xz3GaBn2H1p7uKzhkhKCQV8QVR0t76XD6wmFtXA,3
 pygazpar/resources/monthly_data_sample.json,sha256=yrr4SqrB2MubeVU2HX_FRDZKHIhC0LXCqkO1iqnFWcg,3351
@@ -16,12 +16,12 @@ samples/excelSample.py,sha256=ltAl-bBz9-U9YI802JpcIswra-vDS7tR_KL5VNdxJ5c,765
 samples/jsonSample.py,sha256=sYAIusdEJhZdwDAMgHqoWcwDR0FA2eWhSt_2gL_mJRk,736
 samples/testSample.py,sha256=UeirdEtezHwfZDv_75oxul17YzGWn5yZuHfJYTF3Ez0,387
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_client.py,sha256=UiJCqy_Nctl_YDSVpdaYfrbUEzuu9Td22a4WxNnIhJw,5606
+tests/test_client.py,sha256=BFLz0GNk8qEP4d2-7cAaIcLWpkSnDZo3wRb0AlXBNOQ,5765
 tests/test_datafileparser.py,sha256=nAeUpOHtelblMpmbrrnf-2GuMjK5ai65veDoymceprE,818
-tests/test_datasource.py,sha256=Fkn9BOGVKITAgrx9XipR1_ykT7rdvPyt_PAg3ftjfSU,5983
-pygazpar-1.2.5.dist-info/LICENSE.md,sha256=XsCJx_7_BC9tvmE0ZxS1cTNR7ekurog_ea9ybdZ-8tc,1073
-pygazpar-1.2.5.dist-info/METADATA,sha256=zEubC-8R9fwkm3ZLrCyWHuwAl5phRM7i9FpPGd64wm8,18827
-pygazpar-1.2.5.dist-info/WHEEL,sha256=E6NuJA_MIt2CDvbuRmqVFG99302DhLXEpOp5wbsbDeU,93
-pygazpar-1.2.5.dist-info/entry_points.txt,sha256=c_FMZPYlRv1w9EqfgWhlkdJOoje7FcglI0UMm2oRLoI,53
-pygazpar-1.2.5.dist-info/top_level.txt,sha256=P7qn-XtanDPBLQsTvjvLV71wH8RK0DYbx8tzN_rDS70,23
-pygazpar-1.2.5.dist-info/RECORD,,
+tests/test_datasource.py,sha256=NTeD3yQEi4fTw8ZWz2cuUUdeFC8QVjChBMw5jyqia40,6149
+pygazpar-1.2.7.dist-info/LICENSE.md,sha256=XsCJx_7_BC9tvmE0ZxS1cTNR7ekurog_ea9ybdZ-8tc,1073
+pygazpar-1.2.7.dist-info/METADATA,sha256=zrS5U_76Ad5CdYiixgHJbO-_UZ7amdpp-kwM5Qeus3g,19251
+pygazpar-1.2.7.dist-info/WHEEL,sha256=E6NuJA_MIt2CDvbuRmqVFG99302DhLXEpOp5wbsbDeU,93
+pygazpar-1.2.7.dist-info/entry_points.txt,sha256=c_FMZPYlRv1w9EqfgWhlkdJOoje7FcglI0UMm2oRLoI,53
+pygazpar-1.2.7.dist-info/top_level.txt,sha256=P7qn-XtanDPBLQsTvjvLV71wH8RK0DYbx8tzN_rDS70,23
+pygazpar-1.2.7.dist-info/RECORD,,

tests/test_client.py

@@ -1,159 +1,159 @@
-from pygazpar.enum import Frequency
-from pygazpar.client import Client
-from pygazpar.datasource import JsonWebDataSource, TestDataSource, ExcelWebDataSource
-import os
-import pytest
-
-
-class TestClient:
-
-    @classmethod
-    def setup_class(cls):
-        """ setup any state specific to the execution of the given class (which
-        usually contains tests).
-        """
-
-    @classmethod
-    def teardown_class(cls):
-        """ teardown any state that was previously setup with a call to
-        setup_class.
-        """
-
-    def setup_method(self):
-        """ setup any state tied to the execution of the given method in a
-        class.  setup_method is invoked for every test method of a class.
-        """
-        tmpdir = os.path.normpath(f"{os.getcwd()}/tmp")
-
-        # We create the tmp directory if not already exists.
-        if not os.path.exists(tmpdir):
-            os.mkdir(tmpdir)
-
-        self.__username = os.environ["GRDF_USERNAME"]
-        self.__password = os.environ["GRDF_PASSWORD"]
-        self.__pceIdentifier = os.environ["PCE_IDENTIFIER"]
-        self.__tmp_directory = tmpdir
-
-    def teardown_method(self):
-        """ teardown any state that was previously setup with a setup_method
-        call.
-        """
-
-    def test_login_error(self):
-        client = Client(JsonWebDataSource("WrongUsername", "WrongPassword"))
-
-        with pytest.raises(Exception):
-            client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
-
-    def test_hourly_live(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.HOURLY])
-
-        assert (len(data[Frequency.HOURLY.value]) == 0)
-
-    def test_one_day_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 1, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) <= 1)
-
-    def test_two_days_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 2, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) <= 2)
-
-    # @pytest.mark.skip(reason="Requires live data")
-    def test_daily_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) > 0)
-
-    def test_weekly_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
-
-        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
-
-    def test_monthly_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
-
-        assert (len(data[Frequency.MONTHLY.value]) >= 11 and len(data[Frequency.MONTHLY.value]) <= 13)
-
-    def test_yearly_jsonweb(self):
-        client = Client(JsonWebDataSource(self.__username, self.__password))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
-
-        assert (len(data[Frequency.YEARLY.value]) == 1)
-
-    def test_daily_excelweb(self):
-        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) > 0)
-
-    def test_weekly_excelweb(self):
-        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
-
-        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
-
-    def test_monthly_excelweb(self):
-        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
-
-        assert (len(data[Frequency.MONTHLY.value]) >= 12 and len(data[Frequency.MONTHLY.value]) <= 13)
-
-    def test_yearly_excelweb(self):
-        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
-
-        assert (len(data[Frequency.YEARLY.value]) == 1)
-
-    def test_hourly_sample(self):
-        client = Client(TestDataSource())
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.HOURLY])
-
-        assert (len(data[Frequency.HOURLY.value]) == 0)
-
-    def test_daily_sample(self):
-        client = Client(TestDataSource())
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) == 711)
-
-    def test_weekly_sample(self):
-        client = Client(TestDataSource())
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
-
-        assert (len(data[Frequency.WEEKLY.value]) > 0)
-
-    def test_monthly_sample(self):
-        client = Client(TestDataSource())
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
-
-        assert (len(data[Frequency.MONTHLY.value]) > 0)
-
-    def test_yearly_sample(self):
-        client = Client(TestDataSource())
-
-        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
-
-        assert (len(data[Frequency.YEARLY.value]) == 2)
+from pygazpar.enum import Frequency
+from pygazpar.client import Client
+from pygazpar.datasource import JsonWebDataSource, TestDataSource, ExcelWebDataSource
+import os
+import pytest
+
+
+class TestClient:
+
+    @classmethod
+    def setup_class(cls):
+        """ setup any state specific to the execution of the given class (which
+        usually contains tests).
+        """
+
+    @classmethod
+    def teardown_class(cls):
+        """ teardown any state that was previously setup with a call to
+        setup_class.
+        """
+
+    def setup_method(self):
+        """ setup any state tied to the execution of the given method in a
+        class.  setup_method is invoked for every test method of a class.
+        """
+        tmpdir = os.path.normpath(f"{os.getcwd()}/tmp")
+
+        # We create the tmp directory if not already exists.
+        if not os.path.exists(tmpdir):
+            os.mkdir(tmpdir)
+
+        self.__username = os.environ["GRDF_USERNAME"]
+        self.__password = os.environ["GRDF_PASSWORD"]
+        self.__pceIdentifier = os.environ["PCE_IDENTIFIER"]
+        self.__tmp_directory = tmpdir
+
+    def teardown_method(self):
+        """ teardown any state that was previously setup with a setup_method
+        call.
+        """
+
+    def test_login_error(self):
+        client = Client(JsonWebDataSource("WrongUsername", "WrongPassword"))
+
+        with pytest.raises(Exception):
+            client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
+
+    def test_hourly_live(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.HOURLY])
+
+        assert (len(data[Frequency.HOURLY.value]) == 0)
+
+    def test_one_day_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 1, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) <= 1)
+
+    def test_two_days_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 2, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) <= 2)
+
+    # @pytest.mark.skip(reason="Requires live data")
+    def test_daily_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) > 0)
+
+    def test_weekly_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
+
+        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
+
+    def test_monthly_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
+
+        assert (len(data[Frequency.MONTHLY.value]) >= 11 and len(data[Frequency.MONTHLY.value]) <= 13)
+
+    def test_yearly_jsonweb(self):
+        client = Client(JsonWebDataSource(self.__username, self.__password))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
+
+        assert (len(data[Frequency.YEARLY.value]) >= 1)
+
+    def test_daily_excelweb(self):
+        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) > 0)
+
+    def test_weekly_excelweb(self):
+        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
+
+        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
+
+    def test_monthly_excelweb(self):
+        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
+
+        assert (len(data[Frequency.MONTHLY.value]) >= 12 and len(data[Frequency.MONTHLY.value]) <= 13)
+
+    def test_yearly_excelweb(self):
+        client = Client(ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory))
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
+
+        assert (len(data[Frequency.YEARLY.value]) >= 1)
+
+    def test_hourly_sample(self):
+        client = Client(TestDataSource())
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.HOURLY])
+
+        assert (len(data[Frequency.HOURLY.value]) == 0)
+
+    def test_daily_sample(self):
+        client = Client(TestDataSource())
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) == 711)
+
+    def test_weekly_sample(self):
+        client = Client(TestDataSource())
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.WEEKLY])
+
+        assert (len(data[Frequency.WEEKLY.value]) > 0)
+
+    def test_monthly_sample(self):
+        client = Client(TestDataSource())
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.MONTHLY])
+
+        assert (len(data[Frequency.MONTHLY.value]) > 0)
+
+    def test_yearly_sample(self):
+        client = Client(TestDataSource())
+
+        data = client.loadSince(self.__pceIdentifier, 365, [Frequency.YEARLY])
+
+        assert (len(data[Frequency.YEARLY.value]) == 2)

tests/test_datasource.py

@@ -1,166 +1,166 @@
-import os
-from pygazpar.datasource import TestDataSource, JsonFileDataSource, ExcelFileDataSource, JsonWebDataSource, ExcelWebDataSource
-from pygazpar.enum import Frequency
-from datetime import date, timedelta
-from dotenv import load_dotenv
-
-
-class TestAllDataSource:
-
-    # ------------------------------------------------------
-    @classmethod
-    def setup_class(cls):
-        """ setup any state specific to the execution of the given class (which
-        usually contains tests).
-        """
-
-    # ------------------------------------------------------
-    @classmethod
-    def teardown_class(cls):
-        """ teardown any state that was previously setup with a call to
-        setup_class.
-        """
-
-    # ------------------------------------------------------
-    def setup_method(self):
-        """ setup any state tied to the execution of the given method in a
-        class.  setup_method is invoked for every test method of a class.
-        """
-        tmpdir = os.path.normpath(f"{os.getcwd()}/tmp")
-
-        # We create the tmp directory if not already exists.
-        if not os.path.exists(tmpdir):
-            os.mkdir(tmpdir)
-
-        load_dotenv()
-
-        self.__username = os.environ["GRDF_USERNAME"]
-        self.__password = os.environ["GRDF_PASSWORD"]
-        self.__pceIdentifier = os.environ["PCE_IDENTIFIER"]
-        self.__tmp_directory = tmpdir
-
-    # ------------------------------------------------------
-    def teardown_method(self):
-        """ teardown any state that was previously setup with a setup_method
-        call.
-        """
-
-    # ------------------------------------------------------
-    def test_sample(self):
-
-        dataSource = TestDataSource()
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate)
-
-        assert (len(data[Frequency.DAILY.value]) == 711)
-
-        assert (len(data[Frequency.WEEKLY.value]) == 102)
-
-        assert (len(data[Frequency.MONTHLY.value]) == 24)
-
-        assert (len(data[Frequency.YEARLY.value]) == 2)
-
-    # ------------------------------------------------------
-    def test_jsonfile_sample(self):
-
-        dataSource = JsonFileDataSource("tests/resources/donnees_informatives.json", "tests/resources/temperatures.json")
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
-
-        assert (len(data[Frequency.DAILY.value]) == 1096)
-
-        assert (len(data[Frequency.WEEKLY.value]) == 155)
-
-        assert (len(data[Frequency.MONTHLY.value]) == 36)
-
-        assert (len(data[Frequency.YEARLY.value]) == 3)
-
-    # ------------------------------------------------------
-    def test_daily_excelfile_sample(self):
-
-        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_DAILY.xlsx")
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY])
-
-        assert (len(data[Frequency.DAILY.value]) == 363)
-
-    # ------------------------------------------------------
-    def test_weekly_excelfile_sample(self):
-
-        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_WEEKLY.xlsx")
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.WEEKLY])
-
-        assert (len(data[Frequency.WEEKLY.value]) == 53)
-
-    # ------------------------------------------------------
-    def test_monthly_excelfile_sample(self):
-
-        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_MONTHLY.xlsx")
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.MONTHLY])
-
-        assert (len(data[Frequency.MONTHLY.value]) == 13)
-
-    # ------------------------------------------------------
-    def test_yearly_excelfile_sample(self):
-
-        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_DAILY.xlsx")
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.YEARLY])
-
-        assert (len(data[Frequency.YEARLY.value]) == 1)
-
-    # ------------------------------------------------------
-    def test_jsonweb(self):
-
-        dataSource = JsonWebDataSource(self.__username, self.__password)
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
-
-        assert (len(data[Frequency.DAILY.value]) > 0)
-
-        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
-
-        assert (len(data[Frequency.MONTHLY.value]) >= 11 and len(data[Frequency.MONTHLY.value]) <= 13)
-
-        assert (len(data[Frequency.YEARLY.value]) == 1)
-
-    # ------------------------------------------------------
-    def test_excelweb(self):
-
-        dataSource = ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory)
-
-        endDate = date.today()
-        startDate = endDate + timedelta(days=-365)
-
-        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
-
-        assert (len(data[Frequency.DAILY.value]) > 0)
-
-        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
-
-        assert (len(data[Frequency.MONTHLY.value]) >= 12 and len(data[Frequency.MONTHLY.value]) <= 13)
-
-        assert (len(data[Frequency.YEARLY.value]) == 1)
+import os
+from pygazpar.datasource import TestDataSource, JsonFileDataSource, ExcelFileDataSource, JsonWebDataSource, ExcelWebDataSource
+from pygazpar.enum import Frequency
+from datetime import date, timedelta
+from dotenv import load_dotenv
+
+
+class TestAllDataSource:
+
+    # ------------------------------------------------------
+    @classmethod
+    def setup_class(cls):
+        """ setup any state specific to the execution of the given class (which
+        usually contains tests).
+        """
+
+    # ------------------------------------------------------
+    @classmethod
+    def teardown_class(cls):
+        """ teardown any state that was previously setup with a call to
+        setup_class.
+        """
+
+    # ------------------------------------------------------
+    def setup_method(self):
+        """ setup any state tied to the execution of the given method in a
+        class.  setup_method is invoked for every test method of a class.
+        """
+        tmpdir = os.path.normpath(f"{os.getcwd()}/tmp")
+
+        # We create the tmp directory if not already exists.
+        if not os.path.exists(tmpdir):
+            os.mkdir(tmpdir)
+
+        load_dotenv()
+
+        self.__username = os.environ["GRDF_USERNAME"]
+        self.__password = os.environ["GRDF_PASSWORD"]
+        self.__pceIdentifier = os.environ["PCE_IDENTIFIER"]
+        self.__tmp_directory = tmpdir
+
+    # ------------------------------------------------------
+    def teardown_method(self):
+        """ teardown any state that was previously setup with a setup_method
+        call.
+        """
+
+    # ------------------------------------------------------
+    def test_sample(self):
+
+        dataSource = TestDataSource()
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate)
+
+        assert (len(data[Frequency.DAILY.value]) == 711)
+
+        assert (len(data[Frequency.WEEKLY.value]) == 102)
+
+        assert (len(data[Frequency.MONTHLY.value]) == 24)
+
+        assert (len(data[Frequency.YEARLY.value]) == 2)
+
+    # ------------------------------------------------------
+    def test_jsonfile_sample(self):
+
+        dataSource = JsonFileDataSource("tests/resources/donnees_informatives.json", "tests/resources/temperatures.json")
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
+
+        assert (len(data[Frequency.DAILY.value]) == 1096)
+
+        assert (len(data[Frequency.WEEKLY.value]) == 155)
+
+        assert (len(data[Frequency.MONTHLY.value]) == 36)
+
+        assert (len(data[Frequency.YEARLY.value]) == 3)
+
+    # ------------------------------------------------------
+    def test_daily_excelfile_sample(self):
+
+        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_DAILY.xlsx")
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY])
+
+        assert (len(data[Frequency.DAILY.value]) == 363)
+
+    # ------------------------------------------------------
+    def test_weekly_excelfile_sample(self):
+
+        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_WEEKLY.xlsx")
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.WEEKLY])
+
+        assert (len(data[Frequency.WEEKLY.value]) == 53)
+
+    # ------------------------------------------------------
+    def test_monthly_excelfile_sample(self):
+
+        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_MONTHLY.xlsx")
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.MONTHLY])
+
+        assert (len(data[Frequency.MONTHLY.value]) == 13)
+
+    # ------------------------------------------------------
+    def test_yearly_excelfile_sample(self):
+
+        dataSource = ExcelFileDataSource("tests/resources/Donnees_informatives_PCE_DAILY.xlsx")
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.YEARLY])
+
+        assert (len(data[Frequency.YEARLY.value]) == 1)
+
+    # ------------------------------------------------------
+    def test_jsonweb(self):
+
+        dataSource = JsonWebDataSource(self.__username, self.__password)
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
+
+        assert (len(data[Frequency.DAILY.value]) > 0)
+
+        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
+
+        assert (len(data[Frequency.MONTHLY.value]) >= 11 and len(data[Frequency.MONTHLY.value]) <= 13)
+
+        assert (len(data[Frequency.YEARLY.value]) >= 1)
+
+    # ------------------------------------------------------
+    def test_excelweb(self):
+
+        dataSource = ExcelWebDataSource(self.__username, self.__password, self.__tmp_directory)
+
+        endDate = date.today()
+        startDate = endDate + timedelta(days=-365)
+
+        data = dataSource.load(self.__pceIdentifier, startDate, endDate, [Frequency.DAILY, Frequency.WEEKLY, Frequency.MONTHLY, Frequency.YEARLY])
+
+        assert (len(data[Frequency.DAILY.value]) > 0)
+
+        assert (len(data[Frequency.WEEKLY.value]) >= 51 and len(data[Frequency.WEEKLY.value]) <= 54)
+
+        assert (len(data[Frequency.MONTHLY.value]) >= 12 and len(data[Frequency.MONTHLY.value]) <= 13)
+
+        assert (len(data[Frequency.YEARLY.value]) >= 1)