Ejemplo n.º 1
0
 def getHtmlLabel(self):
     iniObj = IniUtil(self.configIniPath)
     sections = iniObj.getSections()
     list = []
     for sec in sections:
         tmpList = iniObj.getKeysBySection(sec)
         list.extend(tmpList)
     return list
Ejemplo n.º 2
0
 def __init__(self):
     self.fileName = '../config/cookie.txt'
     self.iniObj = IniUtil('../config/config.ini')
     self.loginUrl = self.iniObj.getValue('loginUrl', 'url')
     #self.loginUrl = 'https://passport.suning.com/ids/login?method=GET&loginTheme=b2c'
     self.vCodeName = "../config/vcode.png"
     #MozillaCookieJar提供可读写操作的cookie文件,存储cookie对象
     self.cookiejar = cookielib.MozillaCookieJar(self.fileName)
Ejemplo n.º 3
0
 def checkIniUrl(self):
     iniObj = IniUtil(self.iniPath)
     keys = iniObj.getKeysBySection('url')
     for key in keys:
         url = iniObj.getValue('url',key)
         if self.checkUrl(url) == False:
             self.falseUrl.append(url)
     return self.falseUrl
Ejemplo n.º 4
0
 def checkIniUrl(self):
     iniObj = IniUtil(self.iniPath)
     keys = iniObj.getKeysBySection('url')
     for key in keys:
         url = iniObj.getValue('url', key)
         if self.checkUrl(url) == False:
             self.falseUrl.append(url)
     return self.falseUrl
Ejemplo n.º 5
0
 def getHtmlLabel(self):
     iniObj = IniUtil(self.configIniPath)
     sections = iniObj.getSections()
     list = []
     for sec in sections:
         tmpList = iniObj.getKeysBySection(sec)
         list.extend(tmpList)
     return list
Ejemplo n.º 6
0
class CodeUtil:
    def __init__(self, filePath):
        self.iniObj = IniUtil(filePath)

    #设置code节点
    def codeStat(self):
        keyList = self.iniObj.getKeysBySection('result')
        for title in keyList:
            code_reason_Str = self.iniObj.getValue('result', title)
            codeStr = code_reason_Str.split(";")[0]
            code = codeStr.split(":")[1]
            self.setCode(code)

    #设置单个状态码数量
    def setCode(self, code):
        self.iniObj.addSection("code")
        flag = self.iniObj.getValue("code", code)
        #状态码存在
        if flag:
            num = int(flag) + 1
            self.iniObj.setValue("code", code, str(num))
        #状态码不存在,新增状态码key,初始值为1
        else:
            self.iniObj.setValue("code", code, "1")


#codeUtil = CodeUtil()
#codeUtil.codeStat()
Ejemplo n.º 7
0
 def __init__(self):
     self.fileName = '../config/cookie.txt'
     self.iniObj = IniUtil('../config/config.ini')
     self.loginUrl = self.iniObj.getValue('loginUrl','url')
     #self.loginUrl = 'https://passport.suning.com/ids/login?method=GET&loginTheme=b2c'
     self.vCodeName = "../config/vcode.png"
     #MozillaCookieJar提供可读写操作的cookie文件,存储cookie对象
     self.cookiejar = cookielib.MozillaCookieJar(self.fileName)
Ejemplo n.º 8
0
 def run(self):
     checkKeyResult = self.check.checkIniKeys()
     #key重复性校验失败
     if checkKeyResult:
         self.logUtil.setKeyLog(self.errorLogPath, checkKeyResult)
     #校验成功
     else:
         checkUrlResult = self.check.checkIniUrl()
         #url合法性校验失败
         if checkUrlResult:
             self.logUtil.setUrlLog(self.errorLogPath, checkUrlResult)
         #校验成功
         else:
             self.iniObj = IniUtil(self.nowPath + '/../config/url.ini')
             self.iniObjResult = IniUtil(self.resultIniPath)
             cookie = self.getCookie()
             #解析内容
             self.getContent(cookie)
             #self.doPost(cookie)
             #self.setCode()
             #self.setHtml()
     print '执行完毕!!'
Ejemplo n.º 9
0
 def run(self):
     checkKeyResult = self.check.checkIniKeys()
     #key重复性校验失败
     if checkKeyResult:
         errorKeyPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
         self.logUtil.setKeyLog(errorKeyPath, checkKeyResult)
     #校验成功
     else:
         checkUrlResult = self.check.checkIniUrl()
         #url合法性校验失败
         if checkUrlResult:
             errorUrlPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
             self.logUtil.setUrlLog(errorUrlPath, checkUrlResult)
         #校验成功
         else:
             self.iniObj = IniUtil(self.nowPath + '/../config/url.ini')
             self.iniObjResult = IniUtil(self.resultIniPath)
             cookie = self.getCookie()
             self.doPost(cookie)
             self.setCode()
             self.setHtml()
     print '执行完毕!!'
