"""//////////////////////////////////////////////////////////////////
//                                                                 //
//  Script for saving malicious websites to your local disk        //
//                                                 as *.txt file.  //
//                                                                 //
/////////////////////////////////////////////////////////////////////
//                                                                 //
//         Page: http://sefersimsek.com/malicious-domains          //
//                                                                 //
//////////////////////////////////////////////////////////////////"""


try:
    import requests
    import re
    import sys
except ImportError:
    raise ImportError("Error while importing modules")

listUrl = "http://sefersimsek.com/malicious-domains/"

def parse_urls():
    print "[*] Connecting: %s"% listUrl

    req = requests.get(listUrl)
    htmlText = req.text
    return re.findall(',Domain-Name">(.*?)</div>', htmlText)

def save_txt(file):
    print "[+] Saving URLs/IPs to file: %s"% sys.argv[1]

    fd = open(sys.argv[1], "w")
    for ui in file:
        wr = str(ui)
        wr += "\n"
        fd.write(wr)
    fd.close()

def main():
    argc = len(sys.argv)
    if argc < 2:
        print "usage: %s <file_name.txt>"% sys.argv[0]
        sys.exit()

    urList = parse_urls()[1:]

    save_txt(urList)


if __name__ == "__main__":
    main()