twd-m4sc0 2.0.3__py3-none-any.whl → 3.0.1__py3-none-any.whl

twd/crud.py

@@ -1,104 +0,0 @@
-import os
-import json
-import hashlib
-import time
-import logging
-from collections import OrderedDict
-
-log = logging.getLogger("log")
-error_log = logging.getLogger("error")
-
-
-def create_alias_id():
-    data = str(time.time()) + str(os.urandom(16))
-    return hashlib.sha256(data.encode()).hexdigest()[:12]
-
-
-def get_data_file(config):
-    return os.path.expanduser(config.get("data_file", "~/.twd/data"))
-
-
-def ensure_data_file_exists(config):
-    data_file = get_data_file(config)
-    if not os.path.exists(data_file):
-        try:
-            with open(data_file, "w") as f:
-                json.dump({}, f)
-            log.info(f"Created data file at {data_file}")
-        except OSError as e:
-            error_log.error(f"Error creating data file: {e}")
-
-
-def load_data(config):
-    data_file = get_data_file(config)
-    if not os.path.exists(data_file):
-        ensure_data_file_exists(config)
-    try:
-        with open(data_file, "r") as f:
-            data = json.load(f)
-            log.info(f"Loaded data from {data_file}")
-            return data
-    except json.JSONDecodeError as e:
-        error_log.error(f"Error reading data file: {e}")
-        return {}
-    except OSError as e:
-        error_log.error(f"Error reading data file: {e}")
-        return {}
-
-
-def save_data(config, data):
-    data_file = get_data_file(config)
-    try:
-        sorted_data = OrderedDict(
-            sorted(data.items(), key=lambda item: item[1]["alias"])
-        )
-        with open(data_file, "w") as f:
-            json.dump(sorted_data, f, indent=4)
-        log.info(f"Saved data to {data_file}")
-    except OSError as e:
-        error_log.error(f"Error writing to data file: {e}")
-
-
-def create_entry(config, data, path, alias=None):
-    alias_id = create_alias_id()
-    data[alias_id] = {
-        "path": path,
-        "alias": alias if alias else "no_alias",
-        "created_at": time.time(),
-    }
-    save_data(config, data)
-    log.info(f"Created new entry with alias_id '{alias_id}' and path '{path}'")
-    return alias_id
-
-
-def delete_entry(config, data, entry_id):
-    if entry_id in data:
-        del data[entry_id]
-        save_data(config, data)
-        log.info(f"Deleted entry with alias_id '{entry_id}'")
-    else:
-        error_log.error(f"Entry ID '{entry_id}' not found")
-        raise KeyError(f"Entry ID {entry_id} not found")
-
-
-def update_entry(config, data, entry_id, entry):
-    if entry_id in data:
-        data[entry_id] = entry
-        save_data(config, data)
-        log.info(f"Updated entry with alias_id '{entry_id}'")
-    else:
-        error_log.error(f"Entry ID '{entry_id}' not found")
-        raise KeyError(f"Entry ID {entry_id} not found")
-
-
-def delete_data_file(config):
-    data_file = get_data_file(config)
-    if os.path.exists(data_file):
-        try:
-            os.remove(data_file)
-            log.info(f"Deleted data file at {data_file}")
-        except OSError as e:
-            error_log.error(f"Error deleting data file: {e}")
-            raise
-    else:
-        error_log.error("No data file found to delete")

twd/logger.py

@@ -1,47 +0,0 @@
-import logging
-import os
-from logging.handlers import RotatingFileHandler
-
-
-def setup_logger(config):
-    """Set up loggers based on the configuration provided."""
-    log_file = os.path.expanduser(config.get("log_file"))
-    error_file = os.path.expanduser(config.get("error_file"))
-    log_level = config.get("log_level", "INFO").upper()
-    max_bytes = config.get("log_max_bytes", 5 * 1024 * 1024)  # Default 5MB
-    backup_count = config.get("log_backup_count", 3)  # Default 3 backup files
-
-    os.makedirs(os.path.dirname(log_file), exist_ok=True)
-    os.makedirs(os.path.dirname(error_file), exist_ok=True)
-
-    # General log configuration
-    log_formatter = logging.Formatter(
-        config.get("log_format", "%(asctime)s - %(levelname)s - %(message)s")
-    )
-
-    # Avoid duplicate handlers
-    logger = logging.getLogger("log")
-    if not logger.hasHandlers():
-        logger.setLevel(log_level)
-        log_handler = RotatingFileHandler(
-            log_file, maxBytes=max_bytes, backupCount=backup_count
-        )
-        log_handler.setFormatter(log_formatter)
-        logger.addHandler(log_handler)
-
-    # Error log configuration
-    error_logger = logging.getLogger("error")
-    if not error_logger.hasHandlers():
-        error_logger.setLevel(logging.ERROR)
-        error_handler = RotatingFileHandler(
-            error_file, maxBytes=max_bytes, backupCount=backup_count
-        )
-        error_handler.setFormatter(log_formatter)
-        error_logger.addHandler(error_handler)
-
-    return logger, error_logger
-
-
-def initialize_logging(config):
-    setup_logger(config)
-

