diff --git a/README.md b/README.md
index 7d088f0..30bfcb1 100644
--- a/README.md
+++ b/README.md
@@ -1,11 +1,14 @@
-###运维通用python脚本框架
+### 运维通用python脚本框架
 
 
 * 日常运维:
 [*opsmod.py*](https://github.com/honglongwei/python-scripts/blob/master/opsmod.py)
+ 
+* python操作mysql的原生sql:
+ [*dodb.py*](https://github.com/honglongwei/python-scripts/blob/master/dodb.py)
 
 * zabbix修改可见名称API:
- [*zabbix_update_pj-api*](https://github.com/honglongwei/python-scripts/blob/master/zabbix_update_pj-api.py)
+ [*zabbix_update_pj-api.py*](https://github.com/honglongwei/python-scripts/blob/master/zabbix_update_pj-api.py)
 
 * zabbix通用API:
 [*zabbix-api.py*](https://github.com/honglongwei/python-scripts/blob/master/zabbix-api.py)
@@ -18,3 +21,27 @@
 
 * Python八大排序算法的实现:
 [*psf.py*](https://github.com/honglongwei/python-scripts/blob/master/psf.py)
+
+* 远程执行命令、远程添加信任、远程自动分区挂盘:
+ [*ssh_cmd.py*](https://github.com/honglongwei/python-scripts/blob/master/ssh_cmd.py)
+
+* 基于saltstack的nested输出的运维脚本:
+ [*gameops.py*](https://github.com/honglongwei/python-scripts/blob/master/gameops.py)
+
+* 游戏日志监控脚本:
+ [*monitor_log.py*](https://github.com/honglongwei/python-scripts/blob/master/monitor_log.py)
+ 
+* 扫描服务器生成Excel报表:
+ [*ssh_excel.py*](https://github.com/honglongwei/python-scripts/blob/master/ssh_excel.py)
+ 
+* Web扫描:
+ [*scan_web_banner.py*](https://github.com/honglongwei/python-scripts/blob/master/scan_web_banner.py)
+ 
+* Unix时间戳转换:
+ [*unix_time_format.py*](https://github.com/honglongwei/python-scripts/blob/master/unix_time_format.py)
+
+* LDAP认证:
+ [*ldap.py*](https://github.com/honglongwei/python-scripts/blob/master/ldap.py)
+
+* 获取本机公网出口IP:
+ [*get_local_wip.py*](https://github.com/honglongwei/python-scripts/blob/master/get_local_wip.py)
diff --git a/dodb.py b/dodb.py
new file mode 100644
index 0000000..eaae815
--- /dev/null
+++ b/dodb.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+#_*_ coding:utf-8 _*_
+
+import MySQLdb
+
+
+with open('/var/logs/install.log') as f:
+ a = f.read().strip().split('\n')
+ hostl = []
+ for i in a:
+ b = i.split('\t')
+ if b[3] == 'stop':
+ hostl.append(b[1])
+ else:
+ pass
+ for host in list(set(hostl)):
+ conn= MySQLdb.connect(
+ host='localhost',
+ port = 3306,
+ user='root',
+ passwd='111111',
+ db ='pj_install',
+ )
+ try:
+ #select
+ sql = "select * from automsg where HostName='{0}'".format(host)
+ cur = conn.cursor()
+ cur.execute(sql)
+ ret = cur.fetchone()
+
+ #update
+ update_sql = "update installretmsg set status=2, msg=u'安装成功!' where id={0}".format(int(ret[0]))
+ cur.execute(update_sql)
+
+ cur.close()
+ conn.commit()
+ conn.close()
+ except:
+ pass
diff --git a/gameops.py b/gameops.py
new file mode 100644
index 0000000..27fbf71
--- /dev/null
+++ b/gameops.py
@@ -0,0 +1,146 @@
+#!/usr/bin/env python
+# -*- coding: utf8 -*-
+
+__author__ = 'honglongwei'
+
+import os
+import sys
+import time
+import zipfile
+import datetime
+import subprocess
+import salt.client
+from salt.output.nested import NestDisplay
+from salt.utils import get_colors
+
+#script path
+sct_lt = {'start': '/home/game/startup.sh',
+ 'stop': '/home/game/shutdown.sh'
+ }
+#pid path
+ser_pid = '/home/game/game.pid'
+#src update package path
+src_path = '/home/update'
+#des update package path
+des_path = '/home'
+#src backup package path
+bsrc_path_lt = ['/home/game/config', 
+ '/home/game/data', 
+ '/home/game/hibernate'] 
+#des backup package path
+bdes_path = '/home/backup'
+
+
+# call salt output class
+class NestPut(NestDisplay):
+ def __init__(self):
+ self.colors = get_colors(True)
+ self.__dict__.update(get_colors(True))
+ self.strip_colors = True
+
+def Prest(data):
+ '''
+ Display ret data
+ '''
+ nest = NestPut()
+ print '\n'.join(nest.display(data, 0, '', []))
+
+
+def start():
+ '''
+ Define func start server
+ '''
+ if os.path.exists(ser_pid):
+ return Prest('GameServer is already running !!!')
+ else:
+ cmd = sct_lt['start']
+ proc = subprocess.Popen(cmd, shell=True)
+ return Prest('Start GameServer is successful !!!')
+
+
+def stop():
+ '''
+ Define func stop server
+ '''
+ if os.path.exists(ser_pid):
+ cmd = sct_lt['stop']
+ proc = subprocess.Popen(cmd, shell=True)
+ return Prest('Stop GameServer is successful !!!')
+ else:
+ return Prest('GameServer is already stopped !!!')
+ 
+
+def status():
+ '''
+ Define func status server
+ '''
+ if os.path.exists(ser_pid):
+ return Prest('GameServer is not running !!!')
+ else:
+ cmd = 'ps -ef|grep \'{0}\'|grep -v grep'.format('server')
+ proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
+ ret = proc.stdout.read() 
+ return Prest(ret)
+
+
+def backup():
+ '''
+ Define func backup server
+ '''
+ bakname = 'gs_' + datetime.datetime.now().strftime('%Y%m%d%H%M%S') + '.zip'
+ zipname = os.path.join(bdes_path, bakname)
+ f = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)
+ for bsrc_path in bsrc_path_lt:
+ bac_path = os.path.dirname(bsrc_path)
+ ls_path = bac_path + '/'
+ zg_path = bsrc_path.split(ls_path)[1]
+ os.chdir(bac_path)
+ for dirpath, dirnames, filenames in os.walk(zg_path):
+ for filename in filenames:
+ f.write(os.path.join(dirpath, filename)) 
+ f.close()
+ return 'Backup is successful !'
+
+
+def update(pkg):
+ '''
+ Define func update server
+ '''
+ if pkg:
+ fl = os.path.join(src_path, pkg)
+ try:
+ zfile = zipfile.ZipFile(fl,'r')
+ for filename in zfile.namelist():
+ zfile.extract(filename, des_path)
+ return 'Update is successful !'
+ except IOError:
+ return 'The package is invalid !!!' 
+ else:
+ return 'The package is invalid !!!'
+
+
+
+if __name__== "__main__":
+ # check arguments
+ opts = sys.argv
+ if len(opts) < 2: + print 'start&#124;stop&#124;status&#124;backup&#124;update' + sys.exit(0) + elif len(opts) == 2: + if opts[1]=='start': + start() + elif opts[1]=='stop': + stop() + elif opts[1]=='status': + status() + elif opts[1]=='backup': + backup() + else: + print 'Script Parameter Error !!!' + elif len(opts) == 3: + if opts[1]=='update': + update(opts[2]) + else: + print 'Script Parameter Error !!!' + else: + print 'Script Parameter Error !!!' diff --git a/get_local_wip.py b/get_local_wip.py new file mode 100644 index 0000000..7efb10d --- /dev/null +++ b/get_local_wip.py @@ -0,0 +1,69 @@ +#coding:utf-8 + +import os +import Queue +import codecs +import requests +import threading, subprocess +from time import ctime, sleep, time +from bs4 import BeautifulSoup + +def get_out_ip(url): + req = requests.get(url) + + if req.encoding == 'ISO-8859-1': + encodings = requests.utils.get_encodings_from_content(req.text) + if encodings: + encoding = encodings[0] + else: + encoding = req.apparent_encoding + global encode_content + encode_content = req.content.decode(encoding, 'replace') + + return encode_content[encode_content.find("<div>") + 1: encode_content.find("</div>")].replace("center>", "")
+
+def get_real_url(url=r'http://www.ip138.com/'):
+ r = requests.get(url)
+ txt = r.text
+ soup = BeautifulSoup(txt,"html.parser").iframe
+ return soup["src"]
+
+class ThreadUrl(threading.Thread):
+ def __init__(self,queue):
+ threading.Thread.__init__(self)
+ self.queue = queue
+
+ def run(self):
+ while True:
+ host = self.queue.get()
+ ret = subprocess.Popen('ping -n 600 '+ host, shell=False, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+ with open('iflytek/{0}.log'.format(host), 'a') as f:
+ print>>f, ret.stdout.read()
+ self.queue.task_done()
+
+def startPing():
+ for i in range(100):
+ t=ThreadUrl(queue)
+ t.setDaemon(True)
+ t.start()
+ for host in ['140.143.0.175', '39.107.78.159', '120.92.31.91', '117.121.21.146', '117.121.21.146', '42.62.116.34', '42.62.42.10', '121.201.83.170', '59.107.24.5']:
+ queue.put(host)
+ queue.join()
+
+if __name__ == '__main__':
+ print '*'*40
+ print '*' + u' 欢迎使用网络质量检测工具 ' + '*'
+ print '*' + ' ' + '*'
+ print '*' + u' 程序运行过程中,请不要关闭程序窗口! ' + '*'
+ print '*'*40
+ if not os.path.exists('iflytek'):
+ os.makedirs('iflytek')
+ ip = get_out_ip(get_real_url())
+ f = codecs.open('iflytek/ISP.log', 'w', 'utf-8')
+ f.write(ip)
+ queue = Queue.Queue()
+ ret = startPing()
+
+
+#pip install pyinstaller
+#pyinstaller -F get_local_wip.py>> dist/get_local_wip.exe
diff --git a/ldap.py b/ldap.py
new file mode 100644
index 0000000..c74e892
--- /dev/null
+++ b/ldap.py
@@ -0,0 +1,90 @@
+#!usr/bin/env python 
+#coding: utf-8 
+ 
+import os 
+import sys 
+import ldap 
+ 
+def login_ldap(username, password): 
+ try: 
+ # print("开始执行") 
+ Server = "ldap://8.8.8.8:389" 
+ baseDN = "dc=baidu,dc=com" 
+ searchScope = ldap.SCOPE_SUBTREE 
+ # 设置过滤属性,这里只显示cn=test的信息 
+ searchFilter = "sAMAccountName=" + username 
+ # 为用户名加上域名 
+ username = 'baidu\\' + username 
+ 
+ 
+ # None表示搜索所有属性,['cn']表示只搜索cn属性 
+ retrieveAttributes = None 
+ 
+ conn = ldap.initialize(Server) 
+ #非常重要 
+ conn.set_option(ldap.OPT_REFERRALS, 0) 
+ conn.protocol_version = ldap.VERSION3 
+ # 这里用户名是域账号的全名例如domain/name 
+ #print conn.simple_bind_s(username, password) 
+ conn.simple_bind_s(username, password) 
+ # print 'ldap connect successfully' 
+ 
+ 
+ #调用search方法返回结果id 
+ ldap_result_id = conn.search(baseDN, searchScope, searchFilter, retrieveAttributes) 
+ result_set = [] 
+ #print ldap_result_id 
+ 
+ #print("****************") 
+ while 1: 
+ result_type, result_data = conn.result(ldap_result_id, 0) 
+ if(result_data == []): 
+ break 
+ else: 
+ if result_type == ldap.RES_SEARCH_ENTRY: 
+ result_set.append(result_data) 
+ 
+ #print result_set 
+ Name,Attrs = result_set[0][0] 
+ if hasattr(Attrs, 'has_key') and Attrs.has_key('name'): 
+ ret = {}
+ #distinguishedName = Attrs['mail'][0] 
+ #distinguishedName = Attrs['name'][0] 
+ #distinguishedName = Attrs['displayName'][0] 
+ #distinguishedName = Attrs['mail'][0] 
+ #distinguishedName = Attrs['memberOf'][0] 
+ #distinguishedName = Attrs['mailNickname'][0] 
+ #distinguishedName = Attrs['sAMAccountName'][0] 
+ #distinguishedName = Attrs['distinguishedName'][0] 
+ #distinguishedName = Attrs['title'][0] 
+ #distinguishedName = Attrs['department'][0] 
+ #distinguishedName = Attrs['manager'][0] 
+ # print "Login Info for user : %s" % distinguishedName 
+ 
+ ret['mail'] = Attrs['mail'][0] 
+ ret['username'] = Attrs['name'][0] 
+ ret['nickname'] = Attrs['displayName'][0] 
+ ret['sAMAccountName'] = Attrs['sAMAccountName'][0] 
+ ret['memberOf'] = Attrs['memberOf'][0] 
+ ret['distinguishedName'] = Attrs['distinguishedName'][0] 
+ ret['code'] = 200010
+ # print Attrs['memberOf'][0] 
+ #print Attrs['sAMAccountName'][0] 
+
+ # print Attrs['title'][0] 
+ # print Attrs['department'][0] 
+ 
+ 
+ 
+ return ret
+ 
+ else: 
+ return {'code': 400011, 'msg': u'认证失败'} 
+ except ldap.LDAPError, e: 
+ return {'code': 400012, 'msg': u'认证失败'} 
+ 
+if __name__ == "__main__": 
+ username = "" # ldap中用户名 
+ password = "" # ldap中密码 
+ 
+ print login_ldap(username, password)
diff --git a/monitor_log.py b/monitor_log.py
new file mode 100644
index 0000000..10eeed7
--- /dev/null
+++ b/monitor_log.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+# -*- coding: utf8 -*-
+ 
+import os
+import time
+import datetime
+ 
+ 
+def GetLog(num):
+ fls = []
+ for dirpath, dirnames, filenames in os.walk('/opt/logbak'):
+ for filename in filenames:
+ fn = os.path.join(dirpath, filename)
+ now_time = datetime.datetime.now()
+ last_time = now_time + datetime.timedelta(days=-num)
+ tg_time = last_time.strftime('%Y-%m-%d')
+ endf = tg_time + '.tgz'
+ if filename.startswith('xxx_') and filename.endswith(endf):
+ if os.path.exists(fn):
+ fls.append(fn)
+ else:
+ pass
+ else:
+ pass
+ 
+ xlog = []
+ for a in fls:
+ xlog.append(a[13:17])
+ xlog.sort()
+ return xlog
+ 
+
+blog = GetLog(2)
+alog = GetLog(1)
+'''
+ 交集:list(set(a).intersection(set(b))) 
+ 并集:list(set(a).union(set(b))) 
+ 差集:list(set(b).difference(set(a))) // b中有而a中没有的
+'''
+jdata = set(blog).intersection(set(alog)) #交集
+mor = list(set(alog).difference(jdata)) #新增
+les = list(set(blog).difference(jdata)) #减少
+if len(mor) == 0 and len(les) == 0:
+ print 0
+elif len(mor) == 0 and len(les) != 0:
+ print '-: {0}'.format(','.join(les))
+elif len(mor) != 0 and len(les) == 0:
+ print '+: {0}'.format(','.join(mor))
+else:
+ print '+: {jia}\n-: {shao}'.format(jia=','.join(mor), shao=','.join(les))
diff --git a/scan_web_banner.py b/scan_web_banner.py
new file mode 100644
index 0000000..7bb0ea5
--- /dev/null
+++ b/scan_web_banner.py
@@ -0,0 +1,112 @@
+#/usr/bin/env python
+#-*-coding=utf-8-*-
+
+# __author__ = 'Zline'
+
+import requests
+import re
+from threading import Thread,Lock
+import time
+import sys
+import chardet
+import netaddr
+import struct
+import socket
+
+lock = Lock()
+
+def ip2int(addr):
+ return struct.unpack("!I", socket.inet_aton(addr))[0]
+def int2ip(addr):
+ return socket.inet_ntoa(struct.pack("!I", addr))
+def int_dec(pagehtml):
+
+ charset = None
+ if pagehtml != '':
+ # print 'use charset dect'
+ enc = chardet.detect(pagehtml)
+ # print 'enc= ', enc
+ if enc['encoding'] and enc['confidence']> 0.9:
+ charset = enc['encoding']
+
+ if charset == None:
+ charset_re = re.compile("((^&#124;;)\s*charset\s*=)([^\"']*)", re.M)
+ charset=charset_re.search(pagehtml[:1000]) 
+ charset=charset and charset.group(3) or None
+
+ # test charset
+ try:
+ if charset:
+ unicode('test',charset,errors='replace')
+ except Exception,e:
+ print 'Exception',e
+ charset = None
+ # print 'charset=', charset
+ return charset
+
+
+def http_banner(url):
+ ip=url
+ try:
+ url=requests.get(url,timeout=2) 
+
+ body = url.content
+ 
+ charset = None
+ if body != '':
+ charset = int_dec(body)
+
+ if charset == None or charset == 'ascii':
+ charset = 'ISO-8859-1'
+
+ if charset and charset != 'ascii' and charset != 'unicode':
+ try:
+ body = unicode(body,charset,errors='replace')
+ except Exception, e:
+ body = ''
+ Struts=url.status_code
+ Server=url.headers['server'][0:13]
+ if Struts==200 or Struts==403 or Struts==401:
+ title=re.findall(r"<title>(.*)<\/title>",body)
+ if len(title):
+ title = title[0].strip()
+ else:
+ title = ''
+ lock.acquire()
+ print ('%s\t%d\t%-10s\t%s'%(ip.lstrip('http://'),Struts,Server,title))
+ lock.release()
+ except (requests.HTTPError,requests.RequestException,AttributeError,KeyError),e:
+ pass
+
+
+
+if __name__ == '__main__':
+ if len(sys.argv)>= 2:
+ ips = sys.argv[1]
+ else:
+ print 'usage: python http_banner.py 192.168.1./24 '
+ print 'usage: python http_banner.py 192.168.1.1-192.168.1.254 '
+ print 'usage: python http_banner.py 192.168.1./24 8080'
+ print 'usage: python http_banner.py 192.168.1.1-192.168.1.254 8080'
+ sys.exit()
+ port = '80'
+ if len(sys.argv) == 3:
+ port = sys.argv[2]
+ 
+ if '-' in ips:
+ start, end = ips.split('-')
+ startlong = ip2int(start)
+ endlong = ip2int(end)
+ ips = netaddr.IPRange(start,end)
+ for ip in list(ips):
+ url='http://%s:%s'%(ip,port)
+ t = Thread(target=http_banner,args=(url,))
+ t.daemon=False
+ t.start()
+ elif '/' in ips:
+ ips = netaddr.IPNetwork(ips)
+ for ip in list(ips):
+ url='http://%s:%s'%(ip,port) 
+ t = Thread(target=http_banner,args=(url,))
+ t.daemon=False
+ t.start()
diff --git a/ssh_cmd.py b/ssh_cmd.py
new file mode 100644
index 0000000..e129e0f
--- /dev/null
+++ b/ssh_cmd.py
@@ -0,0 +1,181 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+import time
+import argparse
+import datetime
+import paramiko
+
+reload(sys)
+sys.setdefaultencoding("utf-8")
+
+username = 'root'
+password = '123456'
+
+rsa = [
+'ssh-rsa1',
+'ssh-rsa2'
+]
+
+
+#change return color
+def G(s):
+ return "%s[32;2m%s%s[0m"%(chr(27), s, chr(27))
+def R(s):
+ return "%s[31;2m%s%s[0m"%(chr(27), s, chr(27))
+ 
+
+def cmd_exc(ip, username, password):
+ conn = paramiko.SSHClient()
+ conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ try:
+ conn.connect(hostname = ip, username = username, password = password, timeout = 5)
+ stdin, stdout, stderr = conn.exec_command(cmd)
+ result = stdout.readlines()
+ ret = ''.join(result)
+ except:
+ print R("无法连接")
+ conn.close()
+ try:
+ return G(ret)
+ except UnboundLocalError:
+ pass
+
+
+def copy_rsa(ip, username, password):
+ conn = paramiko.SSHClient()
+ conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ try:
+ conn.connect(hostname = ip, username = username, password = password, timeout = 5)
+ stdin, stdout, stderr = conn.exec_command("echo '{0}'>>/root/.ssh/authorized_keys;echo '{1}'>>/root/.ssh/authorized_keys".format(rsa[0], rsa[1]))
+ result = stdout.readlines()
+ ret = ''.join(result)
+ except:
+ print R("无法连接")
+ conn.close()
+ try:
+ return G(ret)
+ except UnboundLocalError:
+ pass
+
+
+def auto_disk(Disk):
+ if os.path.exists('./auto_disk.sh'):
+ os.remove('./auto_disk.sh') 
+ with open('auto_disk.sh', 'a') as f:
+ print>>f, '#!/bin/bash'
+ print>>f, 'rpm -aq&#124;grep expect'
+ print>>f, 'if [ $? != 0 ];then'
+ print>>f, ' yum install -y expect'
+ print>>f, 'fi'
+ print>>f, '/usr/bin/expect -c"'
+ print>>f, 'set timeout -1'
+ print>>f, 'spawn /sbin/fdisk /dev/{0}'.format(Disk)
+ print>>f, 'expect \"*m for help*:\"'
+ print>>f, 'send -- \"n\r\"' 
+ print>>f, 'expect \"*p*\n\"' 
+ print>>f, 'send -- \"p\r\"'
+ print>>f, 'expect \"*number (1-4):\"' 
+ print>>f, 'send -- \"1\r\"'
+ print>>f, 'expect \"*default 1*:\"'
+ print>>f, 'send -- \"\r\"'
+ print>>f, 'expect \"*default*:\"' 
+ print>>f, 'send -- \"\r\"'
+ print>>f, 'expect \"*m for help*:\"'
+ print>>f, 'send -- \"w\r\"'
+ print>>f, 'expect eof'
+ print>>f, '"'
+ print>>f, 'mkfs.ext4 /dev/{0}1'.format(Disk)
+ print>>f, 'echo "/dev/{0}1 /home/ ext4 defaults 0 0">> /etc/fstab'.format(Disk)
+ print>>f, 'mount /dev/{0}1 /home/'.format(Disk)
+
+
+def sftp_auto(ip, username, password):
+ t = paramiko.Transport((ip,22))
+ t.connect(username = username, password = password)
+ sftp = paramiko.SFTPClient.from_transport(t)
+ sftp.put('./auto_disk.sh','/tmp/auto_disk.sh')
+ t.close()
+ conn = paramiko.SSHClient()
+ conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ try:
+ conn.connect(hostname = ip, username = username, password = password, timeout = 5)
+ stdin, stdout, stderr = conn.exec_command('sh /tmp/auto_disk.sh')
+ result = stdout.readlines()
+ ret = ''.join(result)
+ except:
+ print R("无法连接")
+ conn.close()
+ try:
+ return G(ret)
+ except UnboundLocalError:
+ pass
+ 
+
+
+if __name__ == "__main__":
+ parser=argparse.ArgumentParser(description='ssh_cmd', usage='%(prog)s [options]')
+ parser.add_argument('-H','--host', nargs='?', dest='listhost', help='主机/多个主机用","分割')
+ parser.add_argument('-f','--file', nargs='?', dest='filehost', help='主机列表文件')
+ parser.add_argument('-m','--command', nargs='?', dest='command', help='执行命令')
+ parser.add_argument('-I','--init', nargs='?', dest='init', help='自动分区挂盘')
+ parser.add_argument('-A','--add', nargs='?', dest='add_rsa', help='添加信任')
+ if len(sys.argv)==1:
+ parser.print_help()
+ else:
+ args=parser.parse_args()
+ cmd = args.command
+ if args.listhost is not None and args.filehost is None: 
+ if args.command is not None:
+ for ip in args.listhost.split(','):
+ print G(ip) 
+ print G('-'*80)
+ print cmd_exc(ip, username, password)
+ print 
+ elif args.init is not None:
+ auto_disk(args.init)
+ for ip in args.listhost.split(','):
+ print G(ip)
+ print G('-'*80)
+ print sftp_auto(ip, username, password)
+ print
+ elif args.add_rsa == 'root':
+ for ip in args.listhost.split(','):
+ print G(ip)
+ print G('-'*80)
+ print copy_rsa(ip, username, password)
+ print
+ else:
+ print R('功能选项为空')
+ elif args.listhost is None and args.filehost is not None:
+ if args.command is not None:
+ try:
+ with open(args.filehost) as f:
+ for ips in f.readlines():
+ ip = ips.replace('\n', '')
+ print G(ip)
+ print G('-'*80)
+ print cmd_exc(ip, username, password)
+ print 
+ except IOError:
+ print R('主机列表文件不存在')
+ elif args.init is not None:
+ auto_disk(args.init)
+ for ip in args.listhost.split(','):
+ print G(ip)
+ print G('-'*80)
+ print sftp_auto(ip, username, password)
+ print
+ elif args.add_rsa == 'root':
+ for ip in args.listhost.split(','):
+ print G(ip)
+ print G('-'*80)
+ print copy_rsa(ip, username, password)
+ print
+ else:
+ print R('功能选项为空')
+ else:
+ print R('主机或命令不能为空')
+
diff --git a/ssh_excel.py b/ssh_excel.py
new file mode 100644
index 0000000..e95960e
--- /dev/null
+++ b/ssh_excel.py
@@ -0,0 +1,103 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+
+import os
+import sys
+import xlwt
+import time
+import json
+import urllib
+import urllib2
+import datetime
+import paramiko
+import commands
+
+
+reload(sys)
+sys.setdefaultencoding("utf-8")
+
+username = 'root'
+password = '123456'
+
+url_idc = 'http://www.baidu.com/assetInfo' 
+url_clound = 'http://www.tengxun.com/cloudAsset' 
+
+def GetAllSeal(url):
+ user_agent = 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)' 
+ if url == url_clound:
+ values = {'test': 'demo'} 
+ elif url == url_idc:
+ values = {'product': u'运维'}
+ else:
+ return 'URL is Error !!!'
+ headers = {'User-Agent': user_agent} 
+ data = urllib.urlencode(values) 
+ req = urllib2.Request(url, data, headers) 
+ response = urllib2.urlopen(req) 
+ res = response.read() 
+ a = json.loads(res) 
+ return a
+
+
+def set_style(name, height, bold=False):
+ style = xlwt.XFStyle() 
+
+ font = xlwt.Font() 
+ font.name = name 
+ font.bold = bold
+ font.color_index = 4
+ font.height = height
+
+ style.font = font
+ return style
+
+def WriteDateExcel():
+ Tm = datetime.datetime.now().strftime('%Y-%m-%d')
+ urls = [url_clound, url_idc]
+ wbk = xlwt.Workbook(encoding='utf-8')
+ for url in urls:
+ result = GetAllSeal(url)
+ if url == url_clound:
+ sheet1 = wbk.add_sheet(u'云资产扫描结果', cell_overwrite_ok=True)
+ row0 = [u'业务', u'负责人', u'外网IP', u'内网IP', u'操作系统', u'扫描结果'] 
+ for x in xrange(len(row0)):
+ sheet1.write(0, x, row0[x], set_style('Times New Roman',220,True))
+ for i in xrange(len(result)):
+ conn = paramiko.SSHClient()
+ conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ try:
+ conn.connect(hostname = result[i]['WIp'], username = 'root', timeout = 1)
+ stdin, stdout, stderr = conn.exec_command("iptables -nvL&#124;grep -E '10.0.0.0/8&#124;10.12.0.0/16'")
+ cdt = stdout.readlines()
+ ret = ''.join(cdt)
+ except:
+ ret = u'无法连接'
+ contl = [result[i]['Name'], result[i]['Responser'], result[i]['WIp'], result[i]['LIp'], result[i]['os'], ret]
+ for j in xrange(len(contl)):
+ sheet1.write(i+1, j, contl[j])
+ elif url == url_idc:
+ sheet2 = wbk.add_sheet(u'物理机扫描结果', cell_overwrite_ok=True)
+ row0 = [u'项目', u'负责人', u'外网IP', u'内网IP', u'操作系统', u'扫描结果'] 
+ for x in xrange(len(row0)):
+ sheet2.write(0, x, row0[x], set_style('Times New Roman',220,True))
+ for i in xrange(len(result)):
+ conn = paramiko.SSHClient()
+ conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ try:
+ conn.connect(hostname = result[i]['wIp'], username = 'root', timeout = 1)
+ stdin, stdout, stderr = conn.exec_command("iptables -nvL&#124;grep -E '10.0.0.0/8&#124;10.12.0.0/16'")
+ cdt = stdout.readlines()
+ ret = ''.join(cdt)
+ except:
+ ret = u'无法连接'
+ contl = [result[i]['Name'], result[i]['Owner'], result[i]['wIp'], result[i]['lIp'], result[i]['os'], ret]
+ for j in xrange(len(contl)):
+ sheet2.write(i+1, j, contl[j])
+ else:
+ return 'URL is Error !!!'
+ wbk.save('checkseal_{0}.xls'.format(Tm))
+
+
+if __name__ == "__main__":
+ WriteDateExcel()
diff --git a/unix_time_format.py b/unix_time_format.py
new file mode 100644
index 0000000..fd5cec4
--- /dev/null
+++ b/unix_time_format.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python 
+# -*- coding: utf-8 -*-
+ 
+import time
+ 
+def timestamp_datetime(value):
+ format = '%Y-%m-%d %H:%M:%S'
+ # value为传入的值为时间戳(整形),如:1332888820
+ value = time.localtime(value)
+ ## 经过localtime转换后变成
+ ## time.struct_time(tm_year=2012, tm_mon=3, tm_mday=28, tm_hour=6, tm_min=53, tm_sec=40, tm_wday=2, tm_yday=88, tm_isdst=0)
+ # 最后再经过strftime函数转换为正常日期格式。
+ dt = time.strftime(format, value)
+ return dt
+ 
+def datetime_timestamp(dt):
+ #dt为字符串
+ #中间过程,一般都需要将字符串转化为时间数组
+ time.strptime(dt, '%Y-%m-%d %H:%M:%S')
+ ## time.struct_time(tm_year=2012, tm_mon=3, tm_mday=28, tm_hour=6, tm_min=53, tm_sec=40, tm_wday=2, tm_yday=88, tm_isdst=-1)
+ #将"2012-03-28 06:53:40"转化为时间戳
+ s = time.mktime(time.strptime(dt, '%Y-%m-%d %H:%M:%S'))
+ return int(s)
+ 
+if __name__ == '__main__':
+ d = datetime_timestamp('2015-07-28 21:53:40')
+ print d
+ s = timestamp_datetime(1352889820)
+ print s
</div><div class="naked_ctrl">
<form action="/index.cgi/contrast" method="get" name="gate">
<p><a href="http://altstyle.alfasado.net">AltStyle</a> によって変換されたページ <a href="https://github.com/chenpiao/python-scripts/compare/master...honglongwei:python-scripts:master.diff">(-&gt;オリジナル)</a>
/ <label>アドレス: <input type="text" name="naked_post_url" value="https://github.com/chenpiao/python-scripts/compare/master...honglongwei:python-scripts:master.diff" size="22" /></label> <label>モード: <select name="naked_post_mode">
<option value="default">デフォルト</option>
<option value="speech">音声ブラウザ</option>
<option value="ruby">ルビ付き</option>
<option value="contrast" selected="selected">配色反転</option>
<option value="larger-text">文字拡大</option>
<option value="mobile">モバイル</option>
</select>
<input type="submit" value="表示" />
</p>
</form>
</div>