diff --git a/libs/__init__.py b/libs/__init__.py new file mode 100644 index 0000000..5f030a2 --- /dev/null +++ b/libs/__init__.py @@ -0,0 +1,2 @@ +import files +import config diff --git a/libs/__init__.pyc b/libs/__init__.pyc new file mode 100644 index 0000000..7febc6b Binary files /dev/null and b/libs/__init__.pyc differ diff --git a/libs/config.py b/libs/config.py new file mode 100644 index 0000000..9b0820d --- /dev/null +++ b/libs/config.py @@ -0,0 +1,12 @@ +import files as f +import json + +def readConf(): + try: + jsonData = json.loads(f.openFile("config.json").read()) + except Exception as e: + print e + jasonData = "" + return jsonData + + diff --git a/libs/config.pyc b/libs/config.pyc new file mode 100644 index 0000000..40e9332 Binary files /dev/null and b/libs/config.pyc differ diff --git a/libs/files.py b/libs/files.py new file mode 100644 index 0000000..b3dd1cf --- /dev/null +++ b/libs/files.py @@ -0,0 +1,15 @@ +def openFile(path): + try: + myFile = open(path) + except Exception as e: + print e + pass + return myFile + +def closeFile(fileObj): + try: + fileObj.close() + except Exception as e: + print e + pass + diff --git a/libs/files.pyc b/libs/files.pyc new file mode 100644 index 0000000..3ffdce9 Binary files /dev/null and b/libs/files.pyc differ diff --git a/main.py b/main.py index f730757..6b2712e 100755 --- a/main.py +++ b/main.py @@ -5,6 +5,7 @@ import signal import dns.resolver import json from pprint import pprint +from libs import config running = 0 prompt = '$ ' @@ -13,14 +14,6 @@ whoisrecs = ["NotFound", "DomainName", "Registrar", "Status", "UpdatedDate", "Cr subdomains = ["www", "mail", "ftp"] dnsrecords = ["A", "MX", "TXT", "NS", "SOA"] -def openFile(path): - try: - myfile = open(path) - except Exception as E: - print E + "< - most likely file not found" - pass - return myfile - def domainWhois(inData): whoisData = whois.whois(inData).query() return whois.Parser(inData, whoisData[1]).parse() @@ -50,8 +43,8 @@ def parseInput(inData): signal.signal(signal.SIGINT, signal_handler) -jsonData = json.loads(openFile("config.json").read()) - +jsonData = config.readConf() +pprint(jsonData) print "root" for rdata in dns.resolver.query('yahoo.com', 'A') :