import redis
cf.read("conf.conf")
cf = configparser.ConfigParser()
class Redis_Operation:
def __init__(self, host=cf.get("redis_address", "redis_host"),
port=cf.getint("redis_address", "redis_port"),
db=cf.getint("redis_address", "db"), password=cf.get("redis_address",
"password")):
self.r = redis.Redis(host=host, port=port, db=db, password=password)
def insert_hash_data(self, key, second_key, data):
self.r.hset(key, second_key, data)
self.r.save()
return 1
def get_hash_data(self, key, second_key):
res = self.r.hget(key, second_key)
return res
def del_hash_data(self, key, second_key):
self.r.hdel(key, second_key)
return 1
def query_hash_keys(self, key):
res = self.r.hkeys(key)
return res
def insert_into_train_queue(self, model_name):
res = self.r.rpush("train_queue", model_name)
return 1
def pop_train_queue(self):
res = self.r.lpop("train_queue")
return res
def get_data_by_index(self, index):
res = self.r.lindex("train_queue", index)
return res
cf.get()函数是从配置文件中读取信息,如果使用的redis没有密码就把password=password这个字段去掉就好。每个函数的作用如同函数名一样,导入类之后直接调用就行。