rra
/
Border-Check
mirror of https://github.com/rscmbbng/Border-Check
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

solved indentation erros

pull/3/head
rscmbbng 11 years ago
parent
c1cd4eb67e
commit
cba0658121
1 changed files with 58 additions and 59 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 107
      main.py

107
main.py
View File

@ -1,4 +1,4 @@
#!/usr/bin/python #!/usr/local/bin/python
# -*- coding: iso-8859-15 -*- # -*- coding: iso-8859-15 -*-
""" """
BC (Border-Check) is a tool to retrieve info of traceroute tests over website navigation routes. BC (Border-Check) is a tool to retrieve info of traceroute tests over website navigation routes.
@ -26,12 +26,12 @@ DEBUG = 1
class bc(object): class bc(object):
""" """
BC main Class BC main Class
""" """
def __init__(self): def __init__(self):
""" """
Init defaults Init defaults
""" """
self.browser = "" # "F" Firefox / "C" Chrome self.browser = "" # "F" Firefox / "C" Chrome
self.browser_path = "" self.browser_path = ""
self.url = "" self.url = ""
@ -39,14 +39,14 @@ class bc(object):
def set_options(self, options): def set_options(self, options):
""" """
Set program options Set program options
""" """
self.options = options self.options = options
def create_options(self, args=None): def create_options(self, args=None):
""" """
Create options for OptionParser Create options for OptionParser
""" """
self.optionParser = BCOptions() self.optionParser = BCOptions()
self.options = self.optionParser.get_options(args) self.options = self.optionParser.get_options(args)
if not self.options: if not self.options:
@ -55,8 +55,8 @@ class bc(object):
def try_running(self, func, error, args=None): def try_running(self, func, error, args=None):
""" """
Try running a function and print some error if it fails and exists with a fatal error. Try running a function and print some error if it fails and exists with a fatal error.
""" """
options = self.options options = self.options
args = args or [] args = args or []
try: try:
@ -72,8 +72,8 @@ class bc(object):
def check_browser(self): def check_browser(self):
""" """
Check for browser used by system Check for browser used by system
""" """
if sys.platform == 'darwin': if sys.platform == 'darwin':
f_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Firefox/Profiles') f_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Firefox/Profiles')
c_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Google/Chrome/Default/History') c_osx = os.path.join(os.path.expanduser('~'), 'Library/Application Support/Google/Chrome/Default/History')
@ -81,26 +81,26 @@ class bc(object):
try: try:
if os.path.exists(f_osx): if os.path.exists(f_osx):
if len(os.listdir(f_osx)) > 2: if len(os.listdir(f_osx)) > 2:
print 'you have multiple profiles, choosing the last one used' print 'you have multiple profiles, choosing the last one used'
#filtering the directory that was last modified #filtering the directory that was last modified
all_subdirs = [os.path.join(f_osx,d)for d in os.listdir(f_osx)] all_subdirs = [os.path.join(f_osx,d)for d in os.listdir(f_osx)]
try: try:
all_subdirs.remove(os.path.join(f_osx,'.DS_Store')) #throwing out .DS_store all_subdirs.remove(os.path.join(f_osx,'.DS_Store')) #throwing out .DS_store
except: except:
pass pass
latest_subdir = max(all_subdirs, key=os.path.getmtime) latest_subdir = max(all_subdirs, key=os.path.getmtime)
osx_profile = os.path.join(f_osx, latest_subdir) osx_profile = os.path.join(f_osx, latest_subdir)
osx_history_path = os.path.join(osx_profile, 'places.sqlite') osx_history_path = os.path.join(osx_profile, 'places.sqlite')
self.browser_path = osx_history_path self.browser_path = osx_history_path
else: else:
for folder in os.listdir(f_osx): for folder in os.listdir(f_osx):
if folder.endswith('.default'): if folder.endswith('.default'):
osx_default = os.path.join(f_osx, folder) osx_default = os.path.join(f_osx, folder)
osx_history_path = os.path.join(osx_default, 'places.sqlite') osx_history_path = os.path.join(osx_default, 'places.sqlite')
print "setting:", osx_history_path, "as history file" print "setting:", osx_history_path, "as history file"
self.browser_path = osx_history_path self.browser_path = osx_history_path
self.browser = "F" self.browser = "F"
@ -140,8 +140,8 @@ class bc(object):
def getURL(self): def getURL(self):
""" """
Set urls to visit Set urls to visit
""" """
print "Browser database:", self.browser_path, "\n" print "Browser database:", self.browser_path, "\n"
conn = sqlite3.connect(self.browser_path) conn = sqlite3.connect(self.browser_path)
c = conn.cursor() c = conn.cursor()
@ -164,8 +164,8 @@ class bc(object):
def getGEO(self): def getGEO(self):
""" """
Get Geolocation database (http://dev.maxmind.com/geoip/legacy/geolite/) Get Geolocation database (http://dev.maxmind.com/geoip/legacy/geolite/)
""" """
# Download and extract database # Download and extract database
try: try:
urllib.urlretrieve('http://xsser.sf.net/map/GeoLiteCity.dat.gz', urllib.urlretrieve('http://xsser.sf.net/map/GeoLiteCity.dat.gz',
@ -183,8 +183,8 @@ class bc(object):
def run(self, opts=None): def run(self, opts=None):
""" """
Run BorderCheck Run BorderCheck
""" """
#eprint = sys.stderr.write #eprint = sys.stderr.write
# set options # set options
if opts: if opts:
@ -206,19 +206,19 @@ class bc(object):
BorderCheckWebserver(self) BorderCheckWebserver(self)
while True: while True:
url = urlparse(self.url[0]).netloc url = urlparse(self.url[0]).netloc
url = url.replace('www.','') #--> doing a tracert to for example.com and www.example.com yields different results most of the times. url = url.replace('www.','') #--> doing a tracert to for example.com and www.example.com yields different results most of the times.
url_ip = socket.gethostbyname(url) url_ip = socket.gethostbyname(url)
print url_ip print url_ip
if url != self.old_url: if url != self.old_url:
count = 0 count = 0
print url print url
a = subprocess.Popen(['lft', '-S', '-n', '-E', url_ip], stdout=subprocess.PIPE) # -> using tcp a = subprocess.Popen(['lft', '-S', '-n', '-E', url_ip], stdout=subprocess.PIPE) # -> using tcp
#a = subprocess.Popen(['lft', '-S', '-n', '-u', url_ip], stdout=subprocess.PIPE) # -> using udp #a = subprocess.Popen(['lft', '-S', '-n', '-u', url_ip], stdout=subprocess.PIPE) # -> using udp
logfile = open('logfile', 'a') logfile = open('logfile', 'a')
for line in a.stdout: for line in a.stdout:
logfile.write(line) logfile.write(line)
parts = line.split() parts = line.split()
for ip in parts: for ip in parts:
@ -239,10 +239,10 @@ class bc(object):
print "Not more records. Aborting...", "\n" print "Not more records. Aborting...", "\n"
exit() exit()
self.old_url = url self.old_url = url
print"old url =", self.old_url print"old url =", self.old_url
logfile.close() logfile.close()
time.sleep(5) time.sleep(5)
if __name__ == "__main__": if __name__ == "__main__":
app = bc() app = bc()
@ -250,4 +250,3 @@ if __name__ == "__main__":
if options: if options:
app.set_options(options) app.set_options(options)
app.run() app.run()

Write Preview
Loading…
Cancel
Save