封装带SSH跳板机的REDIS

一、封装ssh的redis

二、setting的配置

三、应用示例

import redis
from sshtunnel import SSHTunnelForwarder
from conf.setting import REDIS_online, REDIS_dev

class MyRedis():

    def __del__(self):
        self.server.close()    

    def __init__(self,host,ip,ssh_username,ssh_password,db,password):
        #构造shh-redis链接，使用setting配置文件
        try:
            server = SSHTunnelForwarder(
                ssh_address_or_host=(host, 22),
                ssh_username=ssh_username,
                ssh_password=ssh_password,
                remote_bind_address=(ip, 6379)
            )
            self.server = server
            self.server.daemon_forward_servers = True #不加server关闭不了
            self.server.start()
            pool = redis.ConnectionPool(host='127.0.0.1', port=self.server.local_bind_port, password=password, db=db,
                                        decode_responses=True)
            redis_connect = redis.Redis(connection_pool=pool)
            self.r = redis_connect
        except Exception as e:
            print('redis连接失败，错误信息%s'%e)

    def str_get(self,k):
        res = self.r.get(k)
        return res

    # def str_set(self,k,v,time=None):
    #     self.r.set(k,v,time)

    # def delete(self,k):
    #     tag = self.r.exists(k) #判断这个key是否存在
    #     if tag:
    #         self.r.delete(k)
    #         print('删除成功')
    #     else:
    #         print('这个key不存在')
    # def hash_get(self,name,k):
    #     res = self.r.hget(name,k)
    #     if res:
    #         return res.decode()
    # def hash_set(self,name,k,v):
    #     self.r.hset(name,k,v)
    # def hash_getall(self,name):
    #     data = {}
    #     # {b'12': b'1212', b'3': b'sdad', b'4': b'asdadsa'}
    #     res = self.r.hgetall(name)
    #     if res:
    #         for k,v in res.items():
    #             k =  k.decode()
    #             v = v.decode()
    #             data[k]=v
    #     return data
    # def hash_del(self,name,k):
    #     res = self.r.hdel(name,k)
    #     if res:
    #         print('删除成功')
    #         return 1
    #     else:
    #         print('删除失败，该key不存在')
    #         return 0


    # @property
    # def clean_redis(self):
    #     self.r.flushdb()  #清1空redis
    #     print('清空redis成功！')
    #     return 0


# my = MyRedis(**REDIS_dev)

# if __name__ == '__main__':
#     my = MyRedis(**REDIS_online)
#     vcode = my.str_get('xxx:vcode:xxx:1xx0000xxxx')
#     print(vcode)

REDIS_online = {
    'host': '1xx.xxx.xxx.xxx',
    'ip' : '1xx.xx.xx.xx',
    'ssh_username' : 'aaa',
    'ssh_password': 'xxx',
    'db':2,
    'password' : 'vvvv'
}

import unittest,requests
from sshtunnel import SSHTunnelForwarder
from lib.my_redis import MyRedis
# from lib.my_sql import my_sql
from conf.setting import BASE_URL,REDIS_online
from urllib.parse import urljoin
from lib.my_request import MyRequest
from lib.tools import login

class Pt_xxx_xxx(unittest.TestCase):
    def test_c_xxx(self):
        '''xxxxx'''
        url = '/xxx/xxx/messageCode'
        real_url = urljoin(BASE_URL, url)

        data = {'xx': 'xxx%2C2',
                'xx': 'xxxA'
                }
        res = MyRequest.get(real_url, data)
        if res.get('msg') == '发xxxxxx,xxx':
            self.assertEqual(401, res.get('code'), msg = '发xxxx==>失败url:%s 失败data:%s'%(real_url,data) )
        else:
            self.assertEqual(200, res.get('code'), msg='发xxxx==>失败url:%s 失败data:%s'%(real_url,data) )


    def test_c_xxxx(self):
        '''xxx'''
        url = '/xx/xxxx/messageCode'
        real_url = urljoin(BASE_URL, url)
        my = MyRedis(**REDIS_online)
        vcode = my.str_get('xx:xx:xxx:1xx000xxx34')

        data = {'xx': 'xx%2C2',
                'xxx': 2,
                'vcode': vcode
                }
        res = MyRequest.get(real_url, data)
        self.assertEqual(200, res.get('code'), msg='获取xx败==>失败url:%s 失败data:%s' % (real_url, data))


if __name__ == '__main__':
    c=xxx()
    c.test_c_xx()
    c.test_c_xx()