#-*- coding=utf-8 -*-
import sys
reload(sys)
sys.setdefaultencoding('utf8')
import xlrd, os
import telnetlib
import string
from xlutils.copy import copy
import logging
from subprocess import Popen
import commands
from StringIO import StringIO
import time
import socket

DEFAULT_HOST = "1.2.3.4"
DEFAULT_USER = "user"
DEFAULT_PASS = "passwd"

LOG_SERV_ADDR = ('136.24.8.45', 1989)

STR_MAX_SIZE = 32700

EXPIRED_TIME = 300

OUTPUT_FILE = "output.xls"

logging.basicConfig(
 level = logging.INFO,
 format='%(asctime)s %(levelname)s - %(message)s',
 datefmt='%Y-%m-%d %H:%M:%S')

def parse_excel(filename):
 read_book = xlrd.open_workbook(filename);
 write_book = copy(read_book)
 read_sheets = read_book.sheets()
 for i in range(len(read_sheets)):
 parse_sheet(read_sheets[i], write_book.get_sheet(i))
 write_book.save(OUTPUT_FILE)

def parse_sheet(read_sheet, write_sheet):
 for i in range(read_sheet.nrows):
 if i < 1:continue # skip head
 bin_name = read_sheet.cell_value(i, 0).encode('gbk').split('.')[0]
 source_path = read_sheet.cell_value(i, 1).encode('gbk')
 bin_path = read_sheet.cell_value(i, 2).encode('gbk')
 build_log = build_program(source_path, bin_path, bin_name)
 build_flag = check_bin(bin_path, bin_name)
 is_running = check_running_on_product(bin_name)
 product_log = check_running_log(bin_name)
 remote_log = check_remote_log(bin_name)
 print 80 * '-'
 logging.info(bin_name)
 logging.info(source_path)
 logging.info(bin_path)
 logging.info(build_log)
 logging.info(build_flag)
 logging.info(is_running)
 logging.info(product_log)
 logging.info(remote_log)
 print 80 * '-'
 # write result to xls out file
 write_sheet.write(i, 3, build_log[0:STR_MAX_SIZE])
 write_sheet.write(i, 4, build_flag)
 write_sheet.write(i, 5, is_running)
 write_sheet.write(i, 6, product_log)
 write_sheet.write(i, 7, remote_log)


def build_program(source_path, bin_path, bin_name):
 build_cmd = "cd " + source_path + " && make clean && make " + bin_name
 build_log = commands.getoutput(build_cmd)
 ret = []
 # skip chinese
 for i in range(len(build_log)):
 v = ord(build_log[i]) 
 if v < 0x20 or v > 0x7e:
 continue
 ret.append(build_log[i])
 return ''.join(ret)

def check_bin(bin_path, bin_name):
 f = os.path.join(bin_path, bin_name)
 if not os.path.exists(f):
 return "FAIL"
 diff = time.time() - os.path.getctime(f)
 if diff > EXPIRED_TIME:
 return "FAIL"
 return "SUCC"

def check_running_on_product(bin_name):
 cmd = "ps -ef|grep -v grep|grep " + bin_name
 ret = do(cmd)
 running_flag = "No"
 # skip first line
 if ret[ret.find('\n')+1:].find(bin_name) != -1:
 running_flag = "Yes" 
 return running_flag

def check_running_log(bin_name):
 #cmd = "l /ibss?/run/log/" + bin_name + ".log|awk '{sum+=5ドル}END{print sum}'"
 cmd = "l /ibss?/run/log/" + bin_name + ".log|awk '{print 5ドル}'"
 ret = StringIO(do(cmd))
 ret_running_log = "NO LOG"
 line_sum = 0
 for line in ret.readlines():
 if line.isdigit():
 line_sum = line_sum + int(line)
 if line_sum > 0:
 ret_running_log = "HAS LOG"
 ret.close()
 return ret_running_log

def check_remote_log(program_name):
 serv_addr = LOG_SERV_ADDR
 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 s.connect(serv_addr)
 cmd = "find /crmbak -name %s.log.gz -mtime 30 2>/dev/null |xargs ls -l --block-size=1024 |grep %s.log.gz$ 2>/dev/null |awk '{sum+=5ドル}END{print sum}'" % (program_name, program_name)
 s.send(cmd)
 data = s.recv(1024)
 s.close()
 return data

def telnetdo(host, user, passwd, command):
 tn = telnetlib.Telnet()
 try:
 tn.open(host)
 except:
 logging.error("telnet open error")
 return
 tn.read_until("login:")
 tn.write(user.encode('UTF-8') + '\n')
 ReadPasswd=tn.read_until(":")
 if(string.find(ReadPasswd,"Password:") or string.find(ReadPasswd,"password:")):
 tn.write(passwd.encode('UTF-8') + '\n')
 else:
 logging.error("cannot input passwd")
 return
 finish_flag = ('>')
 tn.read_until(finish_flag)
 tn.write(command.encode('UTF-8') + '\r\n')
 result = tn.read_until(finish_flag)
 tn.close()
 return result

def do(cmd):
 return telnetdo(DEFAULT_HOST, DEFAULT_USER, DEFAULT_PASS, cmd)

if __name__ == '__main__':
 #res = check_running_log("R2319Cfm")
 #print res
 #print "-" * 80
 #res = check_running_on_product("R2319Cfm")
 #print res
 #res = build_program('/ibss1/run/cfile', '/ibss1/run/bin' , 'R2319Cfm')
 #print res
 #res = check_bin('/ibss1/run/bin', 'R2319Cfm')
 #print res.find("Last modified")
 #print res
 #cmd = "cd /ibss1/run/cfile && make clean && make R2319Cfm"
 #res = commands.getoutput(cmd)
 #print res
 update_excel("69.xls")