99 lines
2.5 KiB
Python
99 lines
2.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Bangs in DDG style without using DDG but just Rofi.
|
||
|
|
|
||
|
|
Load a JSON config file to have the bangs available. Rofi should return, using
|
||
|
|
its dmenu option, the handle followed by your query, e.g. "w burger" if you
|
||
|
|
want a Wikipedia article for "burger", or at least a much needed
|
||
|
|
disambiguation…
|
||
|
|
|
||
|
|
Config file example:
|
||
|
|
{ "bangs": [ {
|
||
|
|
"handle": "w",
|
||
|
|
"name": "Wikipedia",
|
||
|
|
"url": "https://en.wikipedia.org/wiki/Special:Search?search={}&go=Go"
|
||
|
|
} ] }
|
||
|
|
|
||
|
|
By default this scripts attempts to load your config file from
|
||
|
|
`~/.config/rofibangs.json`, but you can specify the ROFIBANGS_CONFIG_PATH
|
||
|
|
environment variable or pass the path through the -c command-line option.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import urllib.parse
|
||
|
|
import webbrowser
|
||
|
|
|
||
|
|
|
||
|
|
def load_config(config_path=None):
|
||
|
|
if config_path is None:
|
||
|
|
config_path = os.environ.get(
|
||
|
|
"ROFIBANGS_CONFIG_PATH",
|
||
|
|
os.path.expanduser("~/.config/rofibangs.json")
|
||
|
|
)
|
||
|
|
try:
|
||
|
|
with open(config_path, "rt") as bangs_file:
|
||
|
|
return json.load(bangs_file)
|
||
|
|
except OSError:
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def list_bangs(config):
|
||
|
|
for item in config["bangs"]:
|
||
|
|
name = item["name"]
|
||
|
|
handle = item["handle"]
|
||
|
|
print(f"- {handle} {name}")
|
||
|
|
|
||
|
|
|
||
|
|
def open_bang(config, handle, query):
|
||
|
|
for bang in config["bangs"]:
|
||
|
|
if handle == bang["handle"]:
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
print("Unknown handle.")
|
||
|
|
return
|
||
|
|
url = bang["url"].format(urllib.parse.quote(query))
|
||
|
|
webbrowser.open_new_tab(url)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
ap = argparse.ArgumentParser()
|
||
|
|
ap.add_argument(
|
||
|
|
"-c", "--config", help="path to JSON config file"
|
||
|
|
)
|
||
|
|
ap.add_argument(
|
||
|
|
"-l", "--list", action="store_true", help="show available bangs"
|
||
|
|
)
|
||
|
|
args = ap.parse_args()
|
||
|
|
|
||
|
|
config = load_config()
|
||
|
|
if config is None:
|
||
|
|
exit("Can't load config file.")
|
||
|
|
|
||
|
|
if args.list:
|
||
|
|
list_bangs(config)
|
||
|
|
return
|
||
|
|
|
||
|
|
rofi_path = config.get("rofi_path", "rofi")
|
||
|
|
process_input = "\n".join(i["handle"] for i in config["bangs"]) + "\n"
|
||
|
|
completed_process = subprocess.run(
|
||
|
|
[rofi_path, "-dmenu", "-p", "bang"],
|
||
|
|
text=True,
|
||
|
|
capture_output=True,
|
||
|
|
input=process_input
|
||
|
|
)
|
||
|
|
output = completed_process.stdout
|
||
|
|
if not output:
|
||
|
|
exit("Empty Rofi output.")
|
||
|
|
|
||
|
|
parts = output.split(maxsplit=1)
|
||
|
|
if len(parts) != 2:
|
||
|
|
exit("Bad Rofi output.")
|
||
|
|
|
||
|
|
open_bang(config, *parts)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|