【Redis】用python操作redis集群

https://blog.csdn.net/bitcarmanlee/article/details/51852126

 密码不能写到列表中去：

1. 有密码 就是这样 redisconn = StrictRedisCluster(startup_nodes=redis_nodes，password=password)

结合上一篇文章，改进脚本后操作redis-cluster

1. [root@ip–172–31–47–226 tmp]# more /root/get_nottl.py
   
2. # encoding: utf–8
3. “”“
   
4. author: yangyi@youzan.com
5. time: 2018/4/26 下午4:34
6. func: 获取数据库中没有设置ttl的 key
7. ““”
8. import redis
9. import argparse
10. import time
11. import sys
12. import rediscluster
14. class ShowProcess:
15.     “”“
    
16.     显示处理进度的类
17.     调用该类相关函数即可实现处理进度的显示
18.     ““”
19.     i = 0 # 当前的处理进度
20.     max_steps = 0 # 总共需要处理的次数
21.     max_arrow = 50 # 进度条的长度
23.     # 初始化函数，需要知道总共的处理次数
24.     def __init__(self, max_steps):
25.         self.max_steps = max_steps
26.         self.i = 0
28.     # 显示函数，根据当前的处理进度i显示进度
29.     # 效果为[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]100.00%
30.     def show_process(self, i = None):
31.         if i is not None:
32.             self.i = i
33.         else:
34.             self.i += 1
35.         num_arrow = int(int(self.i) * int(self.max_arrow) / int(self.max_steps)) # 计算显示多少个‘>’
36.         num_line = self.max_arrow – num_arrow # 计算显示多少个‘-‘
37.         percent = self.i * 100.0 / self.max_steps # 计算完成进度，格式为xx.xx%
38.         process_bar = ‘[‘ + ‘>’ * num_arrow + ‘ ‘ * num_line + ‘]’+ ‘%.2f’ % percent + ‘%’ + ‘\r’ # 带输出的字符串，‘\r’表示不换行回到最左边
39.         sys.stdout.write(process_bar) # 这两句打印字符到终端
40.         sys.stdout.flush()
42.     def close(self, words=‘done’):
43.         print ”
44.         print words
45.         self.i = 0
48. def check_ttl(redis_conn, no_ttl_file, dbindex):
49.     start_time = time.time()
50.     no_ttl_num = 0
51.     allkey = redis_conn.dbsize()
52.     keys_num = sum(list(set(allkey.values())))
53.     print “key分布如下:”,allkey
54.     print “there are {num} keys in db {index} “.format(num=keys_num, index=dbindex)
55.     process_bar = ShowProcess(keys_num)
56.     with open(no_ttl_file, ‘a’) as f:
58.         for key in redis_conn.scan_iter(count=10):
59.             process_bar.show_process()
60.             if redis_conn.ttl(key) == –1:
61.                 no_ttl_num += 1
62.                 if no_ttl_num < 10000:
63.                     f.write(key+‘\n’)
64.             else:
65.                 continue
67.     process_bar.close()
68.     print “cost time(s):”, time.time() – start_time
69.     print “no ttl keys number:”, no_ttl_num
70.     print “we write keys with no ttl to the file: %s” % no_ttl_file
73. def main():
74.     parser = argparse.ArgumentParser()
75.     parser.add_argument(‘-p’, type=int, dest=‘port’, action=‘store’, help=‘port of redis ‘)
76.     parser.add_argument(‘-a’, type=str, dest=‘password’, action=‘store’, help=‘password of redis ‘)
77.     parser.add_argument(‘-d’, type=str, dest=‘db_list’, action=‘store’, default=0, help=‘ex : -d all / -d 1,2,3,4 ‘)
78.     args = parser.parse_args()
79.     port = args.port
80.     pwd = args.password
81.     if args.db_list == ‘all’:
82.         db_list = [i for i in xrange(0, 16)]
83.     else:
84.         db_list = [int(i) for i in args.db_list.split(‘,’)]
86.     for index in db_list:
87.         try:
88.             startup_nodes = [{“host”: “172.31.47.226”, “port”: port,“db”:index}]
89.             r = rediscluster.StrictRedisCluster(startup_nodes=startup_nodes, password=pwd)
90.         except Exception, e:
91.             print e
92.         else:
93.             no_ttl_keys_file = “/tmp/{port}_{db}_no_ttl_keys.txt”.format(port=port, db=index)
94.             check_ttl(r, no_ttl_keys_file, index)
97. if __name__ == ‘__main__’:
98.     main()