Bebop/bebop/__main__.py
2021-03-28 18:28:54 +02:00

31 lines
736 B
Python

import argparse
from bebop.browser.browser import Browser
from bebop.fs import get_user_data_path
from bebop.tofu import load_cert_stash, save_cert_stash
def main():
argparser = argparse.ArgumentParser()
argparser.add_argument("url", default=None)
args = argparser.parse_args()
if args.url:
start_url = args.url
else:
start_url = None
user_data_path = get_user_data_path()
if not user_data_path.exists():
user_data_path.mkdir()
cert_stash_path = user_data_path / "known_hosts.txt"
cert_stash = load_cert_stash(cert_stash_path) or {}
try:
Browser(cert_stash).run(start_url=start_url)
finally:
save_cert_stash(cert_stash, cert_stash_path)
main()