Ejemplo n.º 10
0
 def run(self):
     checkKeyResult = self.check.checkIniKeys()
     # key重复性校验失败
     if checkKeyResult:
         errorKeyPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
         self.logUtil.setKeyLog(errorKeyPath, checkKeyResult)
     # 校验成功
     else:
         checkUrlResult = self.check.checkIniUrl()
         # url合法性校验失败
         if checkUrlResult:
             errorUrlPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
             self.logUtil.setUrlLog(errorUrlPath, checkUrlResult)
         # 校验成功
         else:
             self.iniObj = IniUtil(self.nowPath + "/../config/url.ini")
             self.iniObjResult = IniUtil(self.resultIniPath)
             cookie = self.getCookie()
             self.doPost(cookie)
             self.setCode()
             self.setHtml()
     print "执行完毕!!"
Ejemplo n.º 11
0
class CodeUtil:
    def __init__(self, filePath):
        self.iniObj = IniUtil(filePath)

    # 设置code节点
    def codeStat(self):
        keyList = self.iniObj.getKeysBySection("result")
        for title in keyList:
            code_reason_Str = self.iniObj.getValue("result", title)
            codeStr = code_reason_Str.split(";")[0]
            code = codeStr.split(":")[1]
            self.setCode(code)

    # 设置单个状态码数量
    def setCode(self, code):
        self.iniObj.addSection("code")
        flag = self.iniObj.getValue("code", code)
        # 状态码存在
        if flag:
            num = int(flag) + 1
            self.iniObj.setValue("code", code, str(num))
        # 状态码不存在,新增状态码key,初始值为1
        else:
            self.iniObj.setValue("code", code, "1")
Ejemplo n.º 12
0
 def run(self):
     checkKeyResult = self.check.checkIniKeys()
     # key重复性校验失败
     if checkKeyResult:
         self.logUtil.setKeyLog(self.errorLogPath, checkKeyResult)
     # 校验成功
     else:
         checkUrlResult = self.check.checkIniUrl()
         # url合法性校验失败
         if checkUrlResult:
             self.logUtil.setUrlLog(self.errorLogPath, checkUrlResult)
         # 校验成功
         else:
             self.iniObj = IniUtil(self.nowPath + "/../config/url.ini")
             self.iniObjResult = IniUtil(self.resultIniPath)
             cookie = self.getCookie()
             # 解析内容
             self.getContent(cookie)
             # self.doPost(cookie)
             # self.setCode()
             # self.setHtml()
     print "执行完毕!!"
Ejemplo n.º 13
0
 def __init__(self, filePath):
     self.iniObj = IniUtil(filePath)
Ejemplo n.º 14
0
 def __init__(self, filePath):
     self.iniObj = IniUtil(filePath)
