使用 Python 创建你自己的 Shell(下)

上篇中,我们已经创建了一个 shell 主循环、切分了命令输入,以及通过 fork 和 exec 执行命令。在这部分,我们将会解决剩下的问题。首先,cd test_dir2 命令无法修改我们的当前目录。其次,我们仍无法优雅地从 shell 中退出。

步骤 4:内置命令

cd test_dir2 无法修改我们的当前目录” 这句话是对的,但在某种意义上也是错的。在执行完该命令之后,我们仍然处在同一目录,从这个意义上讲,它是对的。然而,目录实际上已经被修改,只不过它是在子进程中被修改。

还记得我们分叉(fork)了一个子进程,然后执行命令,执行命令的过程没有发生在父进程上。结果是我们只是改变了子进程的当前目录,而不是父进程的目录。

然后子进程退出,而父进程在原封不动的目录下继续运行。

因此,这类与 shell 自己相关的命令必须是内置命令。它必须在 shell 进程中执行而不是在分叉中(forking)。

cd

让我们从 cd 命令开始。

我们首先创建一个 builtins 目录。每一个内置命令都会被放进这个目录中。


  1. yosh_project
  2. |-- yosh
  3. |-- builtins
  4. | |-- __init__.py
  5. | |-- cd.py
  6. |-- __init__.py
  7. |-- shell.py

在 cd.py 中,我们通过使用系统调用 os.chdir 实现自己的 cd 命令。


  1. import os
  2. from yosh.constants import *
  3. def cd(args):
  4. os.chdir(args[0])
  5. return SHELL_STATUS_RUN

注意,我们会从内置函数返回 shell 的运行状态。所以,为了能够在项目中继续使用常量,我们将它们移至yosh/constants.py


  1. yosh_project
  2. |-- yosh
  3. |-- builtins
  4. | |-- __init__.py
  5. | |-- cd.py
  6. |-- __init__.py
  7. |-- constants.py
  8. |-- shell.py

在 constants.py 中,我们将状态常量都放在这里。


  1. SHELL_STATUS_STOP = 0
  2. SHELL_STATUS_RUN = 1

现在,我们的内置 cd 已经准备好了。让我们修改 shell.py 来处理这些内置函数。


  1. ...
  2. ### 导入常量
  3. from yosh.constants import *
  4. ### 使用哈希映射来存储内建的函数名及其引用
  5. built_in_cmds = {}
  6. def tokenize(string):
  7. return shlex.split(string)
  8. def execute(cmd_tokens):
  9. ### 从元组中分拆命令名称与参数
  10. cmd_name = cmd_tokens[0]
  11. cmd_args = cmd_tokens[1:]
  12. ### 如果该命令是一个内建命令,使用参数调用该函数
  13. if cmd_name in built_in_cmds:
  14. return built_in_cmds[cmd_name](cmd_args)
  15. ...

我们使用一个 python 字典变量 built_in_cmds 作为哈希映射hash map,以存储我们的内置函数。我们在execute 函数中提取命令的名字和参数。如果该命令在我们的哈希映射中,则调用对应的内置函数。

(提示:built_in_cmds[cmd_name] 返回能直接使用参数调用的函数引用。)

我们差不多准备好使用内置的 cd 函数了。最后一步是将 cd 函数添加到 built_in_cmds 映射中。


  1. ...
  2. ### 导入所有内建函数引用
  3. from yosh.builtins import *
  4. ...
  5. ### 注册内建函数到内建命令的哈希映射中
  6. def register_command(name, func):
  7. built_in_cmds[name] = func
  8. ### 在此注册所有的内建命令
  9. def init():
  10. register_command("cd", cd)
  11. def main():
  12. ###在开始主循环之前初始化 shell
  13. init()
  14. shell_loop()

我们定义了 register_command 函数,以添加一个内置函数到我们内置的命令哈希映射。接着,我们定义init 函数并且在这里注册内置的 cd 函数。

注意这行 register_command("cd", cd) 。第一个参数为命令的名字。第二个参数为一个函数引用。为了能够让第二个参数 cd 引用到 yosh/builtins/cd.py 中的 cd 函数引用,我们必须将以下这行代码放在yosh/builtins/__init__.py 文件中。


  1. from yosh.builtins.cd import *

因此,在 yosh/shell.py 中,当我们从 yosh.builtins 导入 * 时,我们可以得到已经通过yosh.builtins 导入的 cd 函数引用。

我们已经准备好了代码。让我们尝试在 yosh 同级目录下以模块形式运行我们的 shell,python -m yosh.shell

现在,cd 命令可以正确修改我们的 shell 目录了,同时非内置命令仍然可以工作。非常好!

exit

最后一块终于来了:优雅地退出。

我们需要一个可以修改 shell 状态为 SHELL_STATUS_STOP 的函数。这样,shell 循环可以自然地结束,shell 将到达终点而退出。

和 cd 一样,如果我们在子进程中分叉并执行 exit 函数,其对父进程是不起作用的。因此,exit 函数需要成为一个 shell 内置函数。

让我们从这开始:在 builtins 目录下创建一个名为 exit.py 的新文件。


  1. yosh_project
  2. |-- yosh
  3. |-- builtins
  4. | |-- __init__.py
  5. | |-- cd.py
  6. | |-- exit.py
  7. |-- __init__.py
  8. |-- constants.py
  9. |-- shell.py

exit.py 定义了一个 exit 函数,该函数仅仅返回一个可以退出主循环的状态。


  1. from yosh.constants import *
  2. def exit(args):
  3. return SHELL_STATUS_STOP

