#!/usr/bin/env py
#coding:utf-8
import os
import os.path
import re
import sys
import paramiko
import threading
import time
from IPy import IP as ipy
def syntax_check(): #配置文件检查
return_value = os.system('python /etc/addresspool.conf')
if return_value != 0:
exit()
def ACL_check(value): #获取用户定义的ip和网段
if value == 'ip':
value = re.compile('^allow_ip.*') #获取ip
elif value == 'network':
value = re.compile('^allow_network.*') #获取网段
else:
print '\033[31mparameter error\033[0m'
exit()
num = 0
conf_file = open('/etc/addresspool.conf')
for i in conf_file: #读取配置文件内容
ip_or_net = re.findall(value,i)
if ip_or_net != []:
num += 1
ACL_add = ip_or_net[0]
ACL_add = ACL_add.split('=')[1]
ACL_add = ACL_add.split(',') #获取ip地址池
else:
if num == 0: #如果用户没有定义地址
return 'None'
conf_file.close()
return ACL_add #返回地址池名称
def GET_ip(ip_pool): #处理地址池名称,生成ip地址(IP对象)
path = sys.path[2]
copy_mod = 'cp /etc/addresspool.conf %s/addresspool.py' %path
os.system(copy_mod)
import addresspool as add
ip_list = []
for i in ip_pool:
i = i.strip()
ACL_ip = getattr(add,i)
for ip in ACL_ip:
ip_list.append(ip)
return ip_list #返回ip地址池
def GET_net(net_pool): #处理地址池名称(NET对象)
import addresspool as network
net_list = []
for i in net_pool:
i = i.strip()
ACL_net = getattr(network,i)
for net in ACL_net:
net_list.append(net)
return net_list #返回网段
def IPY(net_list): #将网段解析成ip地址
net_pool = []
net_add = []
for i in net_list:
ip_pool = []
ip_list = ipy(i)
for ip in ip_list:
ip = str(ip)
ip_pool.append(ip)
else:
ip_pool.pop(0)
ip_pool.pop()
net_pool.append(ip_pool)
return net_pool #返回ip地址池
'-----------------------------------config_file--------------------------------------------'
def telnet(ip,port,user,passwd,comd): #shell命名处理
try:
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(ip,port,user,passwd)
stdin,stdout,stderr = ssh.exec_command(comd)
stderr = stderr.read()
if stderr.strip() != '':
print '\033[34m%s %s\033[0m' %(ip,stderr)
except:
print '\033[31mconnect %s\033[0m' %ip
def get_or_put(file2, file4, ip, port, user, passwd): #文件上传处理
try:
get_put = paramiko.Transport((ip,port)) #服务器ip和端口(使用SFTP时使用)
get_put.connect(username=user, password=passwd) #连接服务器，用户名和密码
sftp = paramiko.SFTPClient.from_transport(get_put) #使用SFTP协议
sftp.put(file2,file4)
except:
print '\033[31merror %s\033[0m' %ip
'------------------------------------handle------------------------'
def conf(): #生成配置文件内容
content = '''
#default deny all
#allow appoint ip address or appoint network address
ACL_ip_name1 = [
"192.168.0.2",
"192.168.0.10",
"192.168.0.40",
"192.168.0.60",
]
#ACL_network_name1 = [
# "192.168.1.0/24",
#]
allow_ip = ACL_ip_name1
#allow_network = ACL_network_name1
'''
if os.path.isfile('/etc/addresspool.conf') == False:
os.system('touch /etc/addresspool.conf')
mod_content = "echo '%s' > /etc/addresspool.conf" %content.strip()
os.system(mod_content)
'----------------------------------confing_file_content------------'
if __name__ == '__main__':
conf() #首次执行生成配置文件
def Syntax(): #定义命令语法
print '''
syntax error:
-h display help info
-t test config file syntax
-s execute shell "command option argv"
-p upload file upload_file_path upload_preservation_path
'''
try:
put = ''
comd = ''
argv = sys.argv
opt = argv[1]
if opt == '-s':
comd = argv[2]
elif opt == '-h':
Syntax()
elif opt == '-t':
test = os.system('python /etc/addresspool.conf')
if test == 0:
print '"/etc/addresspool.conf" Syntax ok\n\n'
elif opt == '-p':
put = 'put'
file2 = argv[2]
filename = file2.split('/').pop()
file = argv[3]
file4 = file +'/'+ filename
else:
Syntax()
except:
Syntax()
syntax_check()
ip = ACL_check('ip')
net = ACL_check('network')
if ip != 'None':
ip_list = GET_ip(ip)
if net != 'None':
net_list = GET_net(net)
net_pool = IPY(net_list)
for netadd in net_pool:
for ipadd in netadd:
ip_list.append(ipadd)
user = 'root'
passwd = '123.com'
port = 22
if comd != '': #多线程执行命令
for ip in ip_list:
try:
sshd = threading.Thread(target=telnet,args=(ip ,port ,user ,passwd ,comd,))
sshd.start()
time.sleep(0.1) #延迟打印
except:
pass
if put != '': #多线程上传文件
for ip in ip_list:
try:
exce = threading.Thread(target=get_or_put, args=(file2,file4,ip,port,user,passwd,))
exce.start()
time.sleep(0.1) #延迟打印
except:
pass
path = sys.path[2]
rm_mod = 'rm -f %s/addresspool.py' %path
os.system(rm_mod)