Ejemplo n.º 15
0
 def __init__(self, resultIniPath, htmlPath):
     html = open(htmlPath, 'w+')
     #iniObj = IniUtil('../log/result.ini')
     iniObj = IniUtil(resultIniPath)
     strFirst = '''
     <html>
      <head>
       <meta charset="utf-8" />
       <title>状态码</title>
      </head>
      <body>
       <table align="center" width="80%" border="1">
        <tbody>
         <tr>
          <td colspan="3" align="center">
           <!-- 为ECharts准备一个具备大小(宽高)的Dom -->
           <div id="codeImg" style="height:300px;width:400px;"></div> </td>
         </tr>
         <tr>
          <td colspan="3" align="center">
           <table width="100%" border="1" cellspacing="0" cellpadding="0" class="sp-order-table">
            <thead>
             <tr bgcolor="#CECEFF">
              <td>标题</td>
              <td>URL</td>
              <td>状态码</td>
              <td>结果</td>
             </tr>
            </thead>
     '''
     colorRed = "style='color:#FF0000'"
     colorGreen = "style='color:#00FF7F'"
     titleList = iniObj.getKeysBySection('result')
     strTbody = '<tbody>'
     codeStr = ''
     code = ''
     reasonStr = ''
     reason = ''
     colorFlag = ''
     for title in titleList:
         codeStr = ''
         reasonStr = ''
         code_reason_Str = iniObj.getValue('result', title)
         if ";" in code_reason_Str:
             codeStr = code_reason_Str.split(";")[0]
             #code = codeStr.split(":")[1]
             reasonStr = code_reason_Str.split(";")[1]
             #reason = reasonStr.split(":")[1]
             colorFlag = colorRed
         else:
             reasonStr = '通过'
             codeStr = code_reason_Str.split(";")[0]
             #code = codeStr.split(":")[1]
             colorFlag = colorGreen
         strTbody += "<tr>"
         strTbody += "<td>" + title + "</td>"
         strTbody += "<td>" + iniObj.getValue('result', title) + "</td>"
         strTbody += "<td>" + codeStr + "</td>"
         strTbody += "<td " + colorFlag + ">" + reasonStr + "</td>"
         strTbody += "</tr>"
     strTbody += '</tbody>'
     strSecond = '''
           </table>
           </td>
         </tr>
        </tbody>
       </table>
       <!-- ECharts单文件引入 -->
       <script src="../config/echarts.common.min.js"></script>
       <style>
         body{ text-align:center}
         .sp-order-table {
             clear: both;
             border: 1px solid #ebebeb;
             margin-top: 20px;
         }
       </style>
       <script type="text/javascript">
         // 基于准备好的dom,初始化echarts图表
         var myChart = echarts.init(document.getElementById('codeImg'));
         var option = {
         //标题
         title : {
             text: '状态码',
             //subtext: '纯属虚构',
             x:'center'
         },
     '''
     dataName = '['
     data = ''
     strThird = ''
     codeList = iniObj.getKeysBySection('code')
     for codeName in codeList:
         dataName += "'" + codeName + "',"
         num = iniObj.getValue('code', codeName)
         data += "{value:" + num + ", name:'" + codeName + "'},"
     dataName = dataName[:-1] + ']'
     strThird = '''
         //左侧竖直列
         legend: {
             orient : 'vertical',
             x : 'left',
             data:''' + dataName + '''
         },
         //饼图值
         series : [
             {
                 name:'状态码',
                 type:'pie',
                 radius : '55%',
                 center: ['50%', '60%'],
                 data:[''' + data[:-1] + '''
                 ]
             }
         ]
     };
     '''
     strFour = '''
         // 为echarts对象加载数据
         myChart.setOption(option);
       </script>
      </body>
     </html>
     '''
     strHtml = strFirst + strTbody + strSecond + strThird + strFour
     html.write(strHtml)
     html.close()
Ejemplo n.º 16
0
# -*- coding:utf-8 -*-
import urllib2
from iniUtil import IniUtil
from cookieUtil import CookieUtil

cookieObj = CookieUtil()
iniObj = IniUtil('../config/url.ini')
cookie = cookieObj.getLoginCookie()
#获取配置文件的url
keysList = []
urlDict = {}
keysList = iniObj.getKeysBySection('url')
for key in keysList:
    pageCode = ''
    url = ''
    url = iniObj.getValue('url', key)
    urlDict.setdefault(key, url)
    #url = "http://kxyesit.cnsuning.com/"
    try:
        #创建请求的request
        req = urllib2.Request(url)
        handler = urllib2.HTTPCookieProcessor(cookie)
        #利用urllib2的build_opener方法创建一个opener
        openner = urllib2.build_opener(handler)
        responseCode = openner.open(req).getcode()
        #print '页面状态码为:' + str(responseCode)
        pageCode = u'状态码:' + str(responseCode)
        iniObj.setValue(section=u'状态码', key=key, value=pageCode)
    except urllib2.URLError, e:
        #print '错误原因:'+e.reason
        reason = e.reason
Ejemplo n.º 17
0
Archivo: run.py Proyecto: zw19881/zhaow
# -*- coding:utf-8 -*-
import urllib2
from iniUtil import IniUtil
from cookieUtil import CookieUtil


cookieObj = CookieUtil()
iniObj = IniUtil('../config/url.ini')
cookie = cookieObj.getLoginCookie()
#获取配置文件的url
keysList = []
urlDict = {}
keysList = iniObj.getKeysBySection('url')
for key in keysList:
    pageCode = ''
    url = ''
    url = iniObj.getValue('url',key)
    urlDict.setdefault(key,url)
    #url = "http://kxyesit.cnsuning.com/"
    try:
        #创建请求的request
        req = urllib2.Request(url)
        handler = urllib2.HTTPCookieProcessor(cookie)
        #利用urllib2的build_opener方法创建一个opener
        openner = urllib2.build_opener(handler)
        responseCode = openner.open(req).getcode()
        #print '页面状态码为:' + str(responseCode)
        pageCode = u'状态码:' + str(responseCode)
        iniObj.setValue(section=u'状态码',key=key,value=pageCode)
    except urllib2.URLError,e:
        #print '错误原因:'+e.reason
