add ceuta and melilla

ree/__init__.py

@@ -5,3 +5,4 @@
 from .balearic import Mallorca, Menorca, Ibiza, Formentera
 from .balearicislands import BalearicIslands
 from .iberianpeninsula import IberianPeninsula
+from .cities import Ceuta, Melilla

ree/cities/__init__.py

@@ -0,0 +1,2 @@
+from .ceuta import Ceuta
+from .melilla import Melilla

ree/cities/ceuta.py

@@ -0,0 +1,17 @@
+from typing import Union
+
+from requests import Session
+
+from ..core import Scraper
+
+
+class Ceuta(Scraper):
+
+    def __init__(self, session: Union[Session, None]=None, verify=True):
+        super(self.__class__, self).__init__(session, verify)
+
+    def get(self, date: Union[str, None]=None, last=True):
+        return super(self.__class__, self).get("CEUTA5M", "Europe/Madrid", "Peninsula", date, last)
+
+    def get_all(self, date: Union[str, None]=None):
+        return self.get(date, False)

ree/cities/melilla.py

@@ -0,0 +1,17 @@
+from typing import Union
+
+from requests import Session
+
+from ..core import Scraper
+
+
+class Melilla(Scraper):
+
+    def __init__(self, session: Union[Session, None]=None, verify=True):
+        super(self.__class__, self).__init__(session, verify)
+
+    def get(self, date: Union[str, None]=None, last=True):
+        return super(self.__class__, self).get("MELILLA5M", "Europe/Madrid", "Peninsula", date, last)
+
+    def get_all(self, date: Union[str, None]=None):
+        return self.get(date, False)

tests/cities/test_ceuta.py

@@ -0,0 +1,30 @@
+import unittest
+
+from arrow import get
+
+from ree import Ceuta, Response
+
+
+class TestCeuta(unittest.TestCase):
+
+    def setUp(self):
+        self.instance = Ceuta()
+        self.date = get('2022-06-20').format('YYYY-MM-DD')
+
+    def test_instance(self):
+        self.assertIsInstance(self.instance, Ceuta)
+
+    def test_get(self):
+        response = self.instance.get(self.date)
+        self.assertIsInstance(response, Response)
+        self.assertIsNotNone(response.timestamp)
+        self.assertEqual(response.demand, 18.9)
+        self.assertEqual(response.diesel, 19.83)
+
+    def test_get_all(self):
+        responses = self.instance.get_all()
+        self.assertIsNotNone(responses)
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/cities/test_melilla.py

@@ -0,0 +1,31 @@
+import unittest
+
+from arrow import get
+
+from ree import Melilla, Response
+
+
+class TestCeuta(unittest.TestCase):
+
+    def setUp(self):
+        self.instance = Melilla()
+        self.date = get('2022-06-20').format('YYYY-MM-DD')
+
+    def test_instance(self):
+        self.assertIsInstance(self.instance, Melilla)
+
+    def test_get(self):
+        response = self.instance.get(self.date)
+        self.assertIsInstance(response, Response)
+        self.assertIsNotNone(response.timestamp)
+        self.assertEqual(response.demand, 16.9)
+        self.assertEqual(response.diesel, 9.13)
+        self.assertEqual(response.waste, 1.86)
+
+    def test_get_all(self):
+        responses = self.instance.get_all()
+        self.assertIsNotNone(responses)
+
+
+if __name__ == '__main__':
+    unittest.main()