python 探测网站目录的GUI程序-爱站

1.pyqt4写的界面 find_ui.py

#-*- coding: utf-8 -*-
from PyQt4import QtCore, QtGui
 
try:
    _fromUtf8= QtCore.QString.fromUtf8
except AttributeError:
    def _fromUtf8(s):
        return s
 
try:
    _encoding= QtGui.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig)
 
class Ui_Form(object):
    def setupUi(self, Form):
        Form.setObjectName(_fromUtf8("Form"))
        Form.resize(516,467)
        self.label= QtGui.QLabel(Form)
        self.label.setGeometry(QtCore.QRect(20,10,54,16))
        self.label.setObjectName(_fromUtf8("label"))
        self.edit_address= QtGui.QLineEdit(Form)
        self.edit_address.setGeometry(QtCore.QRect(80,10,351,20))
        self.edit_address.setObjectName(_fromUtf8("edit_address"))
        self.button_search= QtGui.QPushButton(Form)
        self.button_search.setGeometry(QtCore.QRect(440,10,61,23))
        self.button_search.setObjectName(_fromUtf8("button_search"))
        self.text_all= QtGui.QTextEdit(Form)
        self.text_all.setGeometry(QtCore.QRect(20,40,411,261))
        self.text_all.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
        self.text_all.setObjectName(_fromUtf8("text_all"))
        self.label_2= QtGui.QLabel(Form)
        self.label_2.setGeometry(QtCore.QRect(20,320,54,12))
        self.label_2.setObjectName(_fromUtf8("label_2"))
        self.text_exist= QtGui.QTextEdit(Form)
        self.text_exist.setGeometry(QtCore.QRect(20,340,411,64))
        self.text_exist.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
        self.text_exist.setObjectName(_fromUtf8("text_exist"))
        self.label_3= QtGui.QLabel(Form)
        self.label_3.setGeometry(QtCore.QRect(380,310,91,20))
        self.label_3.setText(_fromUtf8(""))
        self.label_3.setObjectName(_fromUtf8("label_3"))
        self.edit_add= QtGui.QLineEdit(Form)
        self.edit_add.setGeometry(QtCore.QRect(20,420,411,20))
        self.edit_add.setObjectName(_fromUtf8("edit_add"))
        self.button_add= QtGui.QPushButton(Form)
        self.button_add.setGeometry(QtCore.QRect(440,420,71,23))
        self.button_add.setObjectName(_fromUtf8("button_add"))
        self.label_4= QtGui.QLabel(Form)
        self.label_4.setGeometry(QtCore.QRect(20,440,251,16))
        self.label_4.setObjectName(_fromUtf8("label_4"))
 
        self.retranslateUi(Form)
        QtCore.QMetaObject.connectSlotsByName(Form)
 
    def retranslateUi(self, Form):
        Form.setWindowTitle(_translate("Form","目录探测工具",None))
        self.label.setText(_translate("Form","网站地址：",None))
        self.button_search.setText(_translate("Form","探测",None))
        self.label_2.setText(_translate("Form","结果：",None))
        self.button_add.setText(_translate("Form","添加地址",None))
        self.label_4.setText(_translate("Form","例如：/admin/manager.asp  请以斜杠开始",None))

2. 启动文件 start.py

100

101

102

#!/usr/local/bin/python
#coding=utf-8
 
import sys
import os
import time
import httplib
import re
from PyQt4import QtCore, QtGui
from threadingimport Thread
 
from find_uiimport Ui_Form
 
 
class MyForm(QtGui.QMainWindow):
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        QtCore.QObject.connect(self.ui.button_search,QtCore.SIGNAL("clicked()"), self.startthread)
        QtCore.QObject.connect(self.ui.button_add,QtCore.SIGNAL("clicked()"), self.addAddress)
    def startSearch(self):
        self.ui.label_3.setText("")
        self.getAddress()
        address=str(self.ui.edit_address.text())
        self.accessAddesss(address)
         
    def startthread(self):
        t1=Thread(target=self.startSearch,)
        t1.start()
         
         
    def getAddress(self):
        try:
            global addresslist
            addresslist=[]
            filePath=os.getcwd()+"\\address.txt"
           #if not os.path.isfile(filePath):
            #    print'aaa'
             #  return 0
                
            fileAddress=file(filePath,"r")
            for address_linein fileAddress.readlines():
                if address_line notin addresslist:
                    addresslist.append(address_line)
                    pass
                pass
            pass
            fileAddress.close()
             
        except:
            #self.ui.text_all.setText('aaa')
            self.ui.text_all.setText(u'打开文件错误')
            pass
        finally:
            #fileAddress.close()
            pass
       # print addresslist[0]
     
    def accessAddesss(self,host):
        try:
            print host
            print len(addresslist)
            for oneAddressin addresslist:
                print len(addresslist)
                oneAddress=oneAddress.replace("\n","")
                print oneAddress
                connection=httplib.HTTPConnection(host,80,timeout=10)
                connection.request("GET",oneAddress)
                response=connection.getresponse()
                result=response.reason
                resultNum=response.status
                 
                if "OK" in result or"Forbidden" in result:
                    getaddress="http://"+host+oneAddress+"------"+str(resultNum)+":"+result
                    self.ui.text_exist.append(getaddress)
                else:
                    self.ui.text_all.append("http://"+host+oneAddress+"------"+str(resultNum)+":"+result)
                     
                connection.close()
        except Exceptionas e:
            print e.message
        self.ui.label_3.setText(u"探测完成")
        self.ui.label_3.colorCount()
        
    def  addAddress(self):
        try:
            filePath=os.getcwd()+"\\address.txt"
            fileAddress=file(filePath,"a")
            newAddress="\n"+str(self.ui.edit_add.text())
            print newAddress
            fileAddress.write(newAddress)
            fileAddress.close()
        except Exceptionas e:
            print e.message
         
 
if __name__ =="__main__":
    app = QtGui.QApplication(sys.argv)
    myapp = MyForm()
    myapp.show()
    sys.exit(app.exec_())