Ejemplo n.º 18
0
 def __init__(self,resultIniPath,htmlPath):
     html = open(htmlPath, 'w+')
     #iniObj = IniUtil('../log/result.ini')
     iniObj = IniUtil(resultIniPath)
     strFirst = '''
     <html>
      <head>
       <meta charset="utf-8" />
       <title>状态码</title>
      </head>
      <body>
       <table align="center" width="80%" border="1">
        <tbody>
         <tr>
          <td colspan="3" align="center">
           <!-- 为ECharts准备一个具备大小(宽高)的Dom -->
           <div id="codeImg" style="height:300px;width:400px;"></div> </td>
         </tr>
         <tr>
          <td colspan="3" align="center">
           <table width="100%" border="1" cellspacing="0" cellpadding="0" class="sp-order-table">
            <thead>
             <tr bgcolor="#CECEFF">
              <td>标题</td>
              <td>URL</td>
              <td>状态码</td>
              <td>结果</td>
             </tr>
            </thead>
     '''
     colorRed = "style='color:#FF0000'"
     colorGreen = "style='color:#00FF7F'"
     titleList = iniObj.getKeysBySection('result')
     strTbody = '<tbody>'
     codeStr = ''
     code = ''
     reasonStr = ''
     reason = ''
     colorFlag = ''
     for title in titleList:
         codeStr = ''
         reasonStr = ''
         code_reason_Str = iniObj.getValue('result',title)
         if ";" in code_reason_Str:
             codeStr = code_reason_Str.split(";")[0]
             #code = codeStr.split(":")[1]
             reasonStr = code_reason_Str.split(";")[1]
            #reason = reasonStr.split(":")[1]
             colorFlag = colorRed
         else:
             reasonStr = '通过'
             codeStr = code_reason_Str.split(";")[0]
             #code = codeStr.split(":")[1]
             colorFlag = colorGreen
         strTbody += "<tr>"
         strTbody += "<td>"+title+"</td>"
         strTbody += "<td>"+iniObj.getValue('result',title)+"</td>"
         strTbody += "<td>"+codeStr+"</td>"
         strTbody += "<td "+colorFlag+">"+reasonStr+"</td>"
         strTbody += "</tr>"
     strTbody +='</tbody>'
     strSecond='''
           </table>
           </td>
         </tr>
        </tbody>
       </table>
       <!-- ECharts单文件引入 -->
       <script src="../config/echarts.common.min.js"></script>
       <style>
         body{ text-align:center}
         .sp-order-table {
             clear: both;
             border: 1px solid #ebebeb;
             margin-top: 20px;
         }
       </style>
       <script type="text/javascript">
         // 基于准备好的dom,初始化echarts图表
         var myChart = echarts.init(document.getElementById('codeImg'));
         var option = {
         //标题
         title : {
             text: '状态码',
             //subtext: '纯属虚构',
             x:'center'
         },
     '''
     dataName = '['
     data = ''
     strThird = ''
     codeList = iniObj.getKeysBySection('code')
     for codeName in codeList:
         dataName += "'" + codeName + "',"
         num = iniObj.getValue('code',codeName)
         data += "{value:"+num+", name:'"+codeName+"'},"
     dataName = dataName[:-1] + ']'
     strThird = '''
         //左侧竖直列
         legend: {
             orient : 'vertical',
             x : 'left',
             data:'''+dataName+'''
         },
         //饼图值
         series : [
             {
                 name:'状态码',
                 type:'pie',
                 radius : '55%',
                 center: ['50%', '60%'],
                 data:['''+data[:-1]+'''
                 ]
             }
         ]
     };
     '''
     strFour = '''
         // 为echarts对象加载数据
         myChart.setOption(option);
       </script>
      </body>
     </html>
     '''
     strHtml = strFirst+strTbody+strSecond+strThird+strFour
     html.write(strHtml)
     html.close()
