简化除法判断,分析apache访问日志,扫描存活主机,利用多线程实现ssh并发访问

简化除法判断

1.1 问题

编写mydiv.py脚本,主要要求以下:python

  • 提示用户输入一个数字做为除数
  • 若是用户按下Ctrl+C或Ctrl+D则退出程序
  • 若是用户输入非数字字符,提示用户应该输入数字
  • 若是用户输入0,提示用户0不能做为除数

1.2 方案

使用if语句判断除数是否合适,须要编写多条语句。有了异常处理,能够本着先作,错了再说的逻辑。直接把除法操做放在try语句中执行,根据产生的异常作相应的处理。程序员

另外,Ctrl+C或Ctrl+D只能经过异常捕获。web

异常捕获的语法以下:正则表达式

  1. try:
  2. A
  3. except:
  4. B
  5. else:
  6. C
  7. finally:
  8. D

把可能发生异常的语句放在A里面执行,若是出现异常则执行B语句,没有异常则执行C语句。无论是否出现异常都会执行D语句。shell

捕获异常时,可使用多个except语句,每一个except语句捕获一个异常,每一个异常给定不一样的处理方法。也能够把多个异常放在同一个except语句后面,可是务必注意,多个异常写在相同的一行,必定要注括号括起来,放在元组中。apache

1.3 步骤

实现此案例须要按照以下步骤进行。编程

步骤一:编写脚本vim

  1. #!/usr/bin/env python
  2. import sys
  3. while True:
  4. try:
  5. result = 100 / int(raw_input('enter a number: '))
  6. except (ValueError, ZeroDivisionError), e: #将异常缘由保存在变量e中
  7. print "invalid input:", e
  8. continue
  9. except (EOFError, KeyboardInterrupt):
  10. sys.exit(1)
  11. break
  12. print result

步骤二:测试脚本执行服务器

  1. [root@py01 bin]# ./mydiv.py
  2. enter a number: 0
  3. invalid input: integer division or modulo by zero
  4. enter a number: abc
  5. invalid input: invalid literal for int() with base 10: 'abc'
  6. enter a number: 3
  7. 33

2 案例2:分析apache访问日志

2.1 问题

编写用于分析apache日志的脚本,主要要求以下:多线程

  • 统计每一个客户端访问apache服务器的次数
  • 将统计信息经过字典的方式显示出来
  • 分别统计客户端是Firefox和MSIE的访问次数
  • 分别使用函数式编程和面向对象编程的方式实现

2.2 方案

涉及到文本处理时,正则表达式将是一个很是强大的工具。匹配客户端的IP地址,可使用正则表达式的元字符,匹配字符串能够直接使用字符的表面含义。

入门级程序员的写法,使用顺序的结构,直接编写。这种方法虽然能够得出结果,可是代码难以重用。参考步骤一。

进阶的写法能够采用函数式编程,方便往后再次使用。参考步骤二。

最后,还可使用OOP的编程方法,先定义一个统计类,该类将正则表达式做为它的数据属性。再定义一个方法,从指定的文件中搜索正则表达式出现的次数,并将其存入到一个字典中。参考步骤三。

2.3 步骤

实现此案例须要按照以下步骤进行。

步骤一:简单实现

  1. [root@py01 bin]# vim countweb.py
  2. #!/usr/bin/env python
  3. import re
  4. logfile = '/var/log/httpd/access_log'
  5. cDict = {}
  6. patt_ip = '^\d+\.\d+\.\d+\.\d+' #定义匹配IP地址的正则表达式
  7. with open(logfile) as f:
  8. for eachLine in f:
  9. m = re.search(patt_ip, eachLine)
  10. if m is not None:
  11. ipaddr = m.group()
  12. #若是IP地址已在字典中,将其值加1,不然初始值设置为1
  13. cDict[ipaddr] = cDict.get(ipaddr, 0) + 1
  14. print cDict

步骤二:使用函数式编程实现

  1. [root@py01 bin]# vim countweb2.py
  2. !/usr/bin/env python
  3. import re
  4. def countPatt(patt, fname): #定义能够在指定文件中搜索指定字符串的函数
  5. cDict = {}
  6. with open(fname) as f:
  7. for eachLine in f:
  8. m = re.search(patt, eachLine)
  9. if m is not None:
  10. k = m.group()
  11. cDict[k] = cDict.get(k, 0) + 1
  12. return cDict
  13. def test():
  14. logfile = '/var/log/httpd/access_log'
  15. patt_ip = '^\d+\.\d+\.\d+\.\d+'
  16. print countPatt(patt_ip, logfile)
  17. patt_br = 'Firefox|MSIE'
  18. print countPatt(patt_br, logfile)
  19. if __name__ == '__main__':
  20. test()

