Сценарий Python не продолжает сценарий - PullRequest
Новая
       12

Сценарий Python не продолжает сценарий

0 голосов
/ 06 декабря 2018

Привет, ребята, у меня есть скрипт на python, но тоже проблема, после того, как он выполнит проверку по списку слов, он просто остановится, но не остановится, у него есть функция main, которую он может сделать, что я могу сделать?

кодировка: utf-8 

import requests
import argparse

usage = ''': python %(prog)s [-h url] [-f file] [-t type]'''
da = argparse.ArgumentParser(usage=usage)
da.add_argument("-u","--url", type=str, help="", dest="host")
da.add_argument("-f","--file", type=str, help="", dest="wordlist")
da.add_argument("-t","--type",type=str, help="D to directory | A to archives)", dest="type")
args = da.parse_args()

url = args.host #URL a ser testada
tipo = args.type

try:
    arquivo = open(args.wordlist, "r") #ARQUIVO com a wordlist

    #Verificação se a última linha da wordlist está com \n no final

    size = arquivo.readlines()
    sizeH = len(size) - 1
    line = size[sizeH]
    if line.endswith('\n'):
        print '[F]Remova última linha em branco na wordlist'
        exit()
    else:
        print '[V]Wordlist válida'
        print'''
        ==================================
        ============INICIANDO=============
        ==================================
            '''
except Exception as erro:
    print '\n[!] ERROR: '+ str(erro)

if __name__ == '__main__':
    if tipo == "D" or tipo == "d" or tipo == "directory" or tipo == "DIRECTORY": 
        for linhas in arquivo:
            link = url+'/'+linhas[:-1]+'/'
            #link = link1[:-1]
            request = requests.get(link, timeout=3)
            status = request.status_code
            if status == 200:
                print "[+] Destino existente: "+link+" STATUS: "+str(status)
            elif status == 403:
                print "[%] Destino privado: "+link+" STATUS: "+str(status)
            else:
                print "[-] Destino inexistente: "+link+" STATUS: "+str(status)
    elif tipo == "A" or tipo == "a" or tipo == "archives" or tipo == "ARCHIVES":
        for linhas in arquivo:
            link = url+linhas
            request = requests.get(link, timeout=3)
            status = request.status_code
            if status == 200:
                print "[+] Destino existente: "+link+" STATUS: "+str(status)
            elif status == 403:
                print "[%] Destino privado: "+link+" STATUS: "+str(status)
            else:
                print "[-] Destino inexistente: "+link+" STATUS: "+str(status)
    else:
        print '[!] Escolha um tipo de teste'

Я пытаюсь указать имя метода main и вызвать его после проверки, но он не работает.И просто работай, если я удалю подтверждение, но я не хочу удалять.

...