Ejemplo n.º 19
0
class RunCode:
    def __init__(self):
        self.cookieObj = CookieUtil()
        self.nowTile = time.strftime("%Y%m%d%H%M%S")
        self.nowPath = os.getcwd()
        if os.path.exists(self.nowPath + "/../log") == False:
            os.mkdir(self.nowPath + "/../log")
        if os.path.exists(self.nowPath + "/../result") == False:
            os.mkdir(self.nowPath + "/../result")
        self.resultIniPath = self.nowPath + "/../log/result_" + self.nowTile + ".ini"
        self.errorLogPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
        self.htmlPath = self.nowPath + "/../result/result_" + self.nowTile + ".html"
        self.check = Check()
        self.logUtil = LogUtil()

    def run(self):
        checkKeyResult = self.check.checkIniKeys()
        # key重复性校验失败
        if checkKeyResult:
            self.logUtil.setKeyLog(self.errorLogPath, checkKeyResult)
        # 校验成功
        else:
            checkUrlResult = self.check.checkIniUrl()
            # url合法性校验失败
            if checkUrlResult:
                self.logUtil.setUrlLog(self.errorLogPath, checkUrlResult)
            # 校验成功
            else:
                self.iniObj = IniUtil(self.nowPath + "/../config/url.ini")
                self.iniObjResult = IniUtil(self.resultIniPath)
                cookie = self.getCookie()
                # 解析内容
                self.getContent(cookie)
                # self.doPost(cookie)
                # self.setCode()
                # self.setHtml()
        print "执行完毕!!"

    def getCookie(self):
        cookie = self.cookieObj.getLoginCookie()
        return cookie

    def getContent(self, cookie):
        # 获取配置文件的url
        keysList = []
        urlDict = {}
        keysList = self.iniObj.getKeysBySection("url")
        num = 1
        excelUtilObj = ExcelUtil()
        for key in keysList:
            url = self.iniObj.getValue("url", key)
            urlDict.setdefault(key, url)
            try:
                # 创建请求的request
                req = urllib2.Request(url)
                handler = urllib2.HTTPCookieProcessor(cookie)
                # 利用urllib2的build_opener方法创建一个opener
                openner = urllib2.build_opener(handler)
                response = openner.open(req)

                content = response.read()
                htmlLabel = excelUtilObj.htmlLabel
                worksheet = excelUtilObj.addSheet(key)
                excelUtilObj.setKeys(worksheet)
                self.setExcel(worksheet, num, excelUtilObj, content, htmlLabel)
                num += 1
            except urllib2.URLError, e:
                reason = e.reason
                pageCode = u"状态码:" + str(e.code) + u";错误原因:" + reason
                self.logUtil.setPostErrorLog(self.errorLogPath, url, pageCode)
        excelUtilObj.closeExcel()
Ejemplo n.º 20
0
class CookieUtil:
    def __init__(self):
        self.fileName = '../config/cookie.txt'
        self.iniObj = IniUtil('../config/config.ini')
        self.loginUrl = self.iniObj.getValue('loginUrl', 'url')
        #self.loginUrl = 'https://passport.suning.com/ids/login?method=GET&loginTheme=b2c'
        self.vCodeName = "../config/vcode.png"
        #MozillaCookieJar提供可读写操作的cookie文件,存储cookie对象
        self.cookiejar = cookielib.MozillaCookieJar(self.fileName)

    #获取登录页面的response信息,用来获取cookie和uuid
    def getLoginPageResp(self):
        #将一个保存cookie对象,和一个HTTP的cookie的处理器绑定
        cookieSupport = urllib2.HTTPCookieProcessor(self.cookiejar)
        #创建一个opener,将保存了cookie的http处理器,还有设置一个handler用于处理http的
        opener = urllib2.build_opener(cookieSupport)
        #将包含了cookie、http处理器、http的handler的资源和urllib2对象绑定在一起,安装opener,此后调用urlopen()时都会使用安装过的opener对象,
        urllib2.install_opener(opener)
        #获取uuid
        sn = urllib2.urlopen(self.loginUrl)
        resp = sn.read()
        return resp
        #保存cookie到文件
        #cookiejar.save(ignore_discard=True,ignore_expires=True)

    #获取uuid
    def getUuid(self, loginPageResp):
        #打开验证码
        pattern = re.compile('var uuid = "(.*?)";', re.S)
        matchResult = re.search(pattern, loginPageResp)
        uuid = matchResult.group(1)
        #print "uuid::"+uuid
        return uuid

    #储存图片
    def saveVCodeImg(self, uuid):
        imgurlStr = self.iniObj.getValue('codeImg', 'url')
        imgurl = imgurlStr + "?uuid=" + uuid + "&yys="
        #imgurl = "https://vcs.suning.com/vcs/imageCode.htm?uuid="+uuid+"&yys="
        request = urllib2.Request(imgurl)
        response = urllib2.urlopen(request)
        content = response.read()
        fp = open(self.vCodeName, "wb")
        fp.write(content)
        fp.close()

    #存储登录后cookie
    def saveLoginCookie(self, userName, passWord, uuid, vcode):
        data = urllib.urlencode({
            'username': userName,
            'password': passWord,
            'jsonViewType': 'true',
            'loginTheme': 'b2c',
            "uuid": uuid,
            'verifyCode': vcode
        })
        request = urllib2.Request(self.loginUrl, data=data)
        response = urllib2.urlopen(request)
        #保存cookie到文件
        self.cookiejar.save(ignore_discard=True, ignore_expires=True)

    #获取cookie
    def getCookie(self):
        #创建MozillaCookieJar实例对象
        cookieFromFile = cookielib.MozillaCookieJar()
        #从文件中读取cookie内容到变量
        cookieFromFile.load(self.fileName,
                            ignore_expires=True,
                            ignore_discard=True)
        return cookieFromFile

    #直接获取登录后cookie
    def getLoginCookie(self):
        resp_ = self.getLoginPageResp()
        uuid_ = self.getUuid(resp_)
        self.saveVCodeImg(uuid_)
        #输入验证码
        vcode_ = raw_input("输入验证码 :")
        username_ = self.iniObj.getValue('user', 'username')
        password_ = self.iniObj.getValue('user', 'password')
        #username_ = '*****@*****.**'
        #password_ = 'yxz0521'
        self.saveLoginCookie(userName=username_,
                             passWord=password_,
                             uuid=uuid_,
                             vcode=vcode_)
        returnCookie = self.getCookie()
        return returnCookie
