[wip] [get][post] /config/last_update

pyheatpump/cli.py

@@ -150,7 +150,7 @@ def fetch(type):
 def supervise(since):
     logger = logger_init()
 
-    from .config import config, mac_address_init
+    from .config import config, mac_address_init, get_last_update, set_last_update
     mac_address = config.get('heatpump','mac_address')
     if mac_address == 'None':
         mac_address = mac_address_init()

pyheatpump/config.py

@@ -31,7 +31,8 @@ default_config = {
         'interval': 10000,
         'auth': 'None',
         'initialization': str(datetime.now()),
-        'heatpump_id': 'None'
+        'heatpump_id': 'None',
+        'last_update': 0
     },
     'api': {
         'host': '127.0.0.1',
@@ -136,6 +137,42 @@ async def mac_route(request, *args, **kwargs):
             ', New -> ', mac_address)))
 
 
+def get_last_update():
+    return config.get('supervisor', 'last_update')
+
+
+def set_last_update(last_update: int=0):
+    global config
+    config.set('supervisor', 'last_update', last_update)
+    save_config()
+    return get_last_update()
+
+
+async def last_update_route(request, *args, **kwargs):
+    """
+    last_update :
+        == 0 : since current date - interval
+        < 0 : since current_date + last_update - interval
+        > 0 : since current_data - last_update - interval (same behaviour as < 0) 
+    """
+    global config
+    if request['method'] == 'GET':
+        return PlainTextResponse(get_last_update())
+
+    elif request['method'] == 'POST':
+        body = await request.body()
+        last_update = int(body.decode() or 0)
+        if int(body.decode() or 0):
+            last_update = int(datetime.now().strftime('%s'))
+        elif last_update <= 0:
+            last_update = int(datetime.now().strftime('%s')) + last_update
+        else:
+            last_update = int(datetime.now().strftime('%s')) - last_update
+
+
+        return PlainTextResponse(str(set_last_update(last_update)))
+
+
 ROUTES=[
     Route('/', config_route, methods=['GET', 'POST']),
     Route('/mac', mac_route, methods=['GET', 'POST'])

tests/test_config.py

@@ -2,13 +2,17 @@
 import pytest
 from starlette.authentication import UnauthenticatedUser
 from starlette.testclient import TestClient
-from pyheatpump.config import app, config, default_config, CONFIG_FILES, get_config, set_config, config_route, ROUTES, mac_address_init, save_config
+from pyheatpump.config import app, config, default_config, \
+    CONFIG_FILES, get_config, set_config, config_route, ROUTES, \
+    mac_address_init, save_config, get_last_update, set_last_update, \
+    last_update_route
 from unittest.mock import patch, MagicMock
 from pprint import pprint
 import json
 from tempfile import mkstemp
 from configparser import ConfigParser
 import os
+from requests import Request
 
 @pytest.fixture
 def tmpconf(testdir):
@@ -75,3 +79,15 @@ def test_mac_address(testdir, tmpconf):
         assert config_mac_address == mac_address
     except Exception as e:
         raise e
+
+
+@pytest.mark.asyncio
+async def test_last_update(testdir, tmpconf):
+    req = Request()
+    req.method = 'POST'
+    res = await last_update_route(req)
+    last_update_0 = int(res.body)
+
+    res = await get_last_update(req)
+    last_update_1 = int(res.body)
+    assert last_update_0 == last_update_1