标签：
import contextlib
@contextlib.contextmanager
def worker_state(state_list,worker_thread):
"""
用于记录线程中正在等待的线程数
"""
state_list.append(worker_thread)
try:
yield
finally:
state_list.remove(worker_thread)
free_list = []
current_thread = ‘alex‘
with worker_state(free_list,current_thread):
print(123)
print(456)
# socket 使用contestlib自动关闭
import contextlib
import socket
@contextlib.contextmanager
def context_socket(host,port):
sk = socket.socket()
sk.bind((host,port))
sk.listen(5)
try:
yield sk
finally:
sk.close()
with context_socket(‘127.0.0.1‘,8888) as sock:
print(sock)
1. REDIS订阅、发布、频道
-----s3.py #不用执行
import redis
class RedisHelper:
def __init__(self):
self.__conn = redis.Redis(host=‘127.0.0.1‘)
def public(self,msg,chan):
self.__conn.publish(chan,msg)
return True
def subscribe(self,chan):
pub = self.__conn.pubsub()
pub.subscribe(chan)
pub.parse_response()
return pub
# 订阅,制定频道wmh.7
-----s4.py
import s3
obj = s3.RedisHelper()
data = obj.subscribe(‘wmh.7‘)
print(data.parse_response())
# 发布
-----s5.py
import s3
obj = s3.RedisHelper()
obj.public(‘alex db‘,‘wmh.7‘) #发给频道“wmh.7”信息
------执行s4.py等待收，执行s5.py发送，s4收结果如下
[b‘message‘, b‘wmh.7‘, b‘alex db‘]
2. rabbitmq 发收
# 安装pika
发 # Queues
import pika
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘172.16.0.2‘))
channel = connection.channel()
channel.queue_declare(queue=‘hello‘)
channel.basic_publish(exchange=‘‘,routing_key=‘hello‘,body=‘Hello World‘)
print(" [x] Sent ‘Hello World!‘ ")
connection.close()
读 #多发几次 消费者Consumers
import pika
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘127.0.0.1‘))
channel = connection.channel()
channel.queue_declare(queue=‘hello1‘)
def callback(ch,method,properties,body):
print(" [x] Received %r" % body)
channel.basic_consume(callback,
queue=‘hello1‘,
no_ack=True)
print(‘ [*] Waiting for messages. To exit press CTRL+C‘)
channel.start_consuming()
在收上查看信息结果
[*] Waiting for messages. To exit press CTRL+C
[x] Received b‘Hello World!‘
[x] Received b‘Hello World!‘
绑定多个队列，exchange调度，随机生成队列,
往exchange发数据,只要是订阅者都能同时收到
--------8.py 订阅者 运行两或多次，自动创建队列
import pika
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘172.16.0.2‘))
channel = connection.channel()
channel.exchange_declare(exchange=‘logs_fanout‘,type=‘fanout‘)
# 随机创建队列
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
# 绑定
channel.queue_bind(exchange=‘logs_fanout‘,queue=queue_name)
print(‘ [*] Waiting for logs. To exit press CTRL+C‘)
def callback(ch, method, properties, body):
print(" [x] %r" % body)
channel.basic_consume(callback,queue=queue_name,no_ack=True)
channel.start_consuming()
---------9.py 发布者
import pika
import sys
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘172.16.0.2‘))
channel = connection.channel()
channel.exchange_declare(exchange=‘logs_fanout‘,type=‘fanout‘)
message = ‘456‘
channel.basic_publish(exchange=‘logs_fanout‘,routing_key=‘‘,body=message)
print(" [x] Sent %r" % message)
connection.close()
关键字发送
--------消费者
import pika
import sys
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘172.16.0.2‘))
channel = connection.channel()
channel.exchange_declare(exchange=‘direct_logs‘,type=‘direct‘)
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
severities = [‘error‘,‘info‘,‘warning‘] #定义三个关键字
# severities = sys.argv[1:]
# if not severities:
# sys.stderr.write("Usage: %s [info] [warning] [error]\n" % sys.argv[0])
# sys.exit(1)
for severity in severities:
channel.queue_bind(exchange=‘direct_logs‘,queue=queue_name,routing_key=severity)
print(‘ [*] Waiting for logs. To exit press CTRL+C‘)
def callback(ch, method, properties, body):
print(" [x] %r:%r" % (method.routing_key, body))
channel.basic_consume(callback,queue=queue_name,no_ack=True)
channel.start_consuming()
-------发布者1
import pika
import sys
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘172.16.0.2‘))
channel = connection.channel()
channel.exchange_declare(exchange=‘direct_logs‘,type=‘direct‘)
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
severities = [‘error‘,]
# severities = sys.argv[1:]
# if not severities:
# sys.stderr.write("Usage: %s [info] [warning] [error]\n" % sys.argv[0])
# sys.exit(1)
for severity in severities:
channel.queue_bind(exchange=‘direct_logs‘,queue=queue_name,routing_key=severity)
print(‘ [*] Waiting for logs. To exit press CTRL+C‘)
def callback(ch, method, properties, body):
print(" [x] %r:%r" % (method.routing_key, body))
channel.basic_consume(callback,queue=queue_name,no_ack=True)
channel.start_consuming()
-------发布者2
import pika
import sys
connection = pika.BlockingConnection(pika.ConnectionParameters(host=‘localhost‘))
channel = connection.channel()
channel.exchange_declare(exchange=‘direct_logs‘,type=‘direct‘)
severity = ‘error‘ #一个关键字
message=‘123‘
channel.basic_publish(exchange=‘direct_logs‘,routing_key=severity,body=message)
print(" [x] Sent %r:%r" % (severity, message))
connection.close()
操作MYSQL
python pymysql
http://www.cnblogs.com/wupeiqi/articles/5699254.html
create table tb1(
nid int not null auto_increment primary key,
num int null
)
或
create table tb1(
nid int not null auto_increment,
num int null,
index(nid)
)
注意：1、对于自增列，必须是索引（含主键）。
2、对于自增可以设置步长和起始值
show session variables like ‘auto_inc%‘;
set session auto_increment_increment=2;
set session auto_increment_offset=10;
shwo global variables like ‘auto_inc%‘;
set global auto_increment_increment=2;
set global auto_increment_offset=10;
自动创建数据库结构 sqlalchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey, UniqueConstraint, Index
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy import create_engine
engine = create_engine("mysql+pymysql://root:123456@127.0.0.1:3306/s13", max_overflow=5)
Base = declarative_base()
# 创建单表
class Users(Base):
__tablename__ = ‘users‘
id = Column(Integer, primary_key=True)
name = Column(String(32))
extra = Column(String(16))
__table_args__ = (
UniqueConstraint(‘id‘, ‘name‘, name=‘uix_id_name‘),
Index(‘ix_id_name‘, ‘name‘, ‘extra‘),
)
# 一对多
class Favor(Base):
__tablename__ = ‘favor‘
nid = Column(Integer, primary_key=True)
caption = Column(String(50), default=‘red‘, unique=True)
class Person(Base):
__tablename__ = ‘person‘
nid = Column(Integer, primary_key=True)
name = Column(String(32), index=True, nullable=True)
favor_id = Column(Integer, ForeignKey("favor.nid"))
# 多对多
class ServerToGroup(Base):
__tablename__ = ‘servertogroup‘
nid = Column(Integer, primary_key=True, autoincrement=True)
server_id = Column(Integer, ForeignKey(‘server.id‘))
group_id = Column(Integer, ForeignKey(‘group.id‘))
class Group(Base):
__tablename__ = ‘group‘
id = Column(Integer, primary_key=True)
name = Column(String(64), unique=True, nullable=False)
class Server(Base):
__tablename__ = ‘server‘
id = Column(Integer, primary_key=True, autoincrement=True)
hostname = Column(String(64), unique=True, nullable=False)
port = Column(Integer, default=22)
Base.metadata.create_all(engine) #创建数据库
#Base.metadata.drop_all(engine) #删除数据库
SSH操作
http://www.cnblogs.com/wupeiqi/articles/5095821.html
# 用于连接远程服务器并执行基本命令
import paramiko
ssh = paramiko.SSHClient() # 创建SSH对象
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 允许连接不在know_hosts文件中的主机
ssh.connect(hostname=‘172.16.0.2‘, port=22, username=‘root‘, password=‘pwd@123‘) # 连接服务器
stdin, stdout, stderr = ssh.exec_command(‘ifconfig‘) # 执行命令
out = stdout.readlines()
for i in out:
print(i)
result = stdout.read() # 获取命令结果
ssh.close() # 关闭连接
# SSHClient 封装 Transport
import paramiko
transport = paramiko.Transport((‘172.16.0.2‘, 22))
transport.connect(username=‘root‘, password=‘pwd@123‘)
ssh = paramiko.SSHClient()
ssh._transport = transport
stdin, stdout, stderr = ssh.exec_command(‘df‘)
out = stdout.readlines()
for i in out:
print(i)
print(stdout.read())
transport.close()
规范化ssh远程
import paramiko
hostname="172.16.0.2"
port=22
username="root"
password="pwd@123"
if __name__=="__main__":
s=paramiko.SSHClient()
s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
s.connect(hostname,port,username,password)
stdin,stdout,sterr=s.exec_command("df -h")
for i in stdout:
print(i)
# print(stdout.read())
s.close()
# 基于公钥密钥连接
import paramiko
private_key = paramiko.RSAKey.from_private_key_file(‘/home/auto/.ssh/id_rsa‘)
ssh = paramiko.SSHClient() # 创建SSH对象
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 允许连接不在know_hosts文件中的主机
ssh.connect(hostname=‘c1.salt.com‘, port=22, username=‘wupeiqi‘, key=private_key) # 连接服务器
stdin, stdout, stderr = ssh.exec_command(‘df‘) # 执行命令
result = stdout.read() # 获取命令结果
ssh.close() # 关闭连接
# 基于用户名密码上传下载
import paramiko
transport = paramiko.Transport((‘hostname‘,22))
transport.connect(username=‘wupeiqi‘,password=‘123‘)
sftp = paramiko.SFTPClient.from_transport(transport)
sftp.put(‘/tmp/location.py‘, ‘/tmp/test.py‘) # 将location.py 上传至服务器 /tmp/test.py
sftp.get(‘remove_path‘, ‘local_path‘) # 将remove_path 下载到本地 local_path
transport.close()
# 基于公钥密钥上传下载
import paramiko
private_key = paramiko.RSAKey.from_private_key_file(‘/home/auto/.ssh/id_rsa‘)
transport = paramiko.Transport((‘hostname‘, 22))
transport.connect(username=‘wupeiqi‘, pkey=private_key )
sftp = paramiko.SFTPClient.from_transport(transport)
sftp.put(‘/tmp/location.py‘, ‘/tmp/test.py‘) # 将location.py 上传至服务器 /tmp/test.py
sftp.get(‘remove_path‘, ‘local_path‘) # 将remove_path 下载到本地 local_path
transport.close()
多线程综合ssh程序(多机、执行命令、上传、下载)
import datetime
import threading
import paramiko
def sshCmd(ip, username, passwd, cmds):
try:
ssh = paramiko.SSHClient() # 创建SSH对象
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 允许连接不在know_hosts文件中的主机
ssh.connect(hostname=ip, port=22, username=username, password=passwd,timeout=5) # 连接服务器
for cmd in cmds:
stdin, stdout, stderr = ssh.exec_command(cmd) # 执行命令
out = stdout.readlines()
for i in out:
print(i,)
print(‘%s\t 运行完毕\r\n‘ % (ip))
except Exception as e:
print(‘%s\t 运行失败,失败原因\r\n%s‘ % (ip, e))
finally:
ssh.close()
#上传文件
def uploadFile(ip,username,passwd):
try:
t=paramiko.Transport((ip,22))
t.connect(username=username,password=passwd)
sftp=paramiko.SFTPClient.from_transport(t)
remotepath=‘/opt/1.txt‘
localpath=r‘D:\Downloads\1.txt‘
sftp.put(localpath,remotepath)
print(‘上传文件成功‘)
except Exception as e:
print(‘%s\t 运行失败,失败原因\r\n%s‘ % (ip, e))
finally:
t.close()
#下载文件
def downloadFile(ip,username,passwd):
try:
t=paramiko.Transport((ip,22))
t.connect(username=username,password=passwd)
sftp=paramiko.SFTPClient.from_transport(t)
remotepath=‘/opt/1.py‘
localpath=‘1.py‘
sftp.get(remotepath,localpath)
print(‘下载文件成功‘)
except Exception as e:
print(‘%s\t 运行失败,失败原因\r\n%s‘ % (ip, e))
finally:
t.close()
if __name__ == ‘__main__‘:
# 需要执行的命令列表
cmds = [‘ls /root‘, ‘ifconfig‘]
# 需要进行远程监控的服务器列表
servers = [‘172.16.0.2‘]
username = "root"
passwd = "pwd@123"
threads = []
print("程序开始运行%s" % datetime.datetime.now())
# 每一台服务器创建一个线程处理
for server in servers:
th = threading.Thread(target=sshCmd, args=(server, username, passwd, cmds))
th.start()
threads.append(th)
# 等待线程运行完毕
for th in threads:
th.join()
print("程序结束运行%s" % datetime.datetime.now())
#测试文件的上传与下载
uploadFile(servers[0],username,passwd)
downloadFile(servers[0],username,passwd)
# pexpect自动输入密码执行命令
import pexpect
def sshCmd(ip, passwd, cmd):
ret = -1
ssh = pexpect.spawn(‘ssh root@%s "%s"‘ % (ip, cmd))
try:
i = ssh.expect([‘password:‘, ‘continue connecting(yes/no)?‘], timeout=5)
if i == 0:
ssh.sendline(passwd)
elif i == 1:
ssh.sendline(‘yes\n‘)
ssh.expect(‘password:‘)
ssh.sendline(passwd)
ssh.sendline(cmd)
r = ssh.read()
print(r)
ret = 0
except pexpect.EOF:
print("EOF")
ret = -1
except pexpect.TIMEOUT:
print("TIMEOUT")
ret = -2
finally:
ssh.close()
return ret
sshCmd(‘172.16.0.2‘,‘pwd@123‘,‘ls /root‘)
视频
http://edu.51cto.com/course/course_id-5686.html
---------------------------------------------------------------------------------------------------------------------------------
#!/usr/bin/env python
#-*- coding:utf-8 -*-
# Author:Minghu Wang
# 导入:
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey, UniqueConstraint, Index
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy import create_engine
# 创建对象的基类:
Base = declarative_base()
# 定义User对象:
class User(Base):
# 表的名字:
__tablename__ = ‘hosts‘
# 表的结构:
id = Column(Integer, primary_key=True, autoincrement=True)
name = Column(String(20))
ip = Column(String(32))
password = Column(String(16))
port = Column(String(16))
business = Column(Integer, ForeignKey("favor.nid"))
class Favor(Base):
__tablename__ = ‘favor‘
nid = Column(Integer, primary_key=True,autoincrement=True)
caption = Column(String(50), default=‘red‘, unique=True)
# 初始化数据库连接:
engine = create_engine("mysql+pymysql://root:123456@192.168.1.103:3306/s13?charset=utf8", max_overflow=5)
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)
# 创建session对象:
session = DBSession()
# # 创建新User对象:
# new_user = User(name=‘DNS‘,ip=‘192.168.1.100‘,password=‘pwd@123‘,port=‘22‘,business=‘1‘)
# # 添加到session:
# session.add(new_user)
# # 提交即保存到数据库:
# session.commit()
# # 关闭session:
# session.close()
# 插入数据
obj = User(name=‘DNS‘,ip=‘192.168.1.100‘,password=‘pwd@123‘,port=‘22‘,business=‘1‘)
session.add(obj)
session.add_all([
Favor(caption=‘kfjsf‘),
User(name=‘DNS‘,ip=‘192.168.1.100‘,password=‘pwd@123‘,port=‘22‘,business=‘1‘),
User(name=‘DNS‘,ip=‘192.168.1.100‘,password=‘pwd@123‘,port=‘22‘,business=‘1‘),
])
session.commit()
# 查询
ret = session.query(User.id, User.name,User.ip,User.password,User.port,User.business).all()
for i in ret:
# ret = session.query(Users).filter_by(name=‘alex1‘).all()
# ret = session.query(Users).filter_by(name=‘alex1‘).first()
print(i)
# 改
# session.query(Users).filter(Users.id > 2).update({"name": "099"})
# session.query(Users).filter(Users.id > 2).update({Users.name: Users.name + "099"}, synchronize_session=False)
# session.query(Users).filter(Users.id > 2).update({"num": Users.num + 1}, synchronize_session="evaluate")
# session.commit()
# 删除
# session.query(Users).filter(Users.id > 2).delete()
# session.commit()
# 创建删除表
# Base.metadata.create_all(engine)
# Base.metadata.drop_all(engine)
标签：
原文地址：http://www.cnblogs.com/wangminghu/p/5720512.html