Ejemplo n.º 21
0
class RunCode:
    def __init__(self):
        self.cookieObj = CookieUtil()
        self.nowTile = time.strftime('%Y%m%d%H%M%S')
        self.nowPath = os.getcwd()
        if os.path.exists(self.nowPath + "/../log") == False:
            os.mkdir(self.nowPath + "/../log")
        if os.path.exists(self.nowPath + "/../result") == False:
            os.mkdir(self.nowPath + "/../result")
        self.resultIniPath = self.nowPath + "/../log/result_" + self.nowTile + ".ini"
        self.errorLogPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
        self.htmlPath = self.nowPath + "/../result/result_" + self.nowTile + ".html"
        self.check = Check()
        self.logUtil = LogUtil()

    def run(self):
        checkKeyResult = self.check.checkIniKeys()
        #key重复性校验失败
        if checkKeyResult:
            self.logUtil.setKeyLog(self.errorLogPath, checkKeyResult)
        #校验成功
        else:
            checkUrlResult = self.check.checkIniUrl()
            #url合法性校验失败
            if checkUrlResult:
                self.logUtil.setUrlLog(self.errorLogPath, checkUrlResult)
            #校验成功
            else:
                self.iniObj = IniUtil(self.nowPath + '/../config/url.ini')
                self.iniObjResult = IniUtil(self.resultIniPath)
                cookie = self.getCookie()
                #解析内容
                self.getContent(cookie)
                #self.doPost(cookie)
                #self.setCode()
                #self.setHtml()
        print '执行完毕!!'

    def getCookie(self):
        cookie = self.cookieObj.getLoginCookie()
        return cookie

    def getContent(self, cookie):
        #获取配置文件的url
        keysList = []
        urlDict = {}
        keysList = self.iniObj.getKeysBySection('url')
        num = 1
        excelUtilObj = ExcelUtil()
        for key in keysList:
            url = self.iniObj.getValue('url', key)
            urlDict.setdefault(key, url)
            try:
                #创建请求的request
                req = urllib2.Request(url)
                handler = urllib2.HTTPCookieProcessor(cookie)
                #利用urllib2的build_opener方法创建一个opener
                openner = urllib2.build_opener(handler)
                response = openner.open(req)

                content = response.read()
                htmlLabel = excelUtilObj.htmlLabel
                worksheet = excelUtilObj.addSheet(key)
                excelUtilObj.setKeys(worksheet)
                self.setExcel(worksheet, num, excelUtilObj, content, htmlLabel)
                num += 1
            except urllib2.URLError, e:
                reason = e.reason
                pageCode = u'状态码:' + str(e.code) + u';错误原因:' + reason
                self.logUtil.setPostErrorLog(self.errorLogPath, url, pageCode)
        excelUtilObj.closeExcel()
