天天看点

Python 脚本学习笔记(四) 定制业务质量报表

一、使用XlsxWriter模块生成Excel表格

安装XlsxWriter模块

pip2.7 install xlsxwriter

常用方法说明:

Workbook(filename [ options ])用于创建一个workbook对象

创建一个Excel文件

workbook = xlsxwriter.Workbook('chart.xlsx') 

add_worksheet(sheetname)用于添加一个工作表,空号中可以指定工作表名称,默认为Sheet1

创建一个工作表对象

worksheet = workbook.add_worksheet()

add_format([properties]) 用于创建一个新的格式对象,参数[properties]为指定一个格式属性的字典如:

bold = workbook.add_format( { 'bold' : True } )

也可以通过Format methods完成格式的设置如:

bold = workbook.add_format()

bold.set_bold( )

worksheet.write(row,column,'data') 用于写入数据到指定位置的单元格row,column,坐标索引的起始位置为0

如果用坐标表示位置的话比如(2,0) 就是 (A3);

这就是表示在A1位置写入'hello'

worksheet.write( 0 , 0 'hello' )

set_row ( row,height,cell_format,options ) 用于设置行单元格的属性,row为指定位置,height设置行高单位像素,cell_format指定定义好的格式对象调用,options设置行hidden(隐藏)、level(组合分级)、collapsed(折叠)操作示例如下:

设置第1行单元格高度为40像素,并调用格式cell_format 。

worksheet.set_row( 0,40,cell_format ) 

隐藏第2行单元格

worksheet.set_row(1,None,None,{ 'hidden': True })

set_column(first_col,last_col,width,cell_format,options) 用于设置一列或多列单元格属性

设置0到1(即A到B)列单元格宽度为10像素,并调用加粗格式。

worksheet.set_column( 0,1,bold ) 

设置C到D列单元格宽度为20像素

worksheet.set_column( C:D,20 ) 

隐藏E到G列单元格

worksheet.set_column( E:G,None,None,{ 'hidden' : 1} ) 

insert_image(row,col,image,[options]) 用于插入图片到指定的单元格

插入图片指定图片超链接

worksheet.insert_image('B5',' img/he.jpg ',{'url':'http://python.org'} )

add_chart(type:图表类型) 用于创建图表对象

图表样式:area面积样式、bar条形样式、column柱形样式、line线条样式、pie饼图样式、scatter散点样式、stock股票样式、radar雷达样式;

创建一个column柱形图表

chart = workbook.add_chart({ type,'column' }) 

在A5单元格插入图表

worksheet.insert_chart( 'A5',chart ) 

chart.add_series( )用于添加一个数据系列到图表

常用的三个选项:categories作为设置图表类别标签范围,values为设置图标数据范围,line为设置图表线条属性。

chart.add_series({

    'categories' : '=Sheet1!$A$1:$A$5',

    'values' : '=Sheet1!$B$1:$B$5',

    'line' : { 'color' : 'red' },

})

定制自动化业务流量报表周报

订制网站5个频道的流量报表周报,通过XlsxWriter模块将流量数据写入Excel文档,同时自动计算各频道周平均流量,再生成数据报表。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

<code>#!/usr/local/python27/bin/python2.7</code>

<code>#coding: utf-8</code>

<code>import</code> <code>sys</code>

<code>reload</code><code>(sys)</code>

<code>sys.setdefaultencoding(</code><code>'utf-8'</code><code>) </code><code>#用于解决windows转linux出现的编码问题</code>

<code>import</code> <code>xlsxwriter</code>

<code>workbook </code><code>=</code> <code>xlsxwriter.Workbook(</code><code>'chart.xlsx'</code><code>)</code>

<code>worksheet </code><code>=</code> <code>workbook.add_worksheet()</code>

<code>#创建图表对象,定义图表类型。</code>

<code>chart </code><code>=</code> <code>workbook.add_chart({</code><code>'type'</code><code>:</code><code>'column'</code><code>})</code>

<code>title </code><code>=</code> <code>[</code><code>'业务名称'</code><code>,</code><code>'星期一'</code><code>,</code><code>'星期二'</code><code>,</code><code>'星期三'</code><code>,</code><code>'星期四'</code><code>,</code><code>'星期五'</code><code>,</code><code>'星期六'</code><code>,</code><code>'星期日'</code><code>,</code><code>'平均流量'</code><code>]</code>

