#!/usr/bin/env python
# -*- coding: UTF-8 -*- 
#功能：掃描某個(gè)key里面的所有元素
#使用方法：python bigkey_save_values.py "zyyset" "m*" 100
#如需獲取全部的元素：python bigkey_save_values.py "zyyset" "*" 100
__author__ = "lcl" 
import sys
import redis 
import os 
pool=redis.ConnectionPool(host='192.168.225.128',port=6379,db=0) 
r = redis.StrictRedis(connection_pool=pool) 
#指定key
key = sys.argv[1]
#掃描匹配值
match = sys.argv[2]
#每次匹配數(shù)量
count = sys.argv[3]
#總數(shù)量
total = 0
#獲取當(dāng)前路徑
path = os.getcwd()
keytype = r.type(key)
print "key的類(lèi)型為%s" % (keytype)
#掃描到的key輸出的文件
txt = path+"/%s.txt" % (key)
#打開(kāi)文件
#f = open(txt,"w")
def list_iter(name):
 list_count = r.llen(key)
 for index in range(list_count):
  yield r.lindex(key, index)
if keytype=='list':
 f = open(txt,"w")
 for item in list_iter(key):
#將匹配到對(duì)應(yīng)key中的member/score輸出到文件中
  f.write("%s %s" % (item,"\n"))
  total = total+1
 f.close
elif keytype=='hash':
 f = open(txt,"w")
 for item in r.hscan_iter(key,match = match,count = count):
#將匹配到對(duì)應(yīng)key中的member/score輸出到文件中
  f.write("%s %s" % (item,"\n"))
  total = total+1
 f.close
elif keytype=='set':
 f = open(txt,"w")
 for item in r.sscan_iter(key,match = match,count = count):
  f.write("%s %s" % (item,"\n"))
  total = total+1
 f.close
elif keytype=='zset':
 f = open(txt,"w")
 for item in r.zscan_iter(key,match = match,count = count):
  f.write("%s %s" % (item,"\n"))
  total = total+1
 f.close
else:
 print("key的類(lèi)型為string，value為：" + r.get(key))
print "key:%s的match：%s的數(shù)量為：%d" % (key,match,total)