def dal_pristine():
dal = DAL(env_config=config)
dal.sim_results_db_url = "sqlite:///:memory:"
with dal as dal:
# noinspection PyProtectedMember
metadata.create_all(dal._sql_alchemy_db_engine)
yield dal
def tags(
# ctx: typer.Context,
tag: str = typer.
Argument(
None,
help=(
"tag for which related tags should be shown. No input: all tags are"
" printed."),
),
verbose: bool = typer.Option(False, "-v", "--verbose"),
):
"""
No parameter: Show all tags
With tag as parameter:
Show related tags, i.e. tags which are used in combination with tag.
"""
if verbose:
typer.echo(f"Using DB: {config.twbm_db_url}", err=True)
if tag is not None:
tag = tag.strip(",").strip().lower()
with DAL(env_config=config) as dal:
if tag is None:
tags = dal.get_all_tags()
else:
tags = dal.get_related_tags(tag=tag)
for t in tags:
typer.echo(f"{t[1]:>4}: {t[0]}", err=True)
def _update_tags(
ids: Sequence[int],
tags: Optional[Sequence[str]] = None,
tags_not: Optional[Sequence[str]] = None,
force: bool = False,
):
bms = Bookmarks(fts_query="").bms
if tags is None:
tags = ()
if tags_not is None:
tags_not = ()
with DAL(env_config=config) as dal:
# to allow for redefinition: set -> list
# (https://mypy.readthedocs.io/en/stable/common_issues.html#redefinitions-with-incompatible-types)
new_tags: Iterable
for id in ids:
bm = bms[id - 1]
if force:
new_tags = set(tags)
else:
new_tags = (set(bm.split_tags) | set(tags)) - set(tags_not)
new_tags = sorted(new_tags)
new_tags_str = f",{','.join(new_tags)},"
_log.debug(f"{new_tags_str=}")
bm.tags = new_tags_str
dal.update_bookmark(bm)
show_bms((bm, ))
def dal(init_db):
dal = DAL(env_config=config)
with dal as dal:
sql_files_path = Path(__file__).parent.absolute() / "sql"
aiosql_queries = aiosql.from_path(f"{sql_files_path}", "sqlite3")
aiosql_queries.load_testdata(dal.conn.connection)
dal.conn.connection.commit()
yield dal
def show(
# ctx: typer.Context,
id_: int = typer.Argument(..., help="id to print"),
verbose: bool = typer.Option(False, "-v", "--verbose"),
):
if verbose:
typer.echo(f"Using DB: {config.twbm_db_url}", err=True)
# _ = BukuDb(dbfile=config.dbfile).print_rec(index=id_) # noqa E800
with DAL(env_config=config) as dal:
bm = dal.get_bookmark_by_id(id_=id_)
show_bms((bm, ))
def delete(
# ctx: typer.Context,
id_: int = typer.Argument(..., help="id to delete"),
verbose: bool = typer.Option(False, "-v", "--verbose"),
):
if verbose:
typer.echo(f"Using DB: {config.twbm_db_url}", err=True)
# use buku because of DB compactdb
with DAL(env_config=config) as dal:
bm = dal.get_bookmark_by_id(id_=id_)
show_bms((bm, ))
_ = BukuDb(dbfile=config.dbfile).delete_rec(index=id_, delay_commit=False)
def open(
# ctx: typer.Context,
ids: str = typer.Argument(
None, help="list of ids, separated by comma, no blanks"),
verbose: bool = typer.Option(False, "-v", "--verbose"),
):
"""
Opens bookmarks
Gotcha:
in order to allow for piped input, ids must be separated by comma with no blanks.
Example for using piped input:
twbm search xxxxx | twbm open
"""
if verbose:
typer.echo(f"Using DB: {config.twbm_db_url}", err=True)
# Gotcha: running from IDE looks like pipe
is_pipe = not isatty(sys.stdin.fileno())
if is_pipe:
ids = sys.stdin.readline()
try:
ids = [int(x.strip()) for x in ids.split(",")] # type: ignore
except ValueError:
typer.secho(f"-E- Wrong input format.", fg=typer.colors.RED, err=True)
raise typer.Abort()
print(ids)
with DAL(env_config=config) as dal:
for id_ in ids:
bm = dal.get_bookmark_by_id(id_=id_)
show_bms((bm, ))
# TODO: generalize opening URI
if bm.URL.startswith("http"):
webbrowser.open(bm.URL, new=2)
else:
open_it(bm.URL)
def __init__(self, fts_query: str):
self.fts_query = fts_query
with DAL(env_config=config) as dal:
self.bms = dal.get_bookmarks(fts_query=fts_query)
_ = None
def check_tags(tags: Sequence[str]) -> Sequence[str]:
with DAL(env_config=config) as dal:
all_tags = set([r[0] for r in dal.get_all_tags()])
return sorted((set(tags) - all_tags))
desc="")
if result is not None:
url, title, tags_in, desc_in = result
else:
raise typer.Abort()
id_ = BukuDb(dbfile=config.dbfile).add_rec(
url=url,
title_in=title,
tags_in=tags_in,
desc=desc,
immutable=0,
delay_commit=False,
fetch=(not nofetch),
)
with DAL(env_config=config) as dal:
bm = dal.get_bookmark_by_id(id_=id_)
show_bms((bm, ))
@app.command()
def show(
# ctx: typer.Context,
id_: int = typer.Argument(..., help="id to print"),
verbose: bool = typer.Option(False, "-v", "--verbose"),
):
if verbose:
typer.echo(f"Using DB: {config.twbm_db_url}", err=True)
# _ = BukuDb(dbfile=config.dbfile).print_rec(index=id_) # noqa E800
with DAL(env_config=config) as dal: