fix flake8

Author: tomkralidis

msc-wis2node-management/msc_wis2node/env.py

@@ -1,6 +1,6 @@
 ###############################################################################
 #
-# Copyright (C) 2024 Tom Kralidis
+# Copyright (C) 2025 Tom Kralidis
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -32,6 +32,7 @@
 TOPIC_PREFIX = os.environ.get('MSC_WIS2NODE_TOPIC_PREFIX', 'origin/a/wis2')
 DISCOVERY_METADATA_ZIP_URL = os.environ.get('MSC_WIS2NODE_DISCOVERY_METADATA_ZIP_URL')  # noqa
 CACHE = os.environ.get('MSC_WIS2NODE_CACHE')
+CACHE_EXPIRY_SECONDS = int(os.environ.get('MSC_WIS2NODE_CACHE_EXPIRY_SECONDS', 86400))  # noqa
 CENTRE_ID = os.environ.get('MSC_WIS2NODE_CENTRE_ID')
 WIS2_GDC = os.environ.get('MSC_WIS2NODE_WIS2_GDC')
 

msc-wis2node-management/msc_wis2node/publisher.py

@@ -23,7 +23,6 @@
 import json
 import logging
 import re
-import sqlite
 from typing import Union
 import uuid
 
@@ -34,8 +33,8 @@
 import yaml
 
 from msc_wis2node.env import (BROKER_HOSTNAME, BROKER_PORT, BROKER_USERNAME,
-                              BROKER_PASSWORD, CACHE, CENTRE_ID, DATASET_CONFIG,
-                              TOPIC_PREFIX)
+                              BROKER_PASSWORD, CACHE, CACHE_EXPIRY_SECONDS,
+                              CENTRE_ID, DATASET_CONFIG, TOPIC_PREFIX)
 from msc_wis2node.util import get_mqtt_client_id, get_mqtt_tls_settings
 
 LOGGER = logging.getLogger(__name__)
@@ -77,7 +76,7 @@ class WIS2Publisher:
     def __init__(self):
         """initialize"""
 
-        self.cache = memcache_base.Client(CACHE]
+        self.cache = memcache_base.Client(CACHE)
         self.datasets = []
         self.tls = None
 
@@ -183,12 +182,13 @@ def publish_to_wis2(self, dataset: dict, url: str) -> None:
         )
 
         LOGGER.debug('Checking for data update')
-        if message['properties']['data_id'] in self.cache:
+        if self.cache.get(message['properties']['data_id']) is not None:
             update_link = deepcopy(message['links'][0])
             update_link['rel'] = 'update'
             message['links'].append(update_link)
         else:
-            self.cache.append(messages['properties']['data_id']
+            self.cache.set(message['properties']['data_id'],
+                           CACHE_EXPIRY_SECONDS)
 
         cache = dataset.get('cache', True)
         if not cache: