RedisHelper.py 747 B

12345678910111213141516171819202122
  1. import logging
  2. import redis
  3. logger = logging.getLogger("RedisHelper.py")
  4. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  5. class RedisHelper(object):
  6. def __init__(self, host: str, port: int, password: str):
  7. self.host = host
  8. self.port = port
  9. self.password = password
  10. self.redis_conn = redis.Redis(host=self.host, port=self.port, password=password)
  11. def add_number_to_set(self, key: str, number):
  12. logger.info(f"Redis Set写入: {key} ---> {number}")
  13. self.redis_conn.sadd(key, number)
  14. def set_expire(self, key: str, expire: int):
  15. logger.info(f"Redis Expire: {key} ---> {expire}")
  16. self.redis_conn.expire(key, expire)