然后,我们导入位于 yosh/builtins/__init__.py 文件的 exit 函数引用。


  1. from yosh.builtins.cd import *
  2. from yosh.builtins.exit import *

最后,我们在 shell.py 中的 init() 函数注册 exit 命令。


  1. ...
  2. ### 在此注册所有的内建命令
  3. def init():
  4. register_command("cd", cd)
  5. register_command("exit", exit)
  6. ...

到此为止!

尝试执行 python -m yosh.shell。现在你可以输入 exit 优雅地退出程序了。

最后的想法

我希望你能像我一样享受创建 yosh (your own shell)的过程。但我的 yosh 版本仍处于早期阶段。我没有处理一些会使 shell 崩溃的极端状况。还有很多我没有覆盖的内置命令。为了提高性能,一些非内置命令也可以实现为内置命令(避免新进程创建时间)。同时,大量的功能还没有实现(请看 公共特性 和 不同特性)。

原文发布时间为:2016-07-29

本文来自合作伙伴“Linux中国”

时间: 2017-05-02

使用 Python 创建你自己的 Shell(下)的相关文章

使用 Python 创建你自己的 Shell (上)

我很想知道一个 shell (像 bash,csh 等)内部是如何工作的.于是为了满足自己的好奇心,我使用 Python 实现了一个名为 yosh (Your Own Shell)的 Shell.本文章所介绍的概念也可以应用于其他编程语言. (提示:你可以在这里查找本博文使用的源代码,代码以 MIT 许可证发布.在 Mac OS X 10.11.5 上,我使用 Python 2.7.10 和 3.4.3 进行了测试.它应该可以运行在其他类 Unix 环境,比如 Linux 和 Windows 上

在Linux中灵活使用shell下的特殊字符

众所周知,shell是Linux操作系统下的主要管理工具.系统管理员除了要掌握shell下的命令外,还需要掌握一些特殊的字符.这些特殊字符可以帮助系统管理员将多个命令组合成一个命令,或者说可以让系统管理员更便捷的操作.笔者就给大家介绍一下这些特殊字符以及相关的应用. 一.快速在目录之间切换. 在Linux操作系统的文本模式下,在目录之间进行切换都需要通过命令来完成.显然这没有图形化界面下使用鼠标点点那么方便.但是,shell中仍然提供了一些特殊的字符.在这些字符的帮助下,系统管理员可以轻松的在不

python创建进程fork用法

  本文实例讲述了python创建进程fork用法.分享给大家供大家参考.具体分析如下: ? 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 #!coding=utf-8 import os ,traceback import time ''' fork()系统调用是Unix下以自身进程创建子进程的系统调用, 一次调用,两次返回,如果返回是0, 则是子进程,如果返回值>0,则是父进程(返回

python创建一个最简单http webserver服务器的方法

  这篇文章主要介绍了python创建一个最简单http webserver服务器的方法,实例分析了Python操作http创建服务器端的相关技巧,需要的朋友可以参考下 ? 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 import sys import BaseHTTPServer from SimpleHTTPServer import SimpleHTTPRequestHandler Handler = SimpleHTTPRequestHandler Serve

在Linux Shell下将汉字转换为Url编码

在linux 的shell环境下有一个od命令可以将文件按不同的进制的ascii码输 出,该功能可以完成在linux shell下将汉字转换为url编码的任务. url编码是十六进制的形如%E6%9C%8D%E5%8A%A1%E5的字符串,test是包含 "数据库网络,"这几个字符的文件,以下命令输出: #od -t x /test 0000000 ddbefdca f8cde2bf 0a2ce7c2 0000014 该字符串对应的url编码为 %CA%FD%BE%DD%BF%E2%C

Python创建模块及模块导入的方法

  本文实例讲述了Python创建模块及模块导入的方法.分享给大家供大家参考.具体分析如下: python学习手册中写道: 定义模块,只要使用文本编辑器,把一些python代码输入到文本中,然后以.py为后缀名进行保存,任何此类文件都会被认为是python模块. 比如说,下面的代码输入到一个文件中,就可以看作是一个模块: ? 1 2 3 4 def printme(var): print var if __name__ == '__main__': printme(1) 假设说输入到a.py中,

python使用wmi模块获取windows下硬盘信息的方法

本文实例讲述了python使用wmi模块获取windows下硬盘信息的方法.分享给大家供大家参考.具体实现方法如下: ? 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45# -*- coding: utf-8 -*- #import ################################

Python创建系统目录的方法_python

本文实例讲述了Python创建系统目录的方法.分享给大家供大家参考.具体如下: Python2 mkdir在没有上级目录时创建会失败.该方法可以创建多级目录. /temp/gapgers/upload/images/1.png 如过temp文件夹不存在,会创建空的文件夹/temp/gapgers/upload/images/以及空文件1.png. 该方法只做抛砖引玉,大神勿喷 复制代码 代码如下: import os     def mkfilePower(path):       '''cre

Python创建xml的方法_python

本文实例讲述了Python创建xml的方法.分享给大家供大家参考.具体实现方法如下: from xml.dom.minidom import Document class write_xml(Document): def __init__(self): Document.__init__(self) def set_tag(self,tag): self.tag = tag self.tag1 = self.createElement(self.tag) self.appendChild(sel

python学习-python创建sql server 数据库问题求助

问题描述 python创建sql server 数据库问题求助 用python来创建一个数据库,如果存在就不用创建,如果不存在就创建一个: cursor.execute('if not exists(select * from master..sysdatabases where name = python) create database python;') 这样写编译不过,求助 解决方案 先用sql语句判断一下表格,不在,再调用创建