aio_elin 0.0.1

data/lib/modules/output/style/excel_table_wps.rb

@@ -0,0 +1,400 @@
+#coding=utf-8
+
+require "aio/core"
+
+class Aio::Module::OutputStyle::ExcelTableWps < Aio::Module::OutputStyle
+	include Aio::Module
+	include Aio::Device
+	include Aio::Base::Toolkit::ExcelWps
+	include Aio::Ui::Verbose
+
+	attr_accessor :worksheet
+
+	#TitleBanner = "湖北中行现场巡检报告"
+	TitleBanner = "现场巡检报告"
+[
+		# 1号网核心
+		["A_HBWHA_COR_CS01", "30.48.1.248"],
+		["A_HBWHA_COR_CS02", "30.48.1.249"],
+		["A_HBA02_WAN_AR01", "30.48.1.253"],
+		["A_HBA02_WAN_AR02", "30.48.1.254"],
+		["A_HBWHA_PUB_AS01", "30.48.14.21"],
+		["A_HBWHA_PUB_AS02", "30.48.14.22"],
+		["A_HBWHA_MAN_DR", "30.49.255.254"],
+		["A_HBWHA_SVR_AS01", "30.48.255.250"],
+		["A_HBWHA_SVR_AS02", "30.48.255.249"],
+		["A_HBWHA_WBN_DR", "30.48.255.254"],
+		["A_HBWHB_CAM_DS01", "30.48.254.254"],
+		["A_HBWHB_CAM_DS02", "30.48.254.253"],
+		["A_HBWHC_MAN_DR", "30.49.255.253"],
+		["A_HBZ02_WAN_AR02", "30.48.1.251"],
+		["A_HBWHC_WBN_DR", "30.48.255.253"],
+
+		# 2号网核心
+		["B_HBWHA_COR_CS01", "21.160.127.1"],
+		["B_HBWHA_COR_CS02", "21.160.127.2"],
+		["B_HBA02_WAN_AR01", "22.160.120.253"],
+		["B_HBA02_WAN_AR02", "22.160.120.254"],
+		["B_HBWHA_WBN_DR", "21.160.127.195"],
+		["B_HBWHA_MAN_DR01", "21.160.127.197"],
+		["B_HBWHA_MAN_DR02", "21.160.127.198"],
+		["B_HBWHA_CAM_DS01", "21.160.127.199"],
+		["B_HBWHA_CAM_DS02", "21.160.127.200"],
+		["B_HBWHA_SVR_DS01", "21.160.127.201"],
+		["B_HBWHA_SVR_DS02", "21.160.127.202"],
+		["B_HBWHA_ATM_DR01", "21.160.223.254"],
+		["B_HBWHA_ATM_DR02", "21.160.223.253"],
+		["B_HBWHC_WBN_DR", "21.160.128.195"],
+		["B_HBWHB_CAM_DS01", "22.160.95.253"],
+		["B_HBZ02_WAN_AR03", "22.160.120.251"],
+		["B_HBWHB_CAM_DS02", "22.160.95.254"],
+	]
+
+	def initialize
+		super({
+			:author 			=> "Elin",
+			:description	=> "这个模块按照表格输出excel文件, 以WPS运行",
+			:file_suffix	=> "xls",
+		})
+	end
+
+	def generate
+		excel = WorkBook.new
+		#excel.show
+		excel.display_alerts = false
+		@worksheet = excel.add_worksheet("sheet")
+
+		# 设置整体列宽
+		@worksheet.width("a", 18)
+		@worksheet.width("b", 10) 
+		@worksheet.width("c", 15) 
+		@worksheet.width("d", 15) 
+		@worksheet.width("e", 17) 
+		@worksheet.width("f", 7)
+		@worksheet.width("g", 7)
+
+		# 设置整体行高
+		@worksheet.height(20)
+
+	#	DevicePool.size.times do |t|
+	#		device_name = DevicePool[t][0]
+		device_manager.devices.each_key do |device_name|
+			device_klass = device_manager[device_name]
+			device_info = { 
+				:device_klass => device_klass,
+				:manager_ip => "", #DevicePool[t][1],
+				#:device_index => t
+			}
+			case device_klass
+			when Cisco, H3C, Maipu
+				generate_table(device_info)
+			else
+				print_error "未生成表格的设备: #{device_name}"
+			end
+		end
+
+		# 设置垂直分页符
+		@worksheet.pagebreak("H") if @worksheet.has_pagebreak?
+
+		# 恢复警告提示
+		excel.display_alerts = true
+
+		# 保存
+		begin
+			excel.save(output_file.to_s)
+		rescue Exception => e
+			puts e.message
+		ensure
+			excel.close
+		end
+			
+	end
+
+	def generate_table(device_info)
+		generate_hander
+		generate_software(device_info)
+		generate_clock(device_info)
+		generate_inventory(device_info)
+		generate_line
+		generate_cpu_memory(device_info)
+		generate_environment(device_info)
+		generate_interface(device_info)
+		pagebreak
+	end
+
+	def generate_hander
+		@worksheet.add_row do |row|
+			row.merge("a", "g")
+			row.height = 30
+			row << TitleBanner
+			row.style("a", "TitleStyle")
+		end
+		@worksheet.add_row do |row|
+			row.merge("b", "g")
+			row << ["巡检命令", "1.基本信息"]
+			row.style("a", "BoldStyle")
+			row.style("b", "BoldStyle")
+		end
+	end
+
+	def generate_software(device_info)
+		device_klass = device_info[:device_klass]
+		cmd = CmdSwitch.version(device_klass)
+		@worksheet.add_row do |row|
+			row.merge("c", "d")
+			row.merge("f", "g")
+			row << [
+				cmd,
+				"设备名称", device_klass.device_name,
+				"设备型号", device_klass.device_info[:device_model]
+			]
+			row.style("a", "BoldStyle")
+		end
+		@begin_row = @worksheet.current_row_id
+		if !device_info[:manager_ip].empty?
+			manager = device_info[:manager_ip]
+		else
+			manager = device_klass.get_manager_ip
+		end
+		@worksheet.add_row do |row|
+			row.merge("c", "d")
+			row.merge("f", "g")
+			row.height = 24
+			row << [
+				cmd,
+				#"管理IP", device_info[:manager_ip],
+				"管理IP", manager,
+				"IOS版本", device_klass.ios_version
+			]
+			row.wraptext("f")
+		end
+		@worksheet.add_row do |row|
+			row.merge("c", "g")
+			row << [
+				cmd,
+				"IOS特性集", device_klass.soft_image
+			]
+		end
+		@worksheet.add_row do |row|
+			row.merge("c", "g")
+			row << [
+				cmd,
+				"运行时间", device_klass.uptime
+			]
+		end
+		@end_row = @worksheet.current_row_id
+		@worksheet.merge("a#{@begin_row}", "a#{@end_row}")
+	end
+
+	def generate_clock(device_info)
+		device_klass = device_info[:device_klass]
+		cmd = case device_klass
+					when Cisco, Maipu
+						"show clock"
+					when H3C
+						"display clock"
+					end
+		@worksheet.add_row do |row|
+			row.merge("c", "g")
+			tmp  = device_klass.clock
+			time = Aio::Base::Toolkit::Date.time_to_s(tmp)
+			time.insert(0, "`") if time.size > 5
+			row << [
+				cmd,
+				"巡检时间", time
+			]
+			row.style("a", "BoldStyle")
+		end
+	end
+
+	def generate_inventory(device_info)
+		device_klass = device_info[:device_klass]
+		cmd = CmdSwitch.inventory(device_klass)
+				
+		@worksheet.add_row do |row|
+			row.merge("b", "c")
+			row.merge("f", "g")
+			row << [
+				cmd,
+				"硬件信息", "接口卡型号", "序列号", "备注"
+			]
+			row.style("a", "BoldStyle")
+			row.wraptext("a")
+		end
+		@begin_row = @worksheet.current_row_id
+		
+		# 循环添加inventory条目
+		inv = device_klass.inventory
+		inv.each do |info|
+			@worksheet.add_row do |row|
+				row.merge("b", "c")
+				row.merge("f", "g")
+				row << [
+					cmd,
+					info[:name], info[:pid],	info[:sn], ""
+				]
+				row.wraptext("b")
+				row.wraptext("e")
+			end
+		end
+
+		# 当不足10条的时候,填充空白条目至10条
+		if inv.size < 10
+			(10 - inv.size).times do |t|
+				@worksheet.add_row do |row|
+					row.merge("b", "c")
+					row.merge("f", "g")
+					row << [
+						cmd,
+						"", "", "", ""
+					]
+				end
+			end
+		end
+		@end_row = @worksheet.current_row_id
+		@worksheet.merge("a#{@begin_row}", "a#{@end_row}")
+	end
+
+	def generate_line
+		@worksheet.add_row do |row|
+			row.merge("b", "g")
+			row << [
+				"", "2.运行状态巡检信息采集"
+			]
+			row.style("b", "BoldStyle")
+		end
+		@worksheet.add_row do |row|
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				"", "项目", "内容", "结果", "备注"
+			]
+			row.style("b", "BoldStyle")
+			row.style("c", "BoldStyle")
+			row.style("d", "BoldStyle")
+			row.style("f", "BoldStyle")
+		end
+	end
+
+	def generate_cpu_memory(device_info)
+		device_klass = device_info[:device_klass]
+		cmd_cpu = CmdSwitch.cpu(device_klass)
+		cmd_mem = CmdSwitch.memory(device_klass)
+
+		@worksheet.add_row do |row|
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				cmd_cpu, "路由状态",
+				"cpu 利用率(%)", device_klass.cpu_percent,
+				""
+			]
+			row.style("a", "BoldStyle")
+		end
+		@begin_row = @worksheet.current_row_id
+	
+		@worksheet.add_row do |row|
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				cmd_mem, "路由状态",
+				"memory 利用率(%)", device_klass.memory_percent,
+				""
+			]
+			row.style("a", "BoldStyle")
+		end
+	end
+
+	def generate_environment(device_info)
+		device_klass = device_info[:device_klass]
+		cmd = case device_klass
+					when Cisco
+						["show environment"] * 3
+					when H3C
+						[nil, "display power", "display fan"]
+					when Maipu
+						["show environment", "show system power",
+							"show system fan" ]
+					else
+						["show environment"] * 3
+					end
+		
+		unless cmd[0].nil?
+			@worksheet.add_row do |row|
+				tmp = alarm_text(device_klass.warning_env?)
+				row.merge("d", "e")
+				row.merge("f", "g")
+				row << [
+					cmd[0], "路由状态",
+					"温度", tmp,
+					""
+				]
+				row.style("a", "BoldStyle")
+			end
+		end
+		beg_row = @worksheet.current_row_id
+
+		@worksheet.add_row do |row|
+			tmp = alarm_text(device_klass.warning_env?)
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				cmd[1], "路由状态",
+				"电源", tmp,
+				""
+			]
+			row.style("a", "BoldStyle")
+		end
+		@worksheet.add_row do |row|
+			tmp = alarm_text(device_klass.warning_env?)
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				cmd[2], "路由状态",
+				"风扇", tmp,
+				""
+			]
+			row.style("a", "BoldStyle")
+		end
+		end_row = @worksheet.current_row_id
+
+		# 如果cmd都是一样的,那么就合并
+		if cmd[0] == cmd[1] and cmd[1] == cmd[2]
+			@worksheet.merge("a#{beg_row}", "a#{end_row}")
+		end
+	end
+
+	def generate_interface(device_info)
+		device_klass = device_info[:device_klass]
+		cmd = CmdSwitch.interface(device_klass)
+
+		@worksheet.add_row do |row|
+			tmp = alarm_text(device_klass.warning_env?)
+			row.merge("d", "e")
+			row.merge("f", "g")
+			row << [
+				cmd, "路由状态",
+				"接口状态", tmp,
+				""
+			]
+			row.style("a", "BoldStyle")
+		end
+		@end_row = @worksheet.current_row_id
+
+		@worksheet.merge("b#{@begin_row}", "b#{@end_row}")
+	end
+
+	def pagebreak
+		@worksheet.current_row.pagebreak
+	end
+
+	def alarm_text(bool)
+		if bool == true
+			return "正常 □  异常 ■"
+		else
+			return "正常 ■  异常 □"
+		end
+	end
+
+end
+