3 案例3:扫描存活主机

3.1 问题

编写扫描存活主机的脚本,主要要求以下:

  • 调用系统的ping命令进行扫描
  • 扫描教室环境下全部存活的主机
  • 采用多线程的方式编写
  • 方案

os模块的system()函数能够调用系统命令,其返回值是系统命令退出码,也就是若是系统命令成功执行,返回0,若是没有成功执行,返回非零值。

本例扫描主机,能够调用系统的ping命令,经过退出码来判断是否ping通了该主机。若是顺序执行,每一个ping操做须要消耗数秒 钟,所有的254个地址须要10分钟以上。而采用多线程,能够实现对这254个地址同时执行ping操做,并发的结果就是将执行时间缩短到了10秒钟左 右。

3.2 步骤

实现此案例须要按照以下步骤进行。

步骤一:编写脚本

  1. [root@py01 bin]# vim mtping.py
  2. #!/usr/bin/env python
  3. import subprocess
  4. import threading
  5. import sys
  6. def ping(ip):
  7. result = subprocess.call("ping -c2 %s &> /dev/null" % ip, shell=True)
  8. if result:
  9. print "%s:down" % ip
  10. else:
  11. print "%s:up" % ip
  12. if __name__ == '__main__':
  13. if len(sys.argv) != 2:
  14. print "Usage: %s subnet" % sys.argv[0]
  15. sys.exit(1)
  16. net_list = sys.argv[1].split('.')
  17. net = '.'.join(net_list[:-1])
  18. ips = ("%s.%s" % (net, i) for i in range(1, 255))
  19. for ip in ips:
  20. t = threading.Thread(target=ping, args=(ip,))
  21. t.start()

步骤二:测试脚本执行

  1. [root@py01 bin]# python mtping.py 172.40.51.0

脚本接受命令行参数,只要给定网段就能够实现对该网段中全部ip地址的ping操做。

4 案例4:利用多线程实现ssh并发访问

4.1 问题

编写ssh客户端脚本,主要要求以下:

  • 在文件中取出全部远程主机IP地址
  • 在shell命令行中接受远程服务器IP地址文件、远程服务器密码以及在远程主机上执行的命令
  • 经过多线程实如今全部的远程服务器上并发执行命令
  • 方案

python的paramiko模块能够实现ssh客户端的功能,使用起来也比较简单。可是当服务器很是多的时候,每台服务器上执行彻底相同的简单操做,也会花费大量的时间。

经过ssh加上多线程,能够实现并发访问。为了将程序写的灵活性更强,把要执行的命令以位置参数的方式来提供。

4.2 步骤

实现此案例须要按照以下步骤进行。

步骤一:编写脚本

  1. [root@py01 bin]# vim remote_comm.py
  2. #!/usr/bin/env python
  3. import paramiko
  4. import os
  5. import sys
  6. import threading
  7. def remote_comm(host, password, comm):
  8. ssh = paramiko.SSHClient()
  9. ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  10. ssh.connect(host, username='root', password=password)
  11. stdin, stdout, stderr = ssh.exec_command(comm)
  12. out = stdout.read()
  13. err = stderr.read()
  14. if out:
  15. print "[%s:out]: %s" % (host, out),
  16. if err:
  17. print "%s:Error: %s", (host, err),
  18. ssh.close()
  19. if __name__ == '__main__':
  20. if len(sys.argv) != 4:
  21. print "Usage: %s ipfile password 'comm'" % sys.argv[0]
  22. sys.exit(1)
  23. ipfile = sys.argv[1]
  24. if not os.path.isfile(ipfile):
  25. print "No such file: %s" % ipfile
  26. sys.exit(2)
  27. password = sys.argv[2]
  28. comm = sys.argv[3]
  29. with open(ipfile) as fobj:
  30. for line in fobj:
  31. ip = line.strip()
  32. t = threading.Thread(target=remote_comm, args=(ip, password, comm))
  33. t.start()

步骤二:测试脚本执行

  1. [root@py01 bin]# python remote_comm.py ipaddr.txt tedu.cn 'useradd bob'
相关文章
相关标签/搜索