12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364 |
- 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(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'})
|