Split some code into libs directory
This commit is contained in:
parent
3028c4646e
commit
84b909e7cb
2
libs/__init__.py
Normal file
2
libs/__init__.py
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
import files
|
||||||
|
import config
|
||||||
BIN
libs/__init__.pyc
Normal file
BIN
libs/__init__.pyc
Normal file
Binary file not shown.
12
libs/config.py
Normal file
12
libs/config.py
Normal file
@ -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
|
||||||
|
|
||||||
|
|
||||||
BIN
libs/config.pyc
Normal file
BIN
libs/config.pyc
Normal file
Binary file not shown.
15
libs/files.py
Normal file
15
libs/files.py
Normal file
@ -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
|
||||||
|
|
||||||
BIN
libs/files.pyc
Normal file
BIN
libs/files.pyc
Normal file
Binary file not shown.
13
main.py
13
main.py
@ -5,6 +5,7 @@ import signal
|
|||||||
import dns.resolver
|
import dns.resolver
|
||||||
import json
|
import json
|
||||||
from pprint import pprint
|
from pprint import pprint
|
||||||
|
from libs import config
|
||||||
|
|
||||||
running = 0
|
running = 0
|
||||||
prompt = '$ '
|
prompt = '$ '
|
||||||
@ -13,14 +14,6 @@ whoisrecs = ["NotFound", "DomainName", "Registrar", "Status", "UpdatedDate", "Cr
|
|||||||
subdomains = ["www", "mail", "ftp"]
|
subdomains = ["www", "mail", "ftp"]
|
||||||
dnsrecords = ["A", "MX", "TXT", "NS", "SOA"]
|
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):
|
def domainWhois(inData):
|
||||||
whoisData = whois.whois(inData).query()
|
whoisData = whois.whois(inData).query()
|
||||||
return whois.Parser(inData, whoisData[1]).parse()
|
return whois.Parser(inData, whoisData[1]).parse()
|
||||||
@ -50,8 +43,8 @@ def parseInput(inData):
|
|||||||
|
|
||||||
signal.signal(signal.SIGINT, signal_handler)
|
signal.signal(signal.SIGINT, signal_handler)
|
||||||
|
|
||||||
jsonData = json.loads(openFile("config.json").read())
|
jsonData = config.readConf()
|
||||||
|
pprint(jsonData)
|
||||||
|
|
||||||
print "root"
|
print "root"
|
||||||
for rdata in dns.resolver.query('yahoo.com', 'A') :
|
for rdata in dns.resolver.query('yahoo.com', 'A') :
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user