class Browser:
def __init__(self, db):
self.db = db
self.findex = Findex(db)
self.env = {}
self.files = None
def parse_incoming_path(self, path):
self.env["isdir"] = path.endswith("/")
spl = path.split("/")
self.env["host"] = spl[0]
self.env["file_path"] = "/" + "/".join(spl[1:-1])
if not self.env["isdir"]:
self.env["file_name"] = path.split("/")[-1]
if self.env["file_path"] != "/":
self.env["file_path"] += "/"
self.env["file_path_quoted"] = quote_plus(self.env["file_path"])
def fetch_files(self):
host = self.db.query(Hosts).filter_by(address=self.env["host"]).first()
if not host:
raise BrowseException("No host found")
self.env["host_id"] = host.id
files = self.findex.get_files_objects(host_id=host.id, file_path=self.env["file_path_quoted"])
if not files:
raise BrowseException("No files found")
self.files = files
def prepare_files(self, sort=None):
# sort files
self.files = sorted(self.files, key=lambda k: k.file_name) # alphabetically
self.files = sorted(self.files, key=lambda k: k.file_isdir, reverse=True) # folders always on top
if not self.env["file_path"] == "/": # add CDUP dirs
x = Files(
file_name="..",
file_path="../",
file_ext="",
file_format=-1,
file_isdir=True,
file_modified=datetime.now(),
file_perm=None,
searchable=None,
file_size=0,
host=self.env["host"],
)
setattr(x, "file_name_human", "..")
self.files.insert(0, x)
self.files = self.findex.set_icons(self.files)
def sort(self):
# calculate total folder file size (non-recursive)
total_size = 0
for f in self.files:
total_size += f.file_size
def generate_action_fetches(self):
url = "ftp://%s" % self.env["host"]
if self.env["file_path"] == "/":
path = ""
elif self.env["file_path"].startswith("/") and url.endswith("/"):
path = self.env["file_path"][1:]
wget_extras = ""
lftp_extras = ""
# if self.source.crawl_authtype:
# wget_extras = 'user=%s password=%s ' % (self.source.crawl_username, self.source.crawl_password)
#
# if self.source.crawl_authtype == 'HTTP_DIGEST':
# lftp_extras = Debug('Authentication using DIGEST is not supported by lftp')
# else:
# lftp_extras = '-u %s,%s ' % (self.source.crawl_username, self.source.crawl_password)
wget = "wget %s-r -nH --no-parent '%s'" % (wget_extras, url + self.env["file_path"])
# if not isinstance(lftp_extras, Debug):
lftp = "lftp %s-e 'mirror' '%s'" % (lftp_extras, url + self.env["file_path"])
# else:
# lftp = lftp_extras.message
return dict(wget=wget, lftp=lftp)
def breadcrumbs(self):
data = [self.env["host"]]
data += [z for z in self.env["file_path"].split("/")[1:] if z]
return data
def output_json(self):
data = []
for source_file in self.files:
if source_file.filename_human == "..":
continue
data.append(
"[%s] %s%s%s"
% (
"D" if source_file.is_directory else "F",
self.folder.source.crawl_url,
source_file.filepath_human,
source_file.filename_human,
)
)
return "\n".join(data)
class Browse():
def __init__(self, cfg, db):
self.cfg = cfg
self.db = db
self.findex = Findex(db=self.db)
@data_strap
def hosts(self, env):
data = {}
data['hosts'] = self.db.query(Hosts).all()
return jinja2_template('main/browse_hosts', env=env, data=data)
@data_strap
def browse(self, path, env):
env['load_dbtime'] = 0
browser = Browser(db=self.db)
try:
browser.parse_incoming_path(path)
start_dbtime = datetime.now()
browser.fetch_files()
env['load_dbtime'] = (datetime.now() - start_dbtime).total_seconds()
browser.prepare_files()
data = {
'files': browser.files,
'breadcrumbs': browser.breadcrumbs(),
'action_fetches': browser.generate_action_fetches(),
'env': browser.env
}
return jinja2_template('main/browse_dir', env=env, data=data)
except Exception as ex:
return jinja2_template('main/error', env=env, data={
'error': 'no files were found'
})
return ''
@data_strap
def goto(self, path, env):
try:
uid = int(path)
f = self.findex.get_files_objects(id=uid)
if not f:
raise Exception()
f = f[0]
h = self.db.query(Hosts).filter_by(
id=f.host_id
).first()
if f and h:
data = {
'file': f,
'host': h
}
else:
raise Exception()
return jinja2_template('main/browse_goto', env=env, data=data)
except Exception as ex:
return 'error :( we could always stay here'
class Browse():
def __init__(self, cfg, db):
self.cfg = cfg
self.db = db
self.findex = Findex(db=self.db)
@data_strap
def hosts(self, env):
data = {}
data['hosts'] = self.db.query(Hosts).all()
return jinja2_template('main/browse_hosts', env=env, data=data)
@data_strap
def browse(self, path, env):
env['load_dbtime'] = 0
browser = Browser(db=self.db)
try:
browser.parse_incoming_path(path)
start_dbtime = datetime.now()
browser.fetch_files()
env['load_dbtime'] = (datetime.now() -
start_dbtime).total_seconds()
browser.prepare_files()
data = {
'files': browser.files,
'breadcrumbs': browser.breadcrumbs(),
'action_fetches': browser.generate_action_fetches(),
'env': browser.env
}
return jinja2_template('main/browse_dir', env=env, data=data)
except Exception as ex:
return jinja2_template('main/error',
env=env,
data={'error': 'no files were found'})
return ''
@data_strap
def goto(self, path, env):
try:
uid = int(path)
f = self.findex.get_files_objects(id=uid)
if not f:
raise Exception()
f = f[0]
h = self.db.query(Hosts).filter_by(id=f.host_id).first()
if f and h:
data = {'file': f, 'host': h}
else:
raise Exception()
return jinja2_template('main/browse_goto', env=env, data=data)
except Exception as ex:
return 'error :( we could always stay here'
class Browser():
def __init__(self, db):
self.db = db
self.findex = Findex(db)
self.env = {}
self.files = None
def parse_incoming_path(self, path):
self.env['isdir'] = path.endswith('/')
spl = path.split('/')
self.env['host'] = spl[0]
self.env['file_path'] = '/' + '/'.join(spl[1:-1])
if not self.env['isdir']: self.env['file_name'] = path.split('/')[-1]
if self.env['file_path'] != '/': self.env['file_path'] += '/'
self.env['file_path_quoted'] = quote_plus(self.env['file_path'])
def fetch_files(self):
host = self.db.query(Hosts).filter_by(
address=self.env['host']
).first()
if not host:
raise BrowseException('No host found')
self.env['host_id'] = host.id
files = self.findex.get_files_objects(host_id=host.id, file_path=self.env['file_path_quoted'])
if not files:
raise BrowseException('No files found')
self.files = files
def prepare_files(self, sort=None):
# sort files
self.files = sorted(self.files, key=lambda k: k.file_name) # alphabetically
self.files = sorted(self.files, key=lambda k: k.file_isdir, reverse=True) # folders always on top
if not self.env['file_path'] == '/': # add CDUP dirs
x = Files(
file_name='..', file_path='../', file_ext='', file_format=-1,
file_isdir=True, file_modified=datetime.now(), file_perm=None, searchable=None,
file_size=0, host=self.env['host']
)
setattr(x, 'file_name_human', '..')
self.files.insert(0, x)
self.files = self.findex.set_icons(self.files)
def sort(self):
# calculate total folder file size (non-recursive)
total_size = 0
for f in self.files:
total_size += f.file_size
def generate_action_fetches(self):
url = 'ftp://%s' % self.env['host']
if self.env['file_path'] == '/':
path = ''
elif self.env['file_path'].startswith('/') and url.endswith('/'):
path = self.env['file_path'][1:]
wget_extras = ''
lftp_extras = ''
# if self.source.crawl_authtype:
# wget_extras = 'user=%s password=%s ' % (self.source.crawl_username, self.source.crawl_password)
#
# if self.source.crawl_authtype == 'HTTP_DIGEST':
# lftp_extras = Debug('Authentication using DIGEST is not supported by lftp')
# else:
# lftp_extras = '-u %s,%s ' % (self.source.crawl_username, self.source.crawl_password)
wget = 'wget %s-r -nH --no-parent \'%s\'' % (wget_extras, url + self.env['file_path'])
#if not isinstance(lftp_extras, Debug):
lftp = 'lftp %s-e \'mirror\' \'%s\'' % (lftp_extras, url + self.env['file_path'])
#else:
# lftp = lftp_extras.message
return dict(wget=wget, lftp=lftp)
def breadcrumbs(self):
data = [self.env['host']]
data += [z for z in self.env['file_path'].split('/')[1:] if z]
return data
def output_json(self):
data = []
for source_file in self.files:
if source_file.filename_human == '..':
continue
data.append('[%s] %s%s%s' % (
'D' if source_file.is_directory else 'F',
self.folder.source.crawl_url,
source_file.filepath_human,
source_file.filename_human))
return '\n'.join(data)
