python登录并校验是否成功

ゝ一纸荒年。 2022-05-15 06:20 380阅读 0赞

# python登录并校验是否成功 #

2007年六月11日 by 贝贝爸 [Leave a reply »][Leave a reply]

为了监控网站运行是否正常，写了一个脚本来检查网站运行的情况。  
主要是自动登录并且校验登陆后的内容来判断网站web服务和数据库服务是否正常。  
代码如下：

\#!/usr/bin/env python   
 \# -\*- coding: gbk -\*-   
 import  urllib   
 import  urllib2   
 import  ClientCookie   
 import  re   
 import  string   
 import  time   
 import  socket   
 timeout = 10   
 server = " [www.juyimeng.com][] "   
 loginurl = " http:// " \+ server \+ " /login.php "   
 v\_account = \[   
\{  " name " : " user1 " , " pass " : " pass1 " , " utype " : " 1 " , " vkey " : " 用户1 " \},  
\{  " name " : " user2 " , " pass " : " pass2 " , " utype " : " 2 " , " vkey " : " 用户2 " \}  
 \]   
 def  fes\_write ( s ) :  
     '''   
    控制输出信息的显示  
     '''   
     print  s   
 def  get\_response ( x ) :  
     '''   
    登录，取得登陆后的页面html代码，用来判断结果  
     '''   
     fes\_write ( " login for  " \+  x \[ " name " \])   
     eform  = \{  ' username ' : x \[ " name " \] ,  ' password ' : x \[ " pass " \] ,  ' usertype ' : x \[ " utype " \] \}  
     qstring  =  urllib . urlencode (  eform  )   
     \#print qstring   
     try :  
         request  =  urllib2 . Request (  loginurl  ,  qstring  )   
         response  =  ClientCookie . urlopen (  request  )   
         return  string . split (  response . read () , " \\n " )   
     except  Exception ,  e :  
         fes\_write ( e )   
         return  " error "   
 def  verify\_info ( k , l ) :  
     '''   
    校验信息是否正确  
     '''   
     for  line  in  l :  
         if  k . match ( line ) :  
             return  True   
             break   
     return  False   
 def  verify\_all () :  
     '''   
    主程序，进行校验  
     '''   
     if  timeout != 0 :  
         socket . setdefaulttimeout ( timeout  )   
     fes\_write ( " socket time out:  "  \+  str ( socket . getdefaulttimeout ()))   
     success\_num = len ( v\_account )   
     for  x  in  v\_account :  
         vkey = re . compile ( ' .\* ' \+ x \[ " vkey " \] \+ ' .\* ' )   
         if  verify\_info ( vkey , get\_response ( x )) == True :  
             success\_num = success\_num \- 1   
             fes\_write ( x \[ " name " \] \+  "  login success! " )   
         else :  
             fes\_write ( " error when  "  \+  x \[ " name " \] \+  " login! " )   
     if  success\_num == 0 :  
         fes\_write ( " all things runs well! " )   
     else :  
         e\_notify ()   
         fes\_write ( " some thing is wrong! " )   
          
 def  e\_notify () :  
     '''   
    出错之后的notify操作  
     '''   
     pass   
 if  \_\_name\_\_  ==  ' \_\_main\_\_ ' :  
     verify\_all ()

v\_account定义了多个账户，因为用户可能不同，后台机制不一样，所以用多个用户来登录  
request = urllib2.Request( loginurl , qstring ) 用的是post的方法，也可以用get的方法。  
ClientCookie,一个python的扩展包，安装请[参考这里][Link 1]  
2个v\_account里的vkey用来和登陆后的页面上的信息来验证是否正常登录的  
socket.setdefaulttimeout用来设置建立socket连接的超时设置，原来没有设置，程序会一直等待回应(如果服务器端做了最长运行时间会好点)，设置了10秒，如果一个登录程序10秒还没有成功，基本上就失败了。  
e\_notify:可以自己设置一些notify的方法，比如邮件，短消息等。

Popularity: 19%

[Leave a reply]: http://www.juyimeng.com/python-simulate-login-with-post-and-examine-the-result.html#comments
[www.juyimeng.com]: http://www.juyimeng.com
[Link 1]: http://www.juyimeng.com/install-activepython-setuptools-clientcookie-linux.html