sonar/sonar_control/memory_storage.py

import configparser

import redis


class MemoryStorage():
    def __init__(self, ):
        config = configparser.ConfigParser()
        config.read("database.conf", encoding="utf-8")
        self.ip = config.get("redis", "redis_host")
        self.port = config.get("redis", "redis_port")
        self.conn = redis.StrictRedis(host=self.ip, port=self.port, db=0, decode_responses=True)

    def connected(self):
        pass

    def set_value(self, data_dict):
        try:
            pipe = self.conn.pipeline(transaction=True)
            for key_name in data_dict.keys():
                pipe.set(key_name, data_dict[key_name], ex=120)
            pipe.execute()
        except Exception as e:
            print('redis set values error: ', e)
            return e
        else:
            return True

    def set_value_permanent(self, data_dict):
        try:
            pipe = self.conn.pipeline(transaction=True)
            for key_name in data_dict.keys():
                pipe.set(key_name, data_dict[key_name])
            pipe.execute()
        except Exception as e:
            print(e)
            return e
        else:
            return True

    def get_value(self, keys):
        dict = {}
        try:
            pipe = self.conn.pipeline(transaction=True)
            for index in range(len(keys)):
                pipe.get(keys[index])
            result = pipe.execute()
            for index in range(len(keys)):
                dict[keys[index]] = result[index]
            return dict
        except Exception as e:
            print('get_value', e)
            return dict

    def is_connected(self):
        pass

    def re_connected(self):
        pass


if __name__ == '__main__':
    redis_db = MemoryStorage()
    redis_db.set_value({'flag': '123'})