Ejemplo n.º 22
0
class CookieUtil:

    def __init__(self):
        self.fileName = '../config/cookie.txt'
        self.iniObj = IniUtil('../config/config.ini')
        self.loginUrl = self.iniObj.getValue('loginUrl','url')
        #self.loginUrl = 'https://passport.suning.com/ids/login?method=GET&loginTheme=b2c'
        self.vCodeName = "../config/vcode.png"
        #MozillaCookieJar提供可读写操作的cookie文件,存储cookie对象
        self.cookiejar = cookielib.MozillaCookieJar(self.fileName)

    #获取登录页面的response信息,用来获取cookie和uuid
    def getLoginPageResp(self):
        #将一个保存cookie对象,和一个HTTP的cookie的处理器绑定
        cookieSupport= urllib2.HTTPCookieProcessor(self.cookiejar)
        #创建一个opener,将保存了cookie的http处理器,还有设置一个handler用于处理http的
        opener = urllib2.build_opener(cookieSupport)
        #将包含了cookie、http处理器、http的handler的资源和urllib2对象绑定在一起,安装opener,此后调用urlopen()时都会使用安装过的opener对象,
        urllib2.install_opener(opener)
        #获取uuid
        sn = urllib2.urlopen(self.loginUrl)
        resp = sn.read()
        return resp
        #保存cookie到文件
        #cookiejar.save(ignore_discard=True,ignore_expires=True)

    #获取uuid
    def getUuid(self,loginPageResp):
        #打开验证码
        pattern = re.compile('var uuid = "(.*?)";',re.S)
        matchResult = re.search(pattern,loginPageResp)
        uuid = matchResult.group(1)
        #print "uuid::"+uuid
        return uuid

    #储存图片
    def saveVCodeImg(self,uuid):
        imgurlStr = self.iniObj.getValue('codeImg','url')
        imgurl = imgurlStr + "?uuid="+uuid+"&yys="
        #imgurl = "https://vcs.suning.com/vcs/imageCode.htm?uuid="+uuid+"&yys="
        request = urllib2.Request(imgurl)
        response = urllib2.urlopen(request)
        content = response.read()
        fp = open(self.vCodeName,"wb")
        fp.write(content)
        fp.close()

    #存储登录后cookie
    def saveLoginCookie(self,userName,passWord,uuid,vcode):
        data = urllib.urlencode({
            'username':userName,
            'password':passWord,
            'jsonViewType':'true',
            'loginTheme':'b2c',
            "uuid":uuid,
            'verifyCode':vcode
        })
        request = urllib2.Request(self.loginUrl,data=data)
        response = urllib2.urlopen(request)
        #保存cookie到文件
        self.cookiejar.save(ignore_discard=True,ignore_expires=True)

    #获取cookie
    def getCookie(self):
        #创建MozillaCookieJar实例对象
        cookieFromFile = cookielib.MozillaCookieJar()
        #从文件中读取cookie内容到变量
        cookieFromFile.load(self.fileName,ignore_expires=True,ignore_discard=True)
        return cookieFromFile

    #直接获取登录后cookie
    def getLoginCookie(self):
        resp_ = self.getLoginPageResp()
        uuid_ = self.getUuid(resp_)
        self.saveVCodeImg(uuid_)
        #输入验证码
        vcode_ = raw_input("输入验证码 :")
        username_ = self.iniObj.getValue('user','username')
        password_ = self.iniObj.getValue('user','password')
        #username_ = '*****@*****.**'
        #password_ = 'yxz0521'
        self.saveLoginCookie(userName=username_,passWord=password_,uuid=uuid_,vcode=vcode_)
        returnCookie = self.getCookie()
        return returnCookie
