53 lines
2.1 KiB
Plaintext
53 lines
2.1 KiB
Plaintext
"""Integrates Johnny Decimal Notation into Xonsh"""
|
|
from pathlib import Path
|
|
|
|
from xonsh.built_ins import XSH
|
|
|
|
HOME = Path("/run/media/riley/drive")
|
|
|
|
def find_category(number: int, path, regex: str):
|
|
"""Find matching JDN folder when given notation"""
|
|
import re
|
|
folders = path.iterdir()
|
|
for folder in folders:
|
|
if not folder.is_dir():
|
|
continue
|
|
folder_category = re.match(regex, folder.name)
|
|
if folder_category is None:
|
|
continue
|
|
if len(folder_category.groups()) == 2:
|
|
if number >= int(folder_category.group(1)) and number <= int(folder_category.group(2)):
|
|
return folder
|
|
elif len(folder_category.groups()) == 1:
|
|
if number == int(folder_category.group((1))):
|
|
return folder
|
|
return path
|
|
|
|
@XSH.builtins.events.on_transform_command
|
|
def jds_preproc(cmd: str, **kw):
|
|
"""processes command strings starting with "`" to cd to matching folder."""
|
|
if cmd.startswith("`") and "." in cmd and not cmd.strip().endswith("`"):
|
|
try:
|
|
johnny_decimal_notation = cmd.replace("`", "")
|
|
bucket = find_category(int(johnny_decimal_notation.split(".")[0]), HOME, r"(^\d0)-(\d9)")
|
|
category = find_category(int(johnny_decimal_notation.split(".")[0]), bucket, r"^(\d\d)")
|
|
folder = find_category(int(johnny_decimal_notation.split(".")[1]), category, r"^(\d\d)")
|
|
cmd = f"cd '{folder.resolve()}'"
|
|
except FileNotFoundError:
|
|
cmd = f'print("\'{HOME}\' not found.")'
|
|
except ValueError:
|
|
cmd = f'print("\'{cmd.strip().replace("`", "")}\' is not a valid JDN.")'
|
|
elif cmd.startswith("`") and not (len(cmd.strip()) >= 2 and cmd.strip().endswith("`")):
|
|
try:
|
|
johnny_decimal_notation = cmd.replace("`", "")
|
|
if johnny_decimal_notation == "\n":
|
|
return f"cd '{HOME.resolve()}'"
|
|
bucket = find_category(int(johnny_decimal_notation.split(".")[0]), HOME, r"(^\d0)-(\d9)")
|
|
category = find_category(int(johnny_decimal_notation.split(".")[0]), bucket, r"^(\d\d)")
|
|
cmd = f"cd '{category.resolve()}'"
|
|
except FileNotFoundError:
|
|
cmd = f'print("\'{HOME}\' not found.")'
|
|
except ValueError:
|
|
cmd = f'print("\'{cmd.strip().replace("`", "")}\' is not a valid JDN.")'
|
|
return cmd
|