Python 实现一个mini shell

mini shell 的想法是通过python实现一个类似与linux的shell。目前只支持4条命令 ls,cat,cd,history。(如果exit也算一条命令的话,那就是五条)。
功能也类似与linux的shell,比如说运行这个python的程序之后,输入ls,它会返回当前目录的文件和文件夹。

通过这个小demo:

  1. 可以熟悉一个python下面os库的一些函数,比如说如何获取当前工作目录,主机名,用户名等等。
  2. 熟悉signal信号机制,比如我们在linux下面输入ctrl+c 是挂起信号,但是你在mini shell里面要屏蔽这个信号,因为并不希望ctrl+c就退出当前的minishell。
  3. 了解一下shell,在学习linux的时候,其实对shell刚接触的时候,会有很大的恐惧感。其实我们在shell下输入的命令,比如cat,ls 他们的本质都是可执行文件,位于系统的bin目录下面。
    输入ls -l,就好像我们现在有一个可执行文件叫 a.out,
//-l 相当于是参数,通过mian的argv获得
./a.out   -l  

ls和自己写得a.out 的区别在于,a.out必须在当前目录下才可以执行,但是ls可以在任何目录。你可以更改环境变量,把a.out的当前目录,加入到环境变量里面,这样就可以了。(这也就是我们安装java,python,mysql 等为什么要配置环境变量的原因了)

前面可以参考:
Python 在linux下获得当前工作目录,主机名,用户名,操作系统平台等信息http://www.jianshu.com/p/5c294b5bc7b9

运行结果:


image.png

这是一个未完工的版本:
history的命令,exit命令还没有实现,也没有写成一个循环。有空来更新。实验是参考实验楼的一个python教程。

解释器的版本是python2.7,换成3也没有问题,稍微改一下print函数就可以。

#coding:utf-8
import os
import sys
import shlex
import getpass
import socket
import signal
import subprocess
import platform
import re


SHELL_STATUS_STOP = 0
SHELL_STATUS_RUN = 1
# 使用 os.path.expanduser('~') 获取当前操作系统平台的当前用户根目录
HISTORY_PATH = os.path.expanduser('~') + os.sep + '.shiyanlou_shell_history'

def ls(args):
    if len(args)>0:
        print os.getcwd()

def cd(args):
    if len(args) > 0:
        os.chdir(args[0])
    else:
        os.chdir(os.getenv('HOME'))
    print os.getcwd()
    return SHELL_STATUS_RUN

def getenv(args):
    #print 'test : ',args
    if len(args) >0:
        print(os.getenv(args))

def exit():
    #TO-DO
    print ('exit')

def history():
    #TO-DO
    print ('history')

def cat(arg):
    #TO-DO
    print ('cat')
    

def display_cmd_prompt():
    user=getpass.getuser()
    hostname=socket.gethostname()
    cwd=os.getcwd()
    base_dir=os.path.basename(cwd)

    home_dir=os.path.expanduser('~')
    if cwd==home_dir:
        base_dir='~'
    #sys.stdout.write("[%s@%s %s]$ " % (user, hostname, base_dir))
    #改變顏色
    sys.stdout.write("[\033[1;33m%s\033[0;0m@%s \033[1;36m%s\033[0;0m] $ " % (user, hostname, base_dir))
    sys.stdout.flush()



#這部分使用啦正則表達式來處理
def getsplit(str):
    # 其实就是按空格符将命令与参数分开
    # 比如,'ls -l /home/shiyanlou' 划分之后就是
    # ['ls', '-l', '/home/shiyanlou']
    #res=re.split(r'\s',str)
    res=re.split(r'\s+',str)
    return res



def isgetenv(tokens):
    token=tokens[0]
    #print token
    if token.startswith('$'):
        getenv(token[1:])
        #print os.getenv(token[1:])
        return True
    else:
        return False

#構建一個字典 字典裏面的value,指的是函數,比如cd是cd函數
built_in_cmds = {}
built_in_cmds['cd']=cd
built_in_cmds['ls']=ls
built_in_cmds['cat']=cat
built_in_cmds['exit']=exit
built_in_cmds['history']=history


#to-do 這裏應該寫成一個循環 

#輸出
display_cmd_prompt()
#等待輸入
cmd=sys.stdin.readline()
#切分
cmd_tokens=getsplit(cmd)
#cmd_tokens=[ls -l]
#cmd_tokens=preprocess(cmd_tokens)
if isgetenv(cmd_tokens)==False:
    #print cmd_tokens[0]
    #print cmd_tokens[1]
    cmd_args=cmd_tokens[1:]
    #print cmd_args
    if cmd_tokens[0] in built_in_cmds:
        #print cmd_args
        built_in_cmds[cmd_tokens[0]](cmd_args)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,133评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,682评论 3 390
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,784评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,508评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,603评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,607评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,604评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,359评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,805评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,121评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,280评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,959评论 5 339
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,588评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,206评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,442评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,193评论 2 367
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,144评论 2 352

推荐阅读更多精彩内容

  • Ubuntu的发音 Ubuntu,源于非洲祖鲁人和科萨人的语言,发作 oo-boon-too 的音。了解发音是有意...
    萤火虫de梦阅读 99,242评论 9 467
  • linux资料总章2.1 1.0写的不好抱歉 但是2.0已经改了很多 但是错误还是无法避免 以后资料会慢慢更新 大...
    数据革命阅读 12,157评论 2 33
  • (一)基本命令 命令格式: 命令 参数 1.ls 显示文件名,等同于dos下dir命令 命令格式:ls [opti...
    飞羽孟德阅读 1,883评论 0 3
  • 我们说:读书学习要会跨界,要学各个基础学科的自然原理和规律,这是我们的底层逻辑规律; 但是在现实世界中,我们要专精...
    小花的蓝田阅读 516评论 0 0
  • 这次旅行定的匆忙,只是一个念头闪过,便再挥之不去,那就一个人去看看吧。 出行前一天看攻略,定青旅,心里大概有了谱,...
    木南妮儿阅读 188评论 4 3