Ejemplo n.º 23
0
class RunCode:
    def __init__(self):
        self.cookieObj = CookieUtil()
        self.nowTile = time.strftime('%Y%m%d%H%M%S')
        self.nowPath = os.getcwd()
        if os.path.exists(self.nowPath + "/../log") == False:
            os.mkdir(self.nowPath + "/../log")
        if os.path.exists(self.nowPath + "/../result") == False:
            os.mkdir(self.nowPath + "/../result")
        self.resultIniPath = self.nowPath + "/../log/result_" + self.nowTile + ".ini"
        self.htmlPath = self.nowPath + "/../result/result_" + self.nowTile + ".html"
        self.check = Check()
        self.logUtil = LogUtil()

    def run(self):
        checkKeyResult = self.check.checkIniKeys()
        #key重复性校验失败
        if checkKeyResult:
            errorKeyPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
            self.logUtil.setKeyLog(errorKeyPath, checkKeyResult)
        #校验成功
        else:
            checkUrlResult = self.check.checkIniUrl()
            #url合法性校验失败
            if checkUrlResult:
                errorUrlPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
                self.logUtil.setUrlLog(errorUrlPath, checkUrlResult)
            #校验成功
            else:
                self.iniObj = IniUtil(self.nowPath + '/../config/url.ini')
                self.iniObjResult = IniUtil(self.resultIniPath)
                cookie = self.getCookie()
                self.doPost(cookie)
                self.setCode()
                self.setHtml()
        print '执行完毕!!'

    def getCookie(self):
        cookie = self.cookieObj.getLoginCookie()
        return cookie

    def doPost(self, cookie):
        #获取配置文件的url
        keysList = []
        urlDict = {}
        keysList = self.iniObj.getKeysBySection('url')
        for key in keysList:
            pageCode = ''
            url = ''
            url = self.iniObj.getValue('url', key)
            urlDict.setdefault(key, url)
            #url = "http://kxyesit.cnsuning.com/"
            try:
                #创建请求的request
                req = urllib2.Request(url)
                handler = urllib2.HTTPCookieProcessor(cookie)
                #利用urllib2的build_opener方法创建一个opener
                openner = urllib2.build_opener(handler)
                responseCode = openner.open(req).getcode()
                #print '页面状态码为:' + str(responseCode)
                pageCode = u'状态码:' + str(responseCode)
                #iniObj.setValue(section='result',key=key,value=pageCode)
                self.iniObjResult.setValue(section='result',
                                           key=key,
                                           value=pageCode)
            except urllib2.URLError, e:
                #print '错误原因:'+e.reason
                reason = e.reason
                pageCode = u'状态码:' + str(e.code) + u';错误原因:' + reason
                #print pageCode
                #iniObj.setValue(section='result',key=key,value=pageCode)
                self.iniObjResult.setValue(section='result',
                                           key=key,
                                           value=pageCode)
Ejemplo n.º 24
0
class RunCode:
    def __init__(self):
        self.cookieObj = CookieUtil()
        self.nowTile = time.strftime("%Y%m%d%H%M%S")
        self.nowPath = os.getcwd()
        if os.path.exists(self.nowPath + "/../log") == False:
            os.mkdir(self.nowPath + "/../log")
        if os.path.exists(self.nowPath + "/../result") == False:
            os.mkdir(self.nowPath + "/../result")
        self.resultIniPath = self.nowPath + "/../log/result_" + self.nowTile + ".ini"
        self.htmlPath = self.nowPath + "/../result/result_" + self.nowTile + ".html"
        self.check = Check()
        self.logUtil = LogUtil()

    def run(self):
        checkKeyResult = self.check.checkIniKeys()
        # key重复性校验失败
        if checkKeyResult:
            errorKeyPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
            self.logUtil.setKeyLog(errorKeyPath, checkKeyResult)
        # 校验成功
        else:
            checkUrlResult = self.check.checkIniUrl()
            # url合法性校验失败
            if checkUrlResult:
                errorUrlPath = self.nowPath + "/../log/result_" + self.nowTile + "_error.ini"
                self.logUtil.setUrlLog(errorUrlPath, checkUrlResult)
            # 校验成功
            else:
                self.iniObj = IniUtil(self.nowPath + "/../config/url.ini")
                self.iniObjResult = IniUtil(self.resultIniPath)
                cookie = self.getCookie()
                self.doPost(cookie)
                self.setCode()
                self.setHtml()
        print "执行完毕!!"

    def getCookie(self):
        cookie = self.cookieObj.getLoginCookie()
        return cookie

    def doPost(self, cookie):
        # 获取配置文件的url
        keysList = []
        urlDict = {}
        keysList = self.iniObj.getKeysBySection("url")
        for key in keysList:
            pageCode = ""
            url = ""
            url = self.iniObj.getValue("url", key)
            urlDict.setdefault(key, url)
            # url = "http://kxyesit.cnsuning.com/"
            try:
                # 创建请求的request
                req = urllib2.Request(url)
                handler = urllib2.HTTPCookieProcessor(cookie)
                # 利用urllib2的build_opener方法创建一个opener
                openner = urllib2.build_opener(handler)
                responseCode = openner.open(req).getcode()
                # print '页面状态码为:' + str(responseCode)
                pageCode = u"状态码:" + str(responseCode)
                # iniObj.setValue(section='result',key=key,value=pageCode)
                self.iniObjResult.setValue(section="result", key=key, value=pageCode)
            except urllib2.URLError, e:
                # print '错误原因:'+e.reason
                reason = e.reason
                pageCode = u"状态码:" + str(e.code) + u";错误原因:" + reason
                # print pageCode
                # iniObj.setValue(section='result',key=key,value=pageCode)
                self.iniObjResult.setValue(section="result", key=key, value=pageCode)