diff options
| -rwxr-xr-x | scripts/Scripts/goto | 29 |
1 files changed, 22 insertions, 7 deletions
diff --git a/scripts/Scripts/goto b/scripts/Scripts/goto index e4180ad..b0c34fb 100755 --- a/scripts/Scripts/goto +++ b/scripts/Scripts/goto @@ -4,11 +4,14 @@ import argparse import os import sys import os.path as path +from subprocess import call + +# Yeah XDG stuff does not really matter here +defaultstore = path.join(os.getenv("HOME"), ".bookmarks") class Bookmarks(): def __init__(self, storepath=None): - # Yeah XDG stuff does not really matter here - self.storepath = storepath or path.join(os.getenv("HOME"), ".bookmarks") + self.storepath = storepath or defaultstore self.store = {} if not path.exists(self.storepath): @@ -30,13 +33,19 @@ class Bookmarks(): # Truncate it with open(self.storepath, "w") as f: for (k, v) in self.store.items(): - # self.store should contain it's old values to we can jump to something we are saving to - if k is name: + # self.store should contain it's old values to we can jump to + # something we are saving to + if k == name: v = path - f.write(f"{name} {path}\n") + f.write(f"{k} {v}\n") def load(self, name): - return self.store[name] + parts = name.split(":") + name = parts[0] + p = self.store[name] + if len(parts) > 1: + p = path.join(p, parts[1]) + return p def list(self): for (k, v) in self.store.items(): @@ -46,10 +55,16 @@ parser = argparse.ArgumentParser() parser.add_argument("--save", "-s", help="Save current location with name, if '.' pwd is used as name") parser.add_argument("--path", "-p", help="Specifies if `dest` is a real path and not a bookmark", action="store_true") parser.add_argument("--list", "-l", help="List bookmarks", action="store_true") +parser.add_argument("--edit", "-e", help="Edit bookmark file with $EDITRO", action="store_true") parser.add_argument("dest", nargs="?", help="Where to go") args = parser.parse_args() +if args.edit: + editor = os.environ.get("EDITOR", "vi") + ret = call([editor, defaultstore]) + sys.exit(ret) + bm = Bookmarks() if args.list: @@ -57,7 +72,7 @@ if args.list: # Whether to save if args.save: - name = args.save if args.save is not "." else os.getcwd() + name = args.save if args.save != "." else os.getcwd() bm.save(name, os.getcwd()) if args.dest: |