<code>buname </code><code>=</code> <code>[</code><code>'业务官网'</code><code>,</code><code>'新闻中心'</code><code>,</code><code>'购物频道'</code><code>,</code><code>'体育频道'</code><code>,</code><code>'亲子频道'</code><code>]</code>

<code>data </code><code>=</code> <code>[</code>

<code>    </code><code>[</code><code>150</code><code>,</code><code>152</code><code>,</code><code>158</code><code>,</code><code>149</code><code>,</code><code>155</code><code>,</code><code>145</code><code>,</code><code>148</code><code>],</code>

<code>    </code><code>[</code><code>89</code><code>,</code><code>88</code><code>,</code><code>95</code><code>,</code><code>93</code><code>,</code><code>98</code><code>,</code><code>100</code><code>,</code><code>99</code><code>],</code>

<code>    </code><code>[</code><code>201</code><code>,</code><code>200</code><code>,</code><code>198</code><code>,</code><code>175</code><code>,</code><code>170</code><code>,</code><code>198</code><code>,</code><code>195</code><code>],</code>

<code>    </code><code>[</code><code>75</code><code>,</code><code>77</code><code>,</code><code>78</code><code>,</code><code>78</code><code>,</code><code>74</code><code>,</code><code>70</code><code>,</code><code>79</code><code>],</code>

<code>    </code><code>[</code><code>88</code><code>,</code><code>85</code><code>,</code><code>87</code><code>,</code><code>90</code><code>,</code><code>93</code><code>,</code><code>88</code><code>,</code><code>84</code><code>],</code>

<code>]</code>

<code>format</code> <code>=</code> <code>workbook.add_format() </code><code>#定义格式,用于在部分单元格调用</code>

<code>format</code><code>.set_border(</code><code>1</code><code>)</code>

<code>format_title </code><code>=</code> <code>workbook.add_format()</code>

<code>format_title.set_border(</code><code>1</code><code>)</code>

<code>format_title.set_bg_color(</code><code>'#cccccc'</code><code>)</code>

<code>format_ave </code><code>=</code> <code>workbook.add_format()</code>

<code>format_ave.set_border(</code><code>1</code><code>)</code>

<code>format_ave.set_num_format(</code><code>'0.00'</code><code>)</code>

<code>#使用行写入第一行业务名称及标题</code>

<code>worksheet.write_row(</code><code>'A1'</code><code>,title,format_title)</code>

<code>#使用列写入从第一列的第二行开始写入频道标题</code>

<code>worksheet.write_column(</code><code>'A2'</code><code>,buname,</code><code>format</code><code>)</code>

<code>#从B2开始写入数据项</code>

<code>worksheet.write_row(</code><code>'B2'</code><code>,data[</code><code>0</code><code>],</code><code>format</code><code>)</code>

<code>worksheet.write_row(</code><code>'B3'</code><code>,data[</code><code>1</code><code>],</code><code>format</code><code>)</code>

<code>worksheet.write_row(</code><code>'B4'</code><code>,data[</code><code>2</code><code>],</code><code>format</code><code>)</code>

<code>worksheet.write_row(</code><code>'B5'</code><code>,data[</code><code>3</code><code>],</code><code>format</code><code>)</code>

<code>worksheet.write_row(</code><code>'B6'</code><code>,data[</code><code>4</code><code>],</code><code>format</code><code>)</code>

<code>#定义图表数据系列函数</code>

<code>def</code> <code>chart_series(cur_row):</code>

<code>#这里是在求周平均值,I当前行的数据 = B:H当前行的值相加再求平均</code>

<code>    </code><code>worksheet.write_formula(</code><code>'I'</code><code>+</code><code>cur_row, </code><code>'= AVERAGE(B'</code><code>+</code><code>cur_row</code><code>+</code><code>':H'</code><code>+</code><code>cur_row</code><code>+</code><code>')'</code><code>,format_ave)</code>