twd/screen.py

@@ -1,201 +0,0 @@
-import curses
-import time
-import os
-from . import crud
-import logging
-
-log = logging.getLogger("log")
-error_log = logging.getLogger("error")
-
-CONFIG = None
-DIRS = None
-filtered_DIRS = None
-search_query = ""
-original_DIRS = None
-
-
-def draw_hr(stdscr, y, mode=None):
-    _, max_cols = stdscr.getmaxyx()
-    mode = mode if mode is not None else curses.A_NORMAL
-    stdscr.addstr(y, 1, "─" * (max_cols - 2), mode)
-
-
-def filter_dirs_by_search(query):
-    global filtered_DIRS
-    filtered_DIRS = (
-        {k: v for k, v in DIRS.items() if query.lower() in v["alias"].lower()}
-        if query
-        else DIRS
-    )
-
-
-def display_select_screen(stdscr):
-    global search_query, filtered_DIRS, original_DIRS
-    selected_entry = 0
-    pre_selected_path = None
-    confirm_mode = False
-    action = None
-    search_mode = False
-    post_search_mode = False
-
-    running = True
-
-    while running:
-        max_items = len(filtered_DIRS)
-        stdscr.clear()
-
-        # Border setup
-        height, width = stdscr.getmaxyx()
-        stdscr.addstr(0, 0, "╭")
-        stdscr.addstr(0, 1, "─" * (width - 2))
-        stdscr.addstr(0, width - 1, "╮")
-        stdscr.addstr(height - 1, 0, "╰")
-        stdscr.addstr(height - 1, 1, "─" * (width - 2))
-        stdscr.addstr(height - 2, width - 1, "╯")
-        for i in range(1, height - 1):
-            stdscr.addstr(i, 0, "│")
-            stdscr.addstr(i, width - 1, "│")
-
-        inner_height = height - 2
-        inner_width = width - 2
-        stdscr.addstr(1, 1, f"Current directory: {os.getcwd()}")
-
-        draw_hr(stdscr, 2)
-
-        # Header
-        max_alias_len = max(
-            max(len(entry["alias"]) for entry in filtered_DIRS.values()), 5
-        )
-        max_path_len = max(
-            max(len(entry["path"]) for entry in filtered_DIRS.values()), 4
-        )
-        max_id_len = max(max(len(alias_id) for alias_id in filtered_DIRS.keys()), 2)
-
-        alias_col = max_alias_len + 2
-        id_col = max_id_len + 2
-        path_col = max_path_len
-
-        header_text = f"{'ALIAS'.ljust(alias_col)}{'ID'.ljust(id_col)}{'PATH'.ljust(path_col)}  CREATED AT"
-        stdscr.addstr(3, 1, header_text[:inner_width])
-
-        draw_hr(stdscr, 4)
-
-        # List entries
-        line_start = 5
-        for entry_id, entry in enumerate(filtered_DIRS.values()):
-            if line_start >= inner_height - 5:
-                break
-            alias = entry["alias"].ljust(max_alias_len)
-            path = entry["path"].ljust(max_path_len)
-            alias_id = list(filtered_DIRS.keys())[entry_id].ljust(max_id_len)
-            created_at = time.strftime(
-                "%Y-%m-%d %H:%M:%S", time.localtime(entry["created_at"])
-            )
-
-            line_text = f"{alias}  {alias_id}  {path}  {created_at}"
-            if entry_id == selected_entry:
-                stdscr.addstr(line_start, 1, line_text[:inner_width], curses.A_REVERSE)
-                pre_selected_path = entry["path"]
-            else:
-                stdscr.addstr(line_start, 1, line_text[:inner_width])
-
-            line_start += 1
-
-        # Controls
-        controls_y = height - 5
-        draw_hr(stdscr, controls_y, curses.A_DIM)
-        controls_text = (
-            "ctrls: enter=select"
-            if search_mode
-            else "ctrls: ↑/k=up  ↓/j=down  enter=select  d/backspace=delete  q=exit search  s=search"
-            if post_search_mode
-            else "ctrls: ↑/k=up  ↓/j=down  enter=select  d/backspace=delete  q=quit  s=search"
-        )
-        stdscr.addstr(controls_y + 1, 1, controls_text, curses.A_DIM)
-
-        # Action area
-        action_area_y = height - 3
-        draw_hr(stdscr, action_area_y)
-
-        if search_mode:
-            stdscr.addstr(action_area_y + 1, 1, f"Search: {search_query}")
-        elif confirm_mode and action == "delete":
-            entry = filtered_DIRS[list(filtered_DIRS.keys())[selected_entry]]
-            stdscr.addstr(
-                action_area_y + 1,
-                1,
-                f"Delete entry '{entry['alias']}' ({entry['path']})? [enter/q]",
-            )
-        elif pre_selected_path:
-            stdscr.addstr(
-                action_area_y + 1,
-                1,
-                f"Command: cd {os.path.abspath(os.path.expanduser(pre_selected_path))}",
-            )
-
-        stdscr.refresh()
-
-        # Handle key events
-        key = stdscr.getch()
-
-        if search_mode:
-            if key == ord("\n"):
-                search_mode = False
-                post_search_mode = True
-            elif key == curses.KEY_BACKSPACE or key == 127:
-                search_query = search_query[:-1]
-                filter_dirs_by_search(search_query)
-            else:
-                search_query += chr(key)
-                filter_dirs_by_search(search_query)
-        elif post_search_mode:
-            if key == ord("q") or key == 27:  # 'q' or 'esc'
-                filtered_DIRS = original_DIRS
-                post_search_mode = False
-            elif key == curses.KEY_UP or key == ord("k"):
-                selected_entry = max(0, selected_entry - 1)
-            elif key == curses.KEY_DOWN or key == ord("j"):
-                selected_entry = min(max_items - 1, selected_entry + 1)
-            elif key == ord("\n"):
-                selected_entry_id = list(filtered_DIRS.keys())[selected_entry]
-                return filtered_DIRS[selected_entry_id]
-        elif confirm_mode:
-            if key == ord("\n") and action == "delete":
-                selected_entry_id = list(filtered_DIRS.keys())[selected_entry]
-                data = crud.load_data(CONFIG)
-                try:
-                    crud.delete_entry(CONFIG, data, selected_entry_id)
-                except KeyError:
-                    error_log.error(f"Entry ID {selected_entry_id} not found")
-                del filtered_DIRS[selected_entry_id]
-                if selected_entry >= len(filtered_DIRS):
-                    selected_entry = max(len(filtered_DIRS) - 1, 0)
-                confirm_mode = False
-            else:
-                confirm_mode = False
-        else:
-            if key == curses.KEY_UP or key == ord("k"):
-                selected_entry = (selected_entry - 1) % max_items
-            elif key == curses.KEY_DOWN or key == ord("j"):
-                selected_entry = (selected_entry + 1) % max_items
-            elif key == ord("\n"):
-                selected_entry_id = list(filtered_DIRS.keys())[selected_entry]
-                return filtered_DIRS[selected_entry_id]
-            elif key == ord("q"):
-                return None
-            elif key == ord("d") or key == curses.KEY_BACKSPACE:
-                confirm_mode = True
-                action = "delete"
-            elif key == ord("s"):
-                search_mode = True
-                selected_entry = 0
-
-
-def display_select(config, dirs):
-    global CONFIG, DIRS, filtered_DIRS, search_query, original_DIRS
-    CONFIG = config
-    DIRS = dirs
-    filtered_DIRS = DIRS
-    original_DIRS = DIRS
-    search_query = ""
-    return curses.wrapper(display_select_screen)