如何在Python 2.7的.py脚本中使用书签函数？

首先得理清楚你原来代码里的核心问题：Python脚本没法直接给你的shell设置别名。因为Python脚本是在子进程里运行的，子进程的任何环境变更（包括别名）都不会影响到启动它的父shell（也就是你平时用的命令行）。而且你代码里的subprocess.call([bookmark, response2])里的bookmark根本没定义，这肯定会报错。

那怎么实现目录书签的功能呢？得换个思路：用Python脚本管理书签的存储，然后让shell自己读取这些书签来切换目录。下面给你一套可行的方案：

第一步：修改Python脚本，把书签存到配置文件里 #

我们把书签信息存在用户主目录下的.dir_bookmarks文件里，这样shell也能方便读取。修改后的代码如下：

#!/usr/bin/env python
import os

# 书签存储文件的路径，放在用户主目录下
BOOKMARK_FILE = os.path.expanduser("~/.dir_bookmarks")

def bookmark_func():
    # 获取要添加书签的目录
    dir_input = raw_input("Please enter the directory name: ").strip()
    
    # 先检查目录是否存在，避免出错
    if not os.path.isdir(dir_input):
        print("Oops! The directory '{}' doesn't exist.".format(dir_input))
        return
    
    # 转成绝对路径，避免相对路径的问题
    target_dir = os.path.abspath(dir_input)
    print("Got it, target directory is: {}".format(target_dir))
    
    # 获取书签名称
    bookmark_name = raw_input("Please enter a bookmark name: ").strip()
    if not bookmark_name:
        print("Bookmark name can't be empty!")
        return
    
    # 读取现有的书签，防止重复添加
    bookmarks = {}
    if os.path.exists(BOOKMARK_FILE):
        with open(BOOKMARK_FILE, 'r') as f:
            for line in f:
                line = line.strip()
                # 跳过注释和空行
                if not line or line.startswith('#'):
                    continue
                # 分割书签名称和路径
                name, path = line.split('=', 1)
                bookmarks[name.strip()] = path.strip()
    
    # 添加或更新书签
    bookmarks[bookmark_name] = target_dir
    
    # 把书签写回文件
    with open(BOOKMARK_FILE, 'w') as f:
        for name, path in bookmarks.items():
            f.write("{}={}\n".format(name, path))
    
    print("Success! Bookmark '{}' added for directory: {}".format(bookmark_name, target_dir))

if __name__ == "__main__":
    bookmark_func()

第二步：在shell里添加切换目录的函数 #

因为Python没法直接控制父shell，所以我们要在你的shell配置文件（比如~/.bashrc或者~/.zshrc，取决于你用的shell）里加一个函数，用来读取书签文件并切换目录。

打开你的shell配置文件，添加这段代码：

# 目录书签切换函数，命名为go
go() {
    local BOOKMARK_FILE="$HOME/.dir_bookmarks"
    # 检查书签文件是否存在
    if [ ! -f "$BOOKMARK_FILE" ]; then
        echo "No bookmarks found yet. Run your Python script to add some first!"
        return 1
    fi
    # 如果没传书签名称，就列出所有书签
    if [ -z "$1" ]; then
        echo "Available bookmarks:"
        cat "$BOOKMARK_FILE"
        return 0
    fi
    # 根据书签名称查找对应的目录
    local target_dir=$(grep "^$1=" "$BOOKMARK_FILE" | cut -d'=' -f2-)
    if [ -z "$target_dir" ]; then
        echo "Bookmark '$1' doesn't exist."
        return 1
    fi
    # 切换目录，如果失败就提示
    cd "$target_dir" || echo "Failed to enter directory: $target_dir"
}

第三步：使用方法 #

1. 先运行你的Python脚本添加书签：

   python bookmark.py
   

   按照提示输入目录（可以是相对路径或绝对路径）和书签名称，比如输入/home/atul/projects，书签名称设为proj。

2. 让shell加载新的配置（或者重新打开一个终端）：

   source ~/.bashrc  # 如果是bash用户
   # 或者
   source ~/.zshrc   # 如果是zsh用户
   

3. 之后就可以用go命令切换目录了：

   • 输入go proj就能直接跳到/home/atul/projects
   • 输入go就能列出所有已添加的书签

为什么原来的方法行不通？ #

再补个小知识点：当你运行Python脚本时，系统会启动一个新的子进程来执行它。这个子进程有自己独立的环境，包括环境变量、shell别名等等。你在子进程里做的任何变更，都不会影响到启动它的父shell（也就是你敲命令的那个终端）。所以直接用subprocess调用alias是没用的，子进程结束后，父shell里根本看不到这个别名。

内容的提问来源于stack exchange，提问作者Atul