<code>    </code><code>chart.add_series({</code>

<code>#这里是在工作表Sheet1中指定该图表类别的标签范围,从B1到H1,就是取了周一到周日。</code>

<code>        </code><code>'categories'</code><code>: </code><code>'=Sheet1!$B$1:$H$1'</code><code>,</code>

<code>#这里是设置图表数据的范围,从B当前数据行到H当前数据行。      </code>

<code>        </code><code>'values'</code><code>: </code><code>'=Sheet1!$B$'</code><code>+</code><code>cur_row</code><code>+</code><code>':$H$'</code><code>+</code><code>cur_row,</code>

<code>        </code><code>'line'</code> <code>: {</code><code>'color'</code><code>:</code><code>'black'</code><code>},</code>

<code>#引用了业务名称为图列项       </code>

<code>        </code><code>'name'</code> <code>: </code><code>'=Sheet1!$A$'</code><code>+</code><code>cur_row,</code>

<code>        </code><code>})</code>

<code>for</code> <code>row </code><code>in</code> <code>range</code><code>(</code><code>2</code><code>,</code><code>7</code><code>):</code>

<code>    </code><code>chart_series(</code><code>str</code><code>(row))</code>

<code>#定义图表样式,每一个值都是一个不一样的样式。</code>

<code>chart.set_style(</code><code>2</code><code>)</code>

<code>#定义图表的大小</code>

<code>chart.set_size({</code><code>'width'</code><code>:</code><code>577</code><code>,</code><code>'height'</code><code>:</code><code>287</code><code>})</code>

<code>#设置图表上方的大标题</code>

<code>chart.set_title({</code><code>'name'</code><code>:</code><code>'业务流量周报图'</code><code>})</code>

<code>#设置图表y轴的小标题</code>

<code>chart.set_y_axis({</code><code>'name'</code><code>:</code><code>'Mb/s'</code><code>})</code>

<code>#插入图表到A8的位置</code>

<code>worksheet.insert_chart(</code><code>'A8'</code><code>,chart)</code>

<code>workbook.close()</code>

二、实现TCP探测目标服务器路由轨迹

scapy模块需要tcpdump支持,生成报表需要graphviz,ImageMagick 图像处理包支持。

yum install tcpdump graphviz ImageMagick -y

安装scapy模块

pip2.7 install scapy

<code> </code> 

<code>import</code> <code>os,sys,time,subprocess</code>

<code>import</code> <code>warnings,logging</code>

<code>#屏蔽无用的告警信息 </code>

<code>warnings.filterwarnings(</code><code>"ignore"</code><code>,category</code><code>=</code><code>DeprecationWarning)</code>

<code>#屏蔽模块IPV6多余告警</code>

<code>logging.getLogger(</code><code>"scapy.runtime"</code><code>).setLevel(logging.ERROR)</code>

<code>from</code> <code>scapy.</code><code>all</code> <code>import</code> <code>traceroute</code>

<code>domains </code><code>=</code> <code>raw_input</code><code>(</code><code>" Please input one or more IP/domain: "</code><code>)</code>

<code>#切片空格,这里就相当于把domains的内容,放到target列表中,可以通过target[number]调用;</code>

<code>target </code><code>=</code> <code>domains.split(</code><code>' '</code><code>)</code>

<code>dport </code><code>=</code> <code>[</code><code>80</code><code>]</code>

<code>if</code> <code>len</code><code>(target) &gt;</code><code>=</code> <code>1</code> <code>and</code> <code>target[</code><code>0</code><code>] !</code><code>=</code><code>'':</code>

<code>#启动路由跟踪</code>

<code>        </code><code>res,unans </code><code>=</code> <code>traceroute(target,dport</code><code>=</code><code>dport,retry</code><code>=</code><code>-</code><code>2</code><code>)</code>

<code>#res保存的是路由跟踪抓的报文数据,通过res.graph方法生成.svg文件;        </code>

<code>        </code><code>res.graph(target</code><code>=</code><code>"&gt; test.svg"</code><code>)</code>

<code>        </code><code>time.sleep(</code><code>1</code><code>)</code>

<code>#调用shell命令,将svg转成png,svg文件可以在浏览器中打开;        </code>

<code>        </code><code>subprocess.Popen(</code><code>"/usr/bin/convert test.svg test.png"</code><code>,shell</code><code>=</code><code>True</code><code>)</code>

<code>else</code><code>:</code>

<code>    </code><code>print</code> <code>"IP/domain Number of errors,exit"</code>

<a href="http://s3.51cto.com/wyfs02/M01/73/6D/wKiom1X9WErD2qv1AAH9xMmStUo554.jpg" target="_blank"></a>

本文转自qw87112 51CTO博客,原文链接:http://blog.51cto.com/tchuairen/1696374