128 lines
4.3 KiB
Python
128 lines
4.3 KiB
Python
from database import Database
|
|
import scrapper
|
|
import utils
|
|
from utils import get
|
|
import platform
|
|
import urllib
|
|
import colorama
|
|
from terminal_menu.Menu import Menu
|
|
|
|
|
|
uri = ""
|
|
preview_site = {}
|
|
preview_database = ""
|
|
|
|
def get_user_choice_by_menu(options: list, title = None, preview_command=None, preview_size: float = 0.3, show_search_hint: bool = False, skip_empty_entries = False, cursor_pos: int = 0, preview_args=None):
|
|
menu = Menu(options,
|
|
title,
|
|
preview_body_function=preview_command,
|
|
preview_ratio=preview_size,
|
|
skip_empty_option=skip_empty_entries,
|
|
cursor_pos=cursor_pos,
|
|
preview_args=preview_args)
|
|
return (menu.show());
|
|
|
|
def add_site(database: Database) -> dict:
|
|
print("add a site:")
|
|
site = {}
|
|
site.update({"url": input("url without protocol (ip:port):")})
|
|
site.update({"user": input("user(leave blank):")})
|
|
site.update({"password": input("password(leave blank):")})
|
|
site.update({"id": database.get_sites_table_len()})
|
|
name = input(f"name[{str(database.get_sites_table_len())}]")
|
|
if (name == ""):
|
|
name = database.get_sites_table_len()
|
|
site.update({"name": name})
|
|
return (site)
|
|
|
|
def files_preview(url: str) -> str:
|
|
if (not url.endswith("/")):
|
|
return (None)
|
|
files = scrapper.get_files_by_url(url);
|
|
return ("\n".join(files))
|
|
|
|
def get_files_formated_by_viewing_data(database: Database, files: list, site: dict, path: str):
|
|
out = []
|
|
current_url = utils.get_base_url(site) + urllib.parse.quote(path);
|
|
for file in files:
|
|
if file == "..":
|
|
out.append(file);
|
|
continue
|
|
url = current_url + urllib.parse.quote(file)
|
|
viewing_data = database.get_viewing_data(url);
|
|
if (viewing_data == None) or (viewing_data.get("finished") == False):
|
|
out.append(colorama.Style.BRIGHT + file)
|
|
else:
|
|
out.append(colorama.Fore.LIGHTBLACK_EX + file);
|
|
return (out);
|
|
|
|
def get_files_path(current_url: str, files: list):
|
|
out = []
|
|
for file in files:
|
|
out.append(current_url + file);
|
|
return (out);
|
|
|
|
def files(site:dict, path:str, database:Database, pos:int):
|
|
|
|
current_url = scrapper.get_url(site, path);
|
|
files = scrapper.get_files(site, path)
|
|
files_displayed = get_files_formated_by_viewing_data(database, files, site, path)
|
|
preview_args = get_files_path(current_url, files)
|
|
choose = get_user_choice_by_menu(files_displayed,
|
|
preview_command=files_preview,
|
|
preview_args=preview_args,
|
|
preview_size=0.3,
|
|
show_search_hint=True,
|
|
title=f"Index of {path}",
|
|
cursor_pos = pos)
|
|
if (choose == None):
|
|
database.set_last_cursor_pos(0)
|
|
return (None)
|
|
database.set_last_cursor_pos(choose)
|
|
return (files[choose])
|
|
|
|
def config_preview(site_name:str):
|
|
id = int(site_name.split(": ")[0])
|
|
site = preview_database.get_site_by_id(id)
|
|
if (site == None):
|
|
return
|
|
str = f"""
|
|
url: {site.get('url')}
|
|
user: {site.get('user')}
|
|
password: {site.get('password')}
|
|
"""
|
|
return (str)
|
|
|
|
def site_deleter(sites, database: Database):
|
|
global preview_database
|
|
|
|
preview_database = database
|
|
choose = get_user_choice_by_menu(sites,
|
|
show_search_hint=True,
|
|
preview_title="delete",
|
|
preview_command=config_preview)
|
|
return (choose)
|
|
|
|
def param_editor(lst: list):
|
|
choose = get_user_choice_by_menu(lst)
|
|
return (choose)
|
|
|
|
def site_editor(lst: list, database: Database):
|
|
global preview_database
|
|
|
|
preview_database = database
|
|
choose = get_user_choice_by_menu(lst + ["", "quit"],
|
|
skip_empty_entries=True,
|
|
show_search_hint=True,
|
|
title="Edit")
|
|
if (choose == len(lst) + 1):
|
|
return (None)
|
|
return (choose)
|
|
|
|
def sites(sites: list):
|
|
nb_site = len(sites)
|
|
choose = get_user_choice_by_menu(get(sites, "name") + ["", "add", "edit", "delete"],
|
|
skip_empty_entries=True,
|
|
show_search_hint=True)
|
|
return (choose)
|