1、现在给我的感觉是,python终于入门了开始越学越简单了。变得更好理解了。
一、memcached
Memcached 是一个高性能的分布式内存对象缓存系统,用于动态Web应用以减轻数据库负载。它通过在内存中缓存数据和对象来减少读取数据库的次数,从而提高动态、数据库驱动网站的速度。Memcached基于一个存储键/值对的hashmap。其守护进程(daemon )是用C写的,但是客户端可以用任何语言来编写,并通过memcached协议与守护进程通信。
Python操作Memcached
1、第一次操作
1 import memcache 2 3 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 4 mc.set("foo", "bar") 5 ret = mc.get(‘foo‘) 6 print ret
Ps:debug = True 表示运行出现错误时,现实错误信息,上线后移除该参数。
2、天生支持集群
python-memcached模块原生支持集群操作,其原理是在内存维护一个主机列表,且集群中主机的权重值和主机在列表中重复出现的次数成正比
1 主机 权重 2 1.1.1.1 1 3 1.1.1.2 2 4 1.1.1.3 1 5 6 那么在内存中主机列表为: 7 host_list = ["1.1.1.1", "1.1.1.2", "1.1.1.2", "1.1.1.3", ]
如果用户根据如果要在内存中创建一个键值对(如:k1 = "v1"),那么要执行一下步骤:
- 根据算法将 k1 转换成一个数字
- 将数字和主机列表长度求余数,得到一个值 N( 0 <= N < 列表长度 )
- 在主机列表中根据 第2步得到的值为索引获取主机,例如:host_list[N]
- 连接 将第3步中获取的主机,将 k1 = "v1" 放置在该服务器的内存中
代码实现如下:
1 mc = memcache.Client([(‘1.1.1.1:12000‘, 1), (‘1.1.1.2:12000‘, 2), (‘1.1.1.3:12000‘, 1)], debug=True) 2 3 mc.set(‘k1‘, ‘v1‘)
3、add
添加一条键值对,如果已经存在的 key,重复执行add操作异常
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 mc.add(‘k1‘, ‘v1‘) 7 # mc.add(‘k1‘, ‘v2‘) # 报错,对已经存在的key重复添加,失败!!!
4、replace
replace 修改某个key的值,如果key不存在,则异常
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 # 如果memcache中存在kkkk,则替换成功,否则一场 7 mc.replace(‘kkkk‘,‘999‘)
5、set 和 set_multi
set 设置一个键值对,如果key不存在,则创建,如果key存在,则修改
set_multi 设置多个键值对,如果key不存在,则创建,如果key存在,则修改
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 7 mc.set(‘key0‘, ‘wupeiqi‘) 8 9 mc.set_multi({‘key1‘: ‘val1‘, ‘key2‘: ‘val2‘})
6、delete 和 delete_multi
delete 在Memcached中删除指定的一个键值对
delete_multi 在Memcached中删除指定的多个键值对
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 7 mc.delete(‘key0‘) 8 mc.delete_multi([‘key1‘, ‘key2‘])
7、get 和 get_multi
get 获取一个键值对
get_multi 获取多一个键值对
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 7 val = mc.get(‘key0‘) 8 item_dict = mc.get_multi(["key1", "key2", "key3"])
8、append 和 prepend
append 修改指定key的值,在该值 后面 追加内容
prepend 修改指定key的值,在该值 前面 插入内容
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 # k1 = "v1" 7 8 mc.append(‘k1‘, ‘after‘) 9 # k1 = "v1after" 10 11 mc.prepend(‘k1‘, ‘before‘) 12 # k1 = "beforev1after"
9、decr 和 incr
incr 自增,将Memcached中的某一个值增加 N ( N默认为1 )
decr 自减,将Memcached中的某一个值减少 N ( N默认为1 )
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 5 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True) 6 mc.set(‘k1‘, ‘777‘) 7 8 mc.incr(‘k1‘) 9 # k1 = 778 10 11 mc.incr(‘k1‘, 10) 12 # k1 = 788 13 14 mc.decr(‘k1‘) 15 # k1 = 787 16 17 mc.decr(‘k1‘, 10) 18 # k1 = 777
10、gets 和 cas
如商城商品剩余个数,假设改值保存在memcache中,product_count = 900
A用户刷新页面从memcache中读取到product_count = 900
B用户刷新页面从memcache中读取到product_count = 900
如果A、B用户均购买商品
A用户修改商品剩余个数 product_count=899
B用户修改商品剩余个数 product_count=899
如此一来缓存内的数据便不在正确,两个用户购买商品后,商品剩余还是 899
如果使用python的set和get来操作以上过程,那么程序就会如上述所示情况!
如果想要避免此情况的发生,只要使用 gets 和 cas 即可,如:
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import memcache 4 mc = memcache.Client([‘10.211.55.4:12000‘], debug=True, cache_cas=True) 5 6 v = mc.gets(‘product_count‘) 7 # ... 8 # 如果有人在gets之后和cas之前修改了product_count,那么,下面的设置将会执行失败,剖出异常,从而避免非正常数据的产生 9 mc.cas(‘product_count‘, "899")
Ps:本质上每次执行gets时,会从memcache中获取一个自增的数字,通过cas去修改gets的值时,会携带之前获取的自增值和memcache中的自增值进行比较,如果相等,则可以提交,如果不想等,那表示在gets和cas执行之间,又有其他人执行了gets(获取了缓冲的指定值), 如此一来有可能出现非正常数据,则不允许修改。
二、redis
redis是一个key-value存储系统。和Memcached类似,它支持存储的value类型相对更多,包括string(字符串)、list(链表)、set(集合)、zset(sorted set --有序集合)和hash(哈希类型)。这些数据类型都支持push/pop、add/remove及取交集并集和差集及更丰富的操作,而且这些操作都是原子性的。在此基础上,redis支持各种不同方式的排序。与memcached一样,为了保证效率,数据都是缓存在内存中。区别的是redis会周期性的把更新的数据写入磁盘或者把修改操作写入追加的记录文件,并且在此基础上实现了master-slave(主从)同步。
API使用
redis-py 的API的使用可以分类为:
- 连接方式
- 连接池
- 操作
- String 操作
- Hash 操作
- List 操作
- Set 操作
- Sort Set 操作
- 管道
- 发布订阅
1.操作模式
redis-py提供两个类Redis和StrictRedis用于实现Redis的命令,StrictRedis用于实现大部分官方的命令,并使用官方的语法和命令,Redis是StrictRedis的子类,用于向后兼容旧版本的redis-py。
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 import redis 5 6 r = redis.Redis(host=‘10.211.55.4‘, port=6379) 7 r.set(‘foo‘, ‘Bar‘) 8 print r.get(‘foo‘)
2.连接池
redis-py使用connection pool来管理对一个redis server的所有连接,避免每次建立、释放连接的开销。默认,每个Redis实例都会维护一个自己的连接池。可以直接建立一个连接池,然后作为参数Redis,这样就可以实现多个Redis实例共享一个连接池。
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 import redis 5 6 pool = redis.ConnectionPool(host=‘10.211.55.4‘, port=6379) 7 8 r = redis.Redis(connection_pool=pool) 9 r.set(‘foo‘, ‘Bar‘) 10 print r.get(‘foo‘)
3、操作
String操作,redis中的String在在内存中按照一个name对应一个value来存储。如图:
set(name, value, ex=None, px=None, nx=False, xx=False)
1 在Redis中设置值,默认,不存在则创建,存在则修改 2 参数: 3 ex,过期时间(秒) 4 px,过期时间(毫秒) 5 nx,如果设置为True,则只有name不存在时,当前set操作才执行 6 xx,如果设置为True,则只有name存在时,岗前set操作才执行
setnx(name, value)
# 设置值,只有name不存在时,执行设置操作(添加)
setex(name, value, time)
1 # 设置值 2 # 参数: 3 # time,过期时间(数字秒 或 timedelta对象)
psetex(name, time_ms, value)
1 # 设置值 2 # 参数: 3 # time_ms,过期时间(数字毫秒 或 timedelta对象)
mset(*args, **kwargs)
1 批量设置值 2 如: 3 mset(k1=‘v1‘, k2=‘v2‘) 4 或 5 mget({‘k1‘: ‘v1‘, ‘k2‘: ‘v2‘})
get(name)
1 获取值
mget(keys, *args)
1 批量获取 2 如: 3 mget(‘ylr‘, ‘wupeiqi‘) 4 或 5 r.mget([‘ylr‘, ‘wupeiqi‘])
getset(name, value)
1 设置新值并获取原来的值
getrange(key, start, end)
1 # 获取子序列(根据字节获取,非字符) 2 # 参数: 3 # name,Redis 的 name 4 # start,起始位置(字节) 5 # end,结束位置(字节) 6 # 如: "武沛齐" ,0-3表示 "武"
setrange(name, offset, value)
1 # 修改字符串内容,从指定字符串索引开始向后替换(新值太长时,则向后添加) 2 # 参数: 3 # offset,字符串的索引,字节(一个汉字三个字节) 4 # value,要设置的值
setbit(name, offset, value)
1 # 对name对应值的二进制表示的位进行操作 2 3 # 参数: 4 # name,redis的name 5 # offset,位的索引(将值变换成二进制后再进行索引) 6 # value,值只能是 1 或 0 7 8 # 注:如果在Redis中有一个对应: n1 = "foo", 9 那么字符串foo的二进制表示为:01100110 01101111 01101111 10 所以,如果执行 setbit(‘n1‘, 7, 1),则就会将第7位设置为1, 11 那么最终二进制则变成 01100111 01101111 01101111,即:"goo" 12 13 # 扩展,转换二进制表示: 14 15 # source = "武沛齐" 16 source = "foo" 17 18 for i in source: 19 num = ord(i) 20 print bin(num).replace(‘b‘,‘‘) 21 22 特别的,如果source是汉字 "武沛齐"怎么办? 23 答:对于utf-8,每一个汉字占 3 个字节,那么 "武沛齐" 则有 9个字节 24 对于汉字,for循环时候会按照 字节 迭代,那么在迭代时,将每一个字节转换 十进制数,然后再将十进制数转换成二进制 25 11100110 10101101 10100110 11100110 10110010 10011011 11101001 10111101 10010000 26 -------------------------- ----------------------------- ----------------------------- 27 武 沛 齐
getbit(name, offset)
1 # 获取name对应的值的二进制表示中的某位的值 (0或1)
bitcount(key, start=None, end=None)
1 # 获取name对应的值的二进制表示中 1 的个数 2 # 参数: 3 # key,Redis的name 4 # start,位起始位置 5 # end,位结束位置
bitop(operation, dest, *keys)
1 # 获取多个值,并将值做位运算,将最后的结果保存至新的name对应的值 2 3 # 参数: 4 # operation,AND(并) 、 OR(或) 、 NOT(非) 、 XOR(异或) 5 # dest, 新的Redis的name 6 # *keys,要查找的Redis的name 7 8 # 如: 9 bitop("AND", ‘new_name‘, ‘n1‘, ‘n2‘, ‘n3‘) 10 # 获取Redis中n1,n2,n3对应的值,然后讲所有的值做位运算(求并集),然后将结果保存 new_name 对应的值中
strlen(name)
1 # 返回name对应值的字节长度(一个汉字3个字节)
incr(self, name, amount=1)
1 # 自增 name对应的值,当name不存在时,则创建name=amount,否则,则自增。 2 3 # 参数: 4 # name,Redis的name 5 # amount,自增数(必须是整数) 6 7 # 注:同incrby
incrbyfloat(self, name, amount=1.0)
1 # 自增 name对应的值,当name不存在时,则创建name=amount,否则,则自增。 2 3 # 参数: 4 # name,Redis的name 5 # amount,自增数(浮点型)
decr(self, name, amount=1)
1 # 自减 name对应的值,当name不存在时,则创建name=amount,否则,则自减。 2 3 # 参数: 4 # name,Redis的name 5 # amount,自减数(整数)
append(key, value)
1 # 在redis name对应的值后面追加内容 2 3 # 参数: 4 key, redis的name 5 value, 要追加的字符串
Hash操作,redis中Hash在内存中的存储格式如下图:
hset(name, key, value)
1 # name对应的hash中设置一个键值对(不存在,则创建;否则,修改) 2 3 # 参数: 4 # name,redis的name 5 # key,name对应的hash中的key 6 # value,name对应的hash中的value 7 8 # 注: 9 # hsetnx(name, key, value),当name对应的hash中不存在当前key时则创建(相当于添加)
hmset(name, mapping)
1 # 在name对应的hash中批量设置键值对 2 3 # 参数: 4 # name,redis的name 5 # mapping,字典,如:{‘k1‘:‘v1‘, ‘k2‘: ‘v2‘} 6 7 # 如: 8 # r.hmset(‘xx‘, {‘k1‘:‘v1‘, ‘k2‘: ‘v2‘})
hget(name,key)
1 # 在name对应的hash中获取根据key获取value
hmget(name, keys, *args)
1 # 在name对应的hash中获取多个key的值 2 3 # 参数: 4 # name,reids对应的name 5 # keys,要获取key集合,如:[‘k1‘, ‘k2‘, ‘k3‘] 6 # *args,要获取的key,如:k1,k2,k3 7 8 # 如: 9 # r.mget(‘xx‘, [‘k1‘, ‘k2‘]) 10 # 或 11 # print r.hmget(‘xx‘, ‘k1‘, ‘k2‘)
hgetall(name)
1 # 获取name对应hash的所有键值
hlen(name)
1 2 # 获取name对应的hash中键值对的个数
hkeys(name)
1 # 获取name对应的hash中所有的key的值
hvals(name)
1 2 # 获取name对应的hash中所有的value的值
hexists(name, key)
1 # 检查name对应的hash是否存在当前传入的key
hdel(name,*keys)
1 2 # 将name对应的hash中指定key的键值对删除
hincrby(name, key, amount=1)
1 # 自增name对应的hash中的指定key的值,不存在则创建key=amount 2 # 参数: 3 # name,redis中的name 4 # key, hash对应的key 5 # amount,自增数(整数)
hincrbyfloat(name, key, amount=1.0)
1 # 自增name对应的hash中的指定key的值,不存在则创建key=amount 2 3 # 参数: 4 # name,redis中的name 5 # key, hash对应的key 6 # amount,自增数(浮点数) 7 8 # 自增name对应的hash中的指定key的值,不存在则创建key=amount
hscan(name, cursor=0, match=None, count=None)
1 # 增量式迭代获取,对于数据大的数据非常有用,hscan可以实现分片的获取数据,并非一次性将数据全部获取完,从而放置内存被撑爆 2 3 # 参数: 4 # name,redis的name 5 # cursor,游标(基于游标分批取获取数据) 6 # match,匹配指定key,默认None 表示所有的key 7 # count,每次分片最少获取个数,默认None表示采用Redis的默认分片个数 8 9 # 如: 10 # 第一次:cursor1, data1 = r.hscan(‘xx‘, cursor=0, match=None, count=None) 11 # 第二次:cursor2, data1 = r.hscan(‘xx‘, cursor=cursor1, match=None, count=None) 12 # ... 13 # 直到返回值cursor的值为0时,表示数据已经通过分片获取完毕
hscan_iter(name, match=None, count=None)
1 # 利用yield封装hscan创建生成器,实现分批去redis中获取数据 2 3 # 参数: 4 # match,匹配指定key,默认None 表示所有的key 5 # count,每次分片最少获取个数,默认None表示采用Redis的默认分片个数 6 7 # 如: 8 # for item in r.hscan_iter(‘xx‘): 9 # print item
List操作,redis中的List在在内存中按照一个name对应一个List来存储。如图:
lpush(name,values)
1 # 在name对应的list中添加元素,每个新的元素都添加到列表的最左边 2 3 # 如: 4 # r.lpush(‘oo‘, 11,22,33) 5 # 保存顺序为: 33,22,11 6 7 # 扩展: 8 # rpush(name, values) 表示从右向左操作
lpushx(name,value)
1 # 在name对应的list中添加元素,只有name已经存在时,值添加到列表的最左边 2 3 # 更多: 4 # rpushx(name, value) 表示从右向左操作
llen(name)
1 # name对应的list元素的个数
linsert(name, where, refvalue, value))
1 # 在name对应的列表的某一个值前或后插入一个新值 2 3 # 参数: 4 # name,redis的name 5 # where,BEFORE或AFTER 6 # refvalue,标杆值,即:在它前后插入数据 7 # value,要插入的数据
r.lset(name, index, value)
1 # 对name对应的list中的某一个索引位置重新赋值 2 3 # 参数: 4 # name,redis的name 5 # index,list的索引位置 6 # value,要设置的值
r.lrem(name, value, num)
1 # 在name对应的list中删除指定的值 2 3 # 参数: 4 # name,redis的name 5 # value,要删除的值 6 # num, num=0,删除列表中所有的指定值; 7 # num=2,从前到后,删除2个; 8 # num=-2,从后向前,删除2个
lpop(name)
1 # 在name对应的列表的左侧获取第一个元素并在列表中移除,返回值则是第一个元素 2 3 # 更多: 4 # rpop(name) 表示从右向左操作
lindex(name, index)
1 # 在name对应的列表中根据索引获取列表元素
lrange(name, start, end)
1 # 在name对应的列表分片获取数据 2 # 参数: 3 # name,redis的name 4 # start,索引的起始位置 5 # end,索引结束位置
ltrim(name, start, end)
1 # 在name对应的列表中移除没有在start-end索引之间的值 2 # 参数: 3 # name,redis的name 4 # start,索引的起始位置 5 # end,索引结束位置
rpoplpush(src, dst)
1 # 从一个列表取出最右边的元素,同时将其添加至另一个列表的最左边 2 # 参数: 3 # src,要取数据的列表的name 4 # dst,要添加数据的列表的name
blpop(keys, timeout)
1 # 将多个列表排列,按照从左到右去pop对应列表的元素 2 3 # 参数: 4 # keys,redis的name的集合 5 # timeout,超时时间,当元素所有列表的元素获取完之后,阻塞等待列表内有数据的时间(秒), 0 表示永远阻塞 6 7 # 更多: 8 # r.brpop(keys, timeout),从右向左获取数据
brpoplpush(src, dst, timeout=0)
1 # 从一个列表的右侧移除一个元素并将其添加到另一个列表的左侧 2 3 # 参数: 4 # src,取出并要移除元素的列表对应的name 5 # dst,要插入元素的列表对应的name 6 # timeout,当src对应的列表中没有数据时,阻塞等待其有数据的超时时间(秒),0 表示永远阻塞
自定义增量迭代
1 # 由于redis类库中没有提供对列表元素的增量迭代,如果想要循环name对应的列表的所有元素,那么就需要: 2 # 1、获取name对应的所有列表 3 # 2、循环列表 4 # 但是,如果列表非常大,那么就有可能在第一步时就将程序的内容撑爆,所有有必要自定义一个增量迭代的功能: 5 6 def list_iter(name): 7 """ 8 自定义redis列表增量迭代 9 :param name: redis中的name,即:迭代name对应的列表 10 :return: yield 返回 列表元素 11 """ 12 list_count = r.llen(name) 13 for index in xrange(list_count): 14 yield r.lindex(name, index) 15 16 # 使用 17 for item in list_iter(‘pp‘): 18 print item
Set操作,Set集合就是不允许重复的列表
sadd(name,values)
1 # name对应的集合中添加元素
scard(name)
1 # 获取name对应的集合中元素个数
sdiff(keys, *args)
1 # 在第一个name对应的集合中且不在其他name对应的集合的元素集合
sdiffstore(dest, keys, *args)
1 # 获取第一个name对应的集合中且不在其他name对应的集合,再将其新加入到dest对应的集合中
sinter(keys, *args)
1 # 获取多一个name对应集合的并集
sinterstore(dest, keys, *args)
1 # 获取多一个name对应集合的并集,再讲其加入到dest对应的集合中
sismember(name, value)
1 # 检查value是否是name对应的集合的成员
smembers(name)
1 # 获取name对应的集合的所有成员
smove(src, dst, value)
1 # 将某个成员从一个集合中移动到另外一个集合
spop(name)
1 # 从集合的右侧(尾部)移除一个成员,并将其返回
srandmember(name, numbers)
1 # 从name对应的集合中随机获取 numbers 个元素
srem(name, values)
1 # 在name对应的集合中删除某些值
sunion(keys, *args)
1 # 获取多一个name对应的集合的并集
sunionstore(dest,keys, *args)
1 # 获取多一个name对应的集合的并集,并将结果保存到dest对应的集合中
sscan(name, cursor=0, match=None, count=None)
sscan_iter(name, match=None, count=None)
1 # 同字符串的操作,用于增量迭代分批获取元素,避免内存消耗太大
有序集合,在集合的基础上,为每元素排序;元素的排序需要根据另外一个值来进行比较,所以,对于有序集合,每一个元素有两个值,即:值和分数,分数专门用来做排序。
zadd(name, *args, **kwargs)
1 # 在name对应的有序集合中添加元素 2 # 如: 3 # zadd(‘zz‘, ‘n1‘, 1, ‘n2‘, 2) 4 # 或 5 # zadd(‘zz‘, n1=11, n2=22)
zcard(name)
1 # 获取name对应的有序集合元素的数量
zcount(name, min, max)
1 # 获取name对应的有序集合中分数 在 [min,max] 之间的个数
zincrby(name, value, amount)
1 # 自增name对应的有序集合的 name 对应的分数
r.zrange( name, start, end, desc=False, withscores=False, score_cast_func=float)
1 # 按照索引范围获取name对应的有序集合的元素 2 3 # 参数: 4 # name,redis的name 5 # start,有序集合索引起始位置(非分数) 6 # end,有序集合索引结束位置(非分数) 7 # desc,排序规则,默认按照分数从小到大排序 8 # withscores,是否获取元素的分数,默认只获取元素的值 9 # score_cast_func,对分数进行数据转换的函数 10 11 # 更多: 12 # 从大到小排序 13 # zrevrange(name, start, end, withscores=False, score_cast_func=float) 14 15 # 按照分数范围获取name对应的有序集合的元素 16 # zrangebyscore(name, min, max, start=None, num=None, withscores=False, score_cast_func=float) 17 # 从大到小排序 18 # zrevrangebyscore(name, max, min, start=None, num=None, withscores=False, score_cast_func=float)
zrank(name, value)
1 # 获取某个值在 name对应的有序集合中的排行(从 0 开始) 2 3 # 更多: 4 # zrevrank(name, value),从大到小排序
zrangebylex(name, min, max, start=None, num=None)
1 # 当有序集合的所有成员都具有相同的分值时,有序集合的元素会根据成员的 值 (lexicographical ordering)来进行排序,而这个命令则可以返回给定的有序集合键 key 中, 元素的值介于 min 和 max 之间的成员 2 # 对集合中的每个成员进行逐个字节的对比(byte-by-byte compare), 并按照从低到高的顺序, 返回排序后的集合成员。 如果两个字符串有一部分内容是相同的话, 那么命令会认为较长的字符串比较短的字符串要大 3 4 # 参数: 5 # name,redis的name 6 # min,左区间(值)。 + 表示正无限; - 表示负无限; ( 表示开区间; [ 则表示闭区间 7 # min,右区间(值) 8 # start,对结果进行分片处理,索引位置 9 # num,对结果进行分片处理,索引后面的num个元素 10 11 # 如: 12 # ZADD myzset 0 aa 0 ba 0 ca 0 da 0 ea 0 fa 0 ga 13 # r.zrangebylex(‘myzset‘, "-", "[ca") 结果为:[‘aa‘, ‘ba‘, ‘ca‘] 14 15 # 更多: 16 # 从大到小排序 17 # zrevrangebylex(name, max, min, start=None, num=None)
zrem(name, values)
1 # 删除name对应的有序集合中值是values的成员 2 3 # 如:zrem(‘zz‘, [‘s1‘, ‘s2‘])
zremrangebyrank(name, min, max)
1 # 根据排行范围删除
zremrangebyscore(name, min, max)
1 # 根据分数范围删除
zremrangebylex(name, min, max)
1 # 根据值返回删除
zscore(name, value)
1 # 获取name对应有序集合中 value 对应的分数
zinterstore(dest, keys, aggregate=None)
1 # 获取两个有序集合的交集,如果遇到相同值不同分数,则按照aggregate进行操作 2 # aggregate的值为: SUM MIN MAX
zunionstore(dest, keys, aggregate=None)
1 # 获取两个有序集合的并集,如果遇到相同值不同分数,则按照aggregate进行操作 2 # aggregate的值为: SUM MIN MAX
zscan(name, cursor=0, match=None, count=None, score_cast_func=float)
zscan_iter(name, match=None, count=None,score_cast_func=float)
1 # 同字符串相似,相较于字符串新增score_cast_func,用来对分数进行操作
其他常用操作
delete(*names) 1 # 根据删除redis中的任意数据类型 exists(name) 1 2 # 检测redis的name是否存在 keys(pattern=‘*‘)
1 # 根据模型获取redis的name 2 3 # 更多: 4 # KEYS * 匹配数据库中所有 key 。 5 # KEYS h?llo 匹配 hello , hallo 和 hxllo 等。 6 # KEYS h*llo 匹配 hllo 和 heeeeello 等。 7 # KEYS h[ae]llo 匹配 hello 和 hallo ,但不匹配 hillo
expire(name ,time)
1 # 为某个redis的某个name设置超时时间
rename(src, dst)
1 # 对redis的name重命名为
move(name, db))
1 # 将redis的某个值移动到指定的db下
randomkey()
1 2 # 随机获取一个redis的name(不删除)
type(name)
1 # 获取name对应值的类型
scan(cursor=0, match=None, count=None)
scan_iter(match=None, count=None)
1 # 同字符串操作,用于增量迭代获取key
4、管道
redis-py默认在执行每次请求都会创建(连接池申请连接)和断开(归还连接池)一次连接操作,如果想要在一次请求中指定多个命令,则可以使用pipline实现一次请求指定多个命令,并且默认情况下一次pipline 是原子性操作。
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 import redis 5 6 pool = redis.ConnectionPool(host=‘10.211.55.4‘, port=6379) 7 8 r = redis.Redis(connection_pool=pool) 9 10 # pipe = r.pipeline(transaction=False) 11 pipe = r.pipeline(transaction=True) 12 13 pipe.set(‘name‘, ‘alex‘) 14 pipe.set(‘role‘, ‘sb‘) 15 16 pipe.execute()
5、发布订阅
发布者:服务器
订阅者:Dashboad和数据处理
Demo如下:
#!/usr/bin/env python # -*- coding:utf-8 -*- import redis class RedisHelper: def __init__(self): self.__conn = redis.Redis(host=‘10.211.55.4‘) self.chan_sub = ‘fm104.5‘ self.chan_pub = ‘fm104.5‘ def public(self, msg): self.__conn.publish(self.chan_pub, msg) return True def subscribe(self): pub = self.__conn.pubsub() pub.subscribe(self.chan_sub) pub.parse_response() return pub
RedisHelper
订阅者:
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 from monitor.RedisHelper import RedisHelper 5 6 obj = RedisHelper() 7 redis_sub = obj.subscribe() 8 9 while True: 10 msg= redis_sub.parse_response() 11 print msg
发布者:
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 4 from monitor.RedisHelper import RedisHelper 5 6 obj = RedisHelper() 7 obj.public(‘hello‘)
更多参见:https://github.com/andymccurdy/redis-py/
http://doc.redisfans.com/
pymsql
pymsql是Python中操作MySQL的模块,其使用方法和MySQLdb几乎相同。
一、下载安装:
1 pip3 install pymysql
二、使用
1、执行SQL
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import pymysql 4 5 # 创建连接 6 conn = pymysql.connect(host=‘127.0.0.1‘, port=3306, user=‘root‘, passwd=‘123‘, db=‘t1‘) 7 # 创建游标 8 cursor = conn.cursor() 9 10 # 执行SQL,并返回收影响行数 11 effect_row = cursor.execute("update hosts set host = ‘1.1.1.2‘") 12 13 # 执行SQL,并返回受影响行数 14 #effect_row = cursor.execute("update hosts set host = ‘1.1.1.2‘ where nid > %s", (1,)) 15 16 # 执行SQL,并返回受影响行数 17 #effect_row = cursor.executemany("insert into hosts(host,color_id)values(%s,%s)", [("1.1.1.11",1),("1.1.1.11",2)]) 18 19 20 # 提交,不然无法保存新建或者修改的数据 21 conn.commit() 22 23 # 关闭游标 24 cursor.close() 25 # 关闭连接 26 conn.close()
2、获取新创建数据自增ID
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import pymysql 4 5 conn = pymysql.connect(host=‘127.0.0.1‘, port=3306, user=‘root‘, passwd=‘123‘, db=‘t1‘) 6 cursor = conn.cursor() 7 cursor.executemany("insert into hosts(host,color_id)values(%s,%s)", [("1.1.1.11",1),("1.1.1.11",2)]) 8 conn.commit() 9 cursor.close() 10 conn.close() 11 12 # 获取最新自增ID 13 new_id = cursor.lastrowid
3、获取查询数据
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import pymysql 4 5 conn = pymysql.connect(host=‘127.0.0.1‘, port=3306, user=‘root‘, passwd=‘123‘, db=‘t1‘) 6 cursor = conn.cursor() 7 cursor.execute("select * from hosts") 8 9 # 获取第一行数据 10 row_1 = cursor.fetchone() 11 12 # 获取前n行数据 13 # row_2 = cursor.fetchmany(3) 14 # 获取所有数据 15 # row_3 = cursor.fetchall() 16 17 conn.commit() 18 cursor.close() 19 conn.close()
注:在fetch数据时按照顺序进行,可以使用cursor.scroll(num,mode)来移动游标位置,如:
- cursor.scroll(1,mode=‘relative‘) # 相对当前位置移动
- cursor.scroll(2,mode=‘absolute‘) # 相对绝对位置移动
4、fetch数据类型
关于默认获取的数据是元祖类型,如果想要或者字典类型的数据,即:
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import pymysql 4 5 conn = pymysql.connect(host=‘127.0.0.1‘, port=3306, user=‘root‘, passwd=‘123‘, db=‘t1‘) 6 7 # 游标设置为字典类型 8 cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) 9 r = cursor.execute("call p1()") 10 11 result = cursor.fetchone() 12 13 conn.commit() 14 cursor.close() 15 conn.close()
SQLAchemy
SQLAlchemy是Python编程语言下的一款ORM框架,该框架建立在数据库API之上,使用关系对象映射进行数据库操作,简言之便是:将对象转换成SQL,然后使用数据API执行SQL并获取执行结果。
SQLAlchemy本身无法操作数据库,其必须以来pymsql等第三方插件,Dialect用于和数据API进行交流,根据配置文件的不同调用不同的数据库API,从而实现对数据库的操作,如:
1 MySQL-Python 2 mysql+mysqldb://<user>:<password>@<host>[:<port>]/<dbname> 3 4 pymysql 5 mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>] 6 7 MySQL-Connector 8 mysql+mysqlconnector://<user>:<password>@<host>[:<port>]/<dbname> 9 10 cx_Oracle 11 oracle+cx_oracle://user:pass@host:port/dbname[?key=value&key=value...] 12 13 更多详见:http://docs.sqlalchemy.org/en/latest/dialects/index.html
一、底层处理
使用 Engine/ConnectionPooling/Dialect 进行数据库操作,Engine使用ConnectionPooling连接数据库,然后再通过Dialect执行SQL语句。
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 from sqlalchemy import create_engine 4 5 6 engine = create_engine("mysql+pymysql://root:[email protected]:3306/t1", max_overflow=5) 7 8 # 执行SQL 9 # cur = engine.execute( 10 # "INSERT INTO hosts (host, color_id) VALUES (‘1.1.1.22‘, 3)" 11 # ) 12 13 # 新插入行自增ID 14 # cur.lastrowid 15 16 # 执行SQL 17 # cur = engine.execute( 18 # "INSERT INTO hosts (host, color_id) VALUES(%s, %s)",[(‘1.1.1.22‘, 3),(‘1.1.1.221‘, 3),] 19 # ) 20 21 22 # 执行SQL 23 # cur = engine.execute( 24 # "INSERT INTO hosts (host, color_id) VALUES (%(host)s, %(color_id)s)", 25 # host=‘1.1.1.99‘, color_id=3 26 # ) 27 28 # 执行SQL 29 # cur = engine.execute(‘select * from hosts‘) 30 # 获取第一行数据 31 # cur.fetchone() 32 # 获取第n行数据 33 # cur.fetchmany(3) 34 # 获取所有数据 35 # cur.fetchall()
二、ORM功能使用
使用 ORM/Schema Type/SQL Expression Language/Engine/ConnectionPooling/Dialect 所有组件对数据进行操作。根据类创建对象,对象转换成SQL,执行SQL。
1、创建表
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 from sqlalchemy.ext.declarative import declarative_base 4 from sqlalchemy import Column, Integer, String, ForeignKey, UniqueConstraint, Index 5 from sqlalchemy.orm import sessionmaker, relationship 6 from sqlalchemy import create_engine 7 8 engine = create_engine("mysql+pymysql://root:[email protected]:3306/t1", max_overflow=5) 9 10 Base = declarative_base() 11 12 # 创建单表 13 class Users(Base): 14 __tablename__ = ‘users‘ 15 id = Column(Integer, primary_key=True) 16 name = Column(String(32)) 17 extra = Column(String(16)) 18 19 __table_args__ = ( 20 UniqueConstraint(‘id‘, ‘name‘, name=‘uix_id_name‘), 21 Index(‘ix_id_name‘, ‘name‘, ‘extra‘), 22 ) 23 24 25 # 一对多 26 class Favor(Base): 27 __tablename__ = ‘favor‘ 28 nid = Column(Integer, primary_key=True) 29 caption = Column(String(50), default=‘red‘, unique=True) 30 31 32 class Person(Base): 33 __tablename__ = ‘person‘ 34 nid = Column(Integer, primary_key=True) 35 name = Column(String(32), index=True, nullable=True) 36 favor_id = Column(Integer, ForeignKey("favor.nid")) 37 38 39 # 多对多 40 class Group(Base): 41 __tablename__ = ‘group‘ 42 id = Column(Integer, primary_key=True) 43 name = Column(String(64), unique=True, nullable=False) 44 port = Column(Integer, default=22) 45 46 47 class Server(Base): 48 __tablename__ = ‘server‘ 49 50 id = Column(Integer, primary_key=True, autoincrement=True) 51 hostname = Column(String(64), unique=True, nullable=False) 52 53 54 class ServerToGroup(Base): 55 __tablename__ = ‘servertogroup‘ 56 nid = Column(Integer, primary_key=True, autoincrement=True) 57 server_id = Column(Integer, ForeignKey(‘server.id‘)) 58 group_id = Column(Integer, ForeignKey(‘group.id‘)) 59 60 61 def init_db(): 62 Base.metadata.create_all(engine) 63 64 65 def drop_db(): 66 Base.metadata.drop_all(engine)
ForeignKeyConstraint([‘other_id‘], [‘othertable.other_id‘]),
2、操作表
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 from sqlalchemy.ext.declarative import declarative_base 4 from sqlalchemy import Column, Integer, String, ForeignKey, UniqueConstraint, Index 5 from sqlalchemy.orm import sessionmaker, relationship 6 from sqlalchemy import create_engine 7 8 engine = create_engine("mysql+pymysql://root:[email protected]:3306/t1", max_overflow=5) 9 10 Base = declarative_base() 11 12 # 创建单表 13 class Users(Base): 14 __tablename__ = ‘users‘ 15 id = Column(Integer, primary_key=True) 16 name = Column(String(32)) 17 extra = Column(String(16)) 18 19 __table_args__ = ( 20 UniqueConstraint(‘id‘, ‘name‘, name=‘uix_id_name‘), 21 Index(‘ix_id_name‘, ‘name‘, ‘extra‘), 22 ) 23 24 def __repr__(self): 25 return "%s-%s" %(self.id, self.name) 26 27 # 一对多 28 class Favor(Base): 29 __tablename__ = ‘favor‘ 30 nid = Column(Integer, primary_key=True) 31 caption = Column(String(50), default=‘red‘, unique=True) 32 33 def __repr__(self): 34 return "%s-%s" %(self.nid, self.caption) 35 36 class Person(Base): 37 __tablename__ = ‘person‘ 38 nid = Column(Integer, primary_key=True) 39 name = Column(String(32), index=True, nullable=True) 40 favor_id = Column(Integer, ForeignKey("favor.nid")) 41 # 与生成表结构无关,仅用于查询方便 42 favor = relationship("Favor", backref=‘pers‘) 43 44 # 多对多 45 class ServerToGroup(Base): 46 __tablename__ = ‘servertogroup‘ 47 nid = Column(Integer, primary_key=True, autoincrement=True) 48 server_id = Column(Integer, ForeignKey(‘server.id‘)) 49 group_id = Column(Integer, ForeignKey(‘group.id‘)) 50 group = relationship("Group", backref=‘s2g‘) 51 server = relationship("Server", backref=‘s2g‘) 52 53 class Group(Base): 54 __tablename__ = ‘group‘ 55 id = Column(Integer, primary_key=True) 56 name = Column(String(64), unique=True, nullable=False) 57 port = Column(Integer, default=22) 58 # group = relationship(‘Group‘,secondary=ServerToGroup,backref=‘host_list‘) 59 60 61 class Server(Base): 62 __tablename__ = ‘server‘ 63 64 id = Column(Integer, primary_key=True, autoincrement=True) 65 hostname = Column(String(64), unique=True, nullable=False) 66 67 68 69 70 def init_db(): 71 Base.metadata.create_all(engine) 72 73 74 def drop_db(): 75 Base.metadata.drop_all(engine) 76 77 78 Session = sessionmaker(bind=engine) 79 session = Session() 80 81 表结构+连接数据库
表结构+连接数据库
?增
1 obj = Users(name="alex0", extra=‘sb‘) 2 session.add(obj) 3 session.add_all([ 4 Users(name="alex1", extra=‘sb‘), 5 Users(name="alex2", extra=‘sb‘), 6 ]) 7 session.commit()
?删
1 session.query(Users).filter(Users.id > 2).delete() 2 session.commit()
?改
1 session.query(Users).filter(Users.id > 2).update({"name" : "099"}) 2 session.query(Users).filter(Users.id > 2).update({Users.name: Users.name + "099"}, synchronize_session=False) 3 session.query(Users).filter(Users.id > 2).update({"num": Users.num + 1}, synchronize_session="evaluate") 4 session.commit()
?查
1 ret = session.query(Users).all() 2 ret = session.query(Users.name, Users.extra).all() 3 ret = session.query(Users).filter_by(name=‘alex‘).all() 4 ret = session.query(Users).filter_by(name=‘alex‘).first()
?其他
1 # 条件 2 ret = session.query(Users).filter_by(name=‘alex‘).all() 3 ret = session.query(Users).filter(Users.id > 1, Users.name == ‘eric‘).all() 4 ret = session.query(Users).filter(Users.id.between(1, 3), Users.name == ‘eric‘).all() 5 ret = session.query(Users).filter(Users.id.in_([1,3,4])).all() 6 ret = session.query(Users).filter(~Users.id.in_([1,3,4])).all() 7 ret = session.query(Users).filter(Users.id.in_(session.query(Users.id).filter_by(name=‘eric‘))).all() 8 from sqlalchemy import and_, or_ 9 ret = session.query(Users).filter(and_(Users.id > 3, Users.name == ‘eric‘)).all() 10 ret = session.query(Users).filter(or_(Users.id < 2, Users.name == ‘eric‘)).all() 11 ret = session.query(Users).filter( 12 or_( 13 Users.id < 2, 14 and_(Users.name == ‘eric‘, Users.id > 3), 15 Users.extra != "" 16 )).all() 17 18 19 # 通配符 20 ret = session.query(Users).filter(Users.name.like(‘e%‘)).all() 21 ret = session.query(Users).filter(~Users.name.like(‘e%‘)).all() 22 23 # 限制 24 ret = session.query(Users)[1:2] 25 26 # 排序 27 ret = session.query(Users).order_by(Users.name.desc()).all() 28 ret = session.query(Users).order_by(Users.name.desc(), Users.id.asc()).all() 29 30 # 分组 31 from sqlalchemy.sql import func 32 33 ret = session.query(Users).group_by(Users.extra).all() 34 ret = session.query( 35 func.max(Users.id), 36 func.sum(Users.id), 37 func.min(Users.id)).group_by(Users.name).all() 38 39 ret = session.query( 40 func.max(Users.id), 41 func.sum(Users.id), 42 func.min(Users.id)).group_by(Users.name).having(func.min(Users.id) >2).all() 43 44 # 连表 45 46 ret = session.query(Users, Favor).filter(Users.id == Favor.nid).all() 47 48 ret = session.query(Person).join(Favor).all() 49 50 ret = session.query(Person).join(Favor, isouter=True).all() 51 52 53 # 组合 54 q1 = session.query(Users.name).filter(Users.id > 2) 55 q2 = session.query(Favor.caption).filter(Favor.nid < 2) 56 ret = q1.union(q2).all() 57 58 q1 = session.query(Users.name).filter(Users.id > 2) 59 q2 = session.query(Favor.caption).filter(Favor.nid < 2) 60 ret = q1.union_all(q2).all()
更多功能参见文档,猛击这里下载PDF
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 from sqlalchemy import create_engine,and_,or_,func,Table 4 from sqlalchemy.ext.declarative import declarative_base 5 from sqlalchemy import Column, Integer, String,ForeignKey,UniqueConstraint,DateTime 6 from sqlalchemy.orm import sessionmaker,relationship 7 8 Base = declarative_base() #生成一个SqlORM 基类 9 10 # 服务器账号和组 11 # HostUser2Group = Table(‘hostuser_2_group‘,Base.metadata, 12 # Column(‘hostuser_id‘,ForeignKey(‘host_user.id‘),primary_key=True), 13 # Column(‘group_id‘,ForeignKey(‘group.id‘),primary_key=True), 14 # ) 15 16 # 用户和组关系表,用户可以属于多个组,一个组可以有多个人 17 UserProfile2Group = Table(‘userprofile_2_group‘,Base.metadata, 18 Column(‘userprofile_id‘,ForeignKey(‘user_profile.id‘),primary_key=True), 19 Column(‘group_id‘,ForeignKey(‘group.id‘),primary_key=True), 20 ) 21 22 # 程序登陆用户和服务器账户,一个人可以有多个服务器账号,一个服务器账号可以给多个人用 23 UserProfile2HostUser= Table(‘userprofile_2_hostuser‘,Base.metadata, 24 Column(‘userprofile_id‘,ForeignKey(‘user_profile.id‘),primary_key=True), 25 Column(‘hostuser_id‘,ForeignKey(‘host_user.id‘),primary_key=True), 26 ) 27 28 29 class Host(Base): 30 __tablename__=‘host‘ 31 id = Column(Integer,primary_key=True,autoincrement=True) 32 hostname = Column(String(64),unique=True,nullable=False) 33 ip_addr = Column(String(128),unique=True,nullable=False) 34 port = Column(Integer,default=22) 35 def __repr__(self): 36 return "<id=%s,hostname=%s, ip_addr=%s>" %(self.id, 37 self.hostname, 38 self.ip_addr) 39 40 41 class HostUser(Base): 42 __tablename__ = ‘host_user‘ 43 id = Column(Integer,primary_key=True) 44 AuthTypes = [ 45 (u‘ssh-passwd‘,u‘SSH/Password‘), 46 (u‘ssh-key‘,u‘SSH/KEY‘), 47 ] 48 # auth_type = Column(ChoiceType(AuthTypes)) 49 auth_type = Column(String(64)) 50 username = Column(String(64),unique=True,nullable=False) 51 password = Column(String(255)) 52 53 host_id = Column(Integer,ForeignKey(‘host.id‘)) 54 55 # groups = relationship(‘Group‘, 56 # secondary=HostUser2Group, 57 # backref=‘host_list‘) 58 59 __table_args__ = (UniqueConstraint(‘host_id‘,‘username‘, name=‘_host_username_uc‘),) 60 61 def __repr__(self): 62 return "<id=%s,name=%s>" %(self.id,self.username) 63 64 65 class Group(Base): 66 __tablename__ = ‘group‘ 67 id = Column(Integer,primary_key=True) 68 name = Column(String(64),unique=True,nullable=False) 69 def __repr__(self): 70 return "<id=%s,name=%s>" %(self.id,self.name) 71 72 73 class UserProfile(Base): 74 __tablename__ = ‘user_profile‘ 75 id = Column(Integer,primary_key=True) 76 username = Column(String(64),unique=True,nullable=False) 77 password = Column(String(255),nullable=False) 78 # host_list = relationship(‘HostUser‘, 79 # secondary=UserProfile2HostUser, 80 # backref=‘userprofiles‘) 81 # groups = relationship(‘Group‘, 82 # secondary=UserProfile2Group, 83 # backref=‘userprofiles‘) 84 def __repr__(self): 85 return "<id=%s,name=%s>" %(self.id,self.username) 86 87 88 class AuditLog(Base): 89 __tablename__ = ‘audit_log‘ 90 id = Column(Integer,primary_key=True) 91 userprofile_id = Column(Integer,ForeignKey(‘user_profile.id‘)) 92 hostuser_id = Column(Integer,ForeignKey(‘host_user.id‘)) 93 action_choices2 = [ 94 (u‘cmd‘,u‘CMD‘), 95 (u‘login‘,u‘Login‘), 96 (u‘logout‘,u‘Logout‘), 97 ] 98 action_type = Column(ChoiceType(action_choices2)) 99 #action_type = Column(String(64)) 100 cmd = Column(String(255)) 101 date = Column(DateTime) 102 103 # user_profile = relationship("UserProfile") 104 #bind_host = relationship("BindHost") 105 106 107 engine = create_engine("mysql+pymsql://root:[email protected]:3306/stupid_jumpserver",echo=False) 108 Base.metadata.create_all(engine) #创建所有表结构
表结构操作练习
Paramiko
paramiko模块,基于SSH用于连接远程服务器并执行相关操作。
一、安装
1 pip3 install paramiko
二、使用
SSHClient
用于连接远程服务器并执行基本命令
基于用户名密码连接:
1 import paramiko 2 3 # 创建SSH对象 4 ssh = paramiko.SSHClient() 5 # 允许连接不在know_hosts文件中的主机 6 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 7 # 连接服务器 8 ssh.connect(hostname=‘c1.salt.com‘, port=22, username=‘wupeiqi‘, password=‘123‘) 9 10 # 执行命令 11 stdin, stdout, stderr = ssh.exec_command(‘ls‘) 12 # 获取命令结果 13 result = stdout.read() 14 15 # 关闭连接 16 ssh.close()
1 import paramiko 2 3 transport = paramiko.Transport((‘hostname‘, 22)) 4 transport.connect(username=‘wupeiqi‘, password=‘123‘) 5 6 ssh = paramiko.SSHClient() 7 ssh._transport = transport 8 9 stdin, stdout, stderr = ssh.exec_command(‘df‘) 10 print stdout.read() 11 12 transport.close()
SSHClient 封装 Transport
基于公钥密钥连接:
1 import paramiko 2 3 private_key = paramiko.RSAKey.from_private_key_file(‘/home/auto/.ssh/id_rsa‘) 4 5 # 创建SSH对象 6 ssh = paramiko.SSHClient() 7 # 允许连接不在know_hosts文件中的主机 8 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 9 # 连接服务器 10 ssh.connect(hostname=‘c1.salt.com‘, port=22, username=‘wupeiqi‘, key=private_key) 11 12 # 执行命令 13 stdin, stdout, stderr = ssh.exec_command(‘df‘) 14 # 获取命令结果 15 result = stdout.read() 16 17 # 关闭连接 18 ssh.close()
1 import paramiko 2 3 private_key = paramiko.RSAKey.from_private_key_file(‘/home/auto/.ssh/id_rsa‘) 4 5 transport = paramiko.Transport((‘hostname‘, 22)) 6 transport.connect(username=‘wupeiqi‘, pkey=private_key) 7 8 ssh = paramiko.SSHClient() 9 ssh._transport = transport 10 11 stdin, stdout, stderr = ssh.exec_command(‘df‘) 12 13 transport.close()
SSHClient 封装 Transport
SFTPClient
用于连接远程服务器并执行上传下载
基于用户名密码上传下载:
1 import paramiko 2 3 transport = paramiko.Transport((‘hostname‘,22)) 4 transport.connect(username=‘wupeiqi‘,password=‘123‘) 5 6 sftp = paramiko.SFTPClient.from_transport(transport) 7 # 将location.py 上传至服务器 /tmp/test.py 8 sftp.put(‘/tmp/location.py‘, ‘/tmp/test.py‘) 9 # 将remove_path 下载到本地 local_path 10 sftp.get(‘remove_path‘, ‘local_path‘) 11 12 transport.close()
基于公钥密钥上传下载:
1 import paramiko 2 3 private_key = paramiko.RSAKey.from_private_key_file(‘/home/auto/.ssh/id_rsa‘) 4 5 transport = paramiko.Transport((‘hostname‘, 22)) 6 transport.connect(username=‘wupeiqi‘, pkey=private_key ) 7 8 sftp = paramiko.SFTPClient.from_transport(transport) 9 # 将location.py 上传至服务器 /tmp/test.py 10 sftp.put(‘/tmp/location.py‘, ‘/tmp/test.py‘) 11 # 将remove_path 下载到本地 local_path 12 sftp.get(‘remove_path‘, ‘local_path‘) 13 14 transport.close()
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 import paramiko 4 import uuid 5 6 class SSHConnection(object): 7 8 def __init__(self, host=‘172.16.103.191‘, port=22, username=‘wupeiqi‘,pwd=‘123‘): 9 self.host = host 10 self.port = port 11 self.username = username 12 self.pwd = pwd 13 self.__k = None 14 15 def create_file(self): 16 file_name = str(uuid.uuid4()) 17 with open(file_name,‘w‘) as f: 18 f.write(‘sb‘) 19 return file_name 20 21 def run(self): 22 self.connect() 23 self.upload(‘/home/wupeiqi/tttttttttttt.py‘) 24 self.rename(‘/home/wupeiqi/tttttttttttt.py‘, ‘/home/wupeiqi/ooooooooo.py) 25 self.close() 26 27 def connect(self): 28 transport = paramiko.Transport((self.host,self.port)) 29 transport.connect(username=self.username,password=self.pwd) 30 self.__transport = transport 31 32 def close(self): 33 34 self.__transport.close() 35 36 def upload(self,target_path): 37 # 连接,上传 38 file_name = self.create_file() 39 40 sftp = paramiko.SFTPClient.from_transport(self.__transport) 41 # 将location.py 上传至服务器 /tmp/test.py 42 sftp.put(file_name, target_path) 43 44 def rename(self, old_path, new_path): 45 46 ssh = paramiko.SSHClient() 47 ssh._transport = self.__transport 48 # 执行命令 49 cmd = "mv %s %s" % (old_path, new_path,) 50 stdin, stdout, stderr = ssh.exec_command(cmd) 51 # 获取命令结果 52 result = stdout.read() 53 54 def cmd(self, command): 55 ssh = paramiko.SSHClient() 56 ssh._transport = self.__transport 57 # 执行命令 58 stdin, stdout, stderr = ssh.exec_command(command) 59 # 获取命令结果 60 result = stdout.read() 61 return result 62 63 64 65 ha = SSHConnection() 66 ha.run()
Demo
1 # 对于更多限制命令,需要在系统中设置 2 /etc/sudoers 3 4 Defaults requiretty 5 Defaults:cmdb !requiretty
1 import paramiko 2 import uuid 3 4 class SSHConnection(object): 5 6 def __init__(self, host=‘192.168.11.61‘, port=22, username=‘alex‘,pwd=‘alex3714‘): 7 self.host = host 8 self.port = port 9 self.username = username 10 self.pwd = pwd 11 self.__k = None 12 13 def run(self): 14 self.connect() 15 pass 16 self.close() 17 18 def connect(self): 19 transport = paramiko.Transport((self.host,self.port)) 20 transport.connect(username=self.username,password=self.pwd) 21 self.__transport = transport 22 23 def close(self): 24 self.__transport.close() 25 26 def cmd(self, command): 27 ssh = paramiko.SSHClient() 28 ssh._transport = self.__transport 29 # 执行命令 30 stdin, stdout, stderr = ssh.exec_command(command) 31 # 获取命令结果 32 result = stdout.read() 33 return result 34 35 def upload(self,local_path, target_path): 36 # 连接,上传 37 sftp = paramiko.SFTPClient.from_transport(self.__transport) 38 # 将location.py 上传至服务器 /tmp/test.py 39 sftp.put(local_path, target_path) 40 41 ssh = SSHConnection() 42 ssh.connect() 43 r1 = ssh.cmd(‘df‘) 44 ssh.upload(‘s2.py‘, "/home/alex/s7.py") 45 ssh.close()
Demo