python 操作redis 的脚本

绝地灬酷狼 2021-11-04 14:22 230阅读 0赞

需要向认证服务缓存sp 操作 黑名单的各种操作

[![image][]][image 1]

\#!/usr/bin/python  
\# -\*- coding: UTF-8 -\*-  
import redis  
import datetime  
import sys  
import os  
import time

\#IPS-LOCK-smtp-ip  
\#IPS-LOCK-web-183.14.29.22   
def usage():  
     program\_name=sys.argv\[0\]  
     print "操作sp的IP黑名单 ---------"  
     print "其中 ipfile的格式为每一行一个ip EX是 key lifetime ttl"  
     print program\_name+" set|get|ttl|del all|web|smtp|pop|imap ip|ipfile"  
     print program\_name+" set|get|ttl|del all|web|smtp|pop|imap ip|ipfile EX"  
     sys.exit(0)  
def is\_file(pathfile):  
     return os.path.isfile(pathfile)  
def timestamp\_toString(sp):  
     return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(sp))  
def datetime\_toTimestamp(dt):  
     return time.mktime(dt.timetuple())

**\#操作逻辑函数**

def fun(redis\_c,operation,extime=60,\*list\_key):  
     try:     
         list\_key=list(list\_key)  
          
         datetime\_n=datetime.datetime.now()  
         nowtime=datetime\_n.strftime("%Y-%m-%d %H:%M:%S")  
         time\_n=datetime\_toTimestamp(datetime\_n)+extime  
         end\_strtime=timestamp\_toString(time\_n)  
         \#print "xxx %s %s %s %d,%d" %(nowtime,end\_strtime,str(extime),datetime\_toTimestamp(datetime\_n),time\_n)  
          
         for key in list\_key:  
             if operation == "get":     
                 GetKey=conn\_redis.get(key)  
                 \#print "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"  
                 print "get "+key+" rc="+str(GetKey)  
             elif operation == "ttl":  
                 ttl=conn\_redis.ttl(key)  
                 print "ttl "+key+" rc="+str(ttl)  
             elif operation == "del":  
                 del\_ret=conn\_redis.delete(key)  
                 print "delete "+key+" rc="+str(del\_ret)  
             elif operation == "set":  
                 uip = key.split('-')\[-1\] \#IPS-LOCK-web-183.14.29.22   
                 value = "FROM "+nowtime+" "+end\_strtime+" %s sys\_duration=%d" % (key,extime)  
                 rc=redis\_c.setex(key,value,extime)  
                 print "set "+key+" rc="+str(rc)

except Exception,e:  
         print "Exception:"+str(e)

if len(sys.argv) < 4:  
      
     usage()  
else:  
     operation=sys.argv\[1\]  
     op\_type=sys.argv\[2\]  
     op\_ip=sys.argv\[3\]  
     op\_ipfile=None  
     list\_key=\[\]  
     if is\_file(op\_ip): \#如果是ip文件  
             op\_ipfile=op\_ip  
             fdhh=open(op\_ipfile,"rb")  
             line = fdhh.readline()  
             while line:  
                 line= line.strip()  
                 if op\_type == "all":  
                     list\_key.append('IPS-LOCK-web-'+line)  
                     list\_key.append('IPS-LOCK-pop-'+line)  
                     list\_key.append('IPS-LOCK-smtp-'+line)  
                     list\_key.append('IPS-LOCK-imap-'+line)  
                 else:  
                     list\_key.append('IPS-LOCK-'+op\_type+'-'+line)  
                 line=fdhh.readline()

else:  
         if op\_type == "all":  
                     list\_key.append('IPS-LOCK-web-'+op\_ip)  
                     list\_key.append('IPS-LOCK-pop-'+op\_ip)  
                     list\_key.append('IPS-LOCK-smtp-'+op\_ip)  
                     list\_key.append('IPS-LOCK-imap-'+op\_ip)  
         else:  
             list\_key.append('IPS-LOCK-'+op\_type+'-'+op\_ip)

if operation == "get" or operation == "ttl" or operation == "del" or operation == "set" :  
         if operation == "set":  
             if len(sys.argv) != 5:  
                  useage()

else:  
             if len(sys.argv) != 4:  
                 usage()  
         conn\_pool=redis.ConnectionPool(password=xxx ,host='172.16.183.aaaa',port=6379)  
         conn\_redis=redis.Redis(connection\_pool=conn\_pool)

if operation == "set":  
                 fun(conn\_redis,operation,600,\*list\_key)  
         else:  
             print list\_key  
             fun(conn\_redis,operation,0,\*list\_key)

else:  
         usage()

\#connection redis  
\#conn\_redis=redis.Redis(host='172.16.183.xxx',port=6379,password=aaaaa)  
\#connection pool

\#GetKey=conn\_redis.get('IPS-LOCK-web-114.99.23.182')

\#conn\_redis.setex('IPS-LOCK-web-183.14.29.188','web-114.99.23.182 1628036',1628036)

转载于:https://www.cnblogs.com/xiaocry/articles/11308708.html

[image]: /images/20211104/5861e7f56cd441cdbbf4912db4ea35e0.png
[image 1]: https://img2018.cnblogs.com/blog/732632/201908/732632-20190806142900342-1253286295.png