python交互式shell_用python -i写交互式shell

cabinet是公司的一个数据存储服务，需要添加一个shell client，查看数据，做简单操作。

用python写了一个比想象的简单。代码如下:

#! /usr/bin/python -i

# coding: utf-8

import sys

import os

dirname = os.path.dirname(sys.argv[0])

sys.path.append(dirname + '/gen-py')

sys.path.append('/usr/lib/python2.7/site-packages')

from cabinet import CabinetStorage

from document.ttypes import Document

from thrift import Thrift

from thrift.transport import TSocket

from thrift.transport import TTransport

from thrift.protocol import TCompactProtocol

from optparse import OptionParser

# avoid UnicodeDecodeError

reload(sys)

sys.setdefaultencoding( "utf-8" )

try:

# command line arguments.

parser = OptionParser(usage="usage: %prog [options]", version="%prog 1.0")

parser.add_option("-c", "--host", dest = "host", action = "store", default = "127.0.0.1", type = "string", help = "Specify cabinet host.")

parser.add_option("-p", "--port", dest = "port", action = "store", default = 9090, type = "int", help = "Specify cabinet port.")

(options, args) = parser.parse_args()

# connect to cabinet server.

transport = TSocket.TSocket(options.host, options.port)

transport = TTransport.TBufferedTransport(transport)

protocol = TCompactProtocol.TCompactProtocol(transport)

client = CabinetStorage.Client(protocol)

transport.open()

except Exception as e:

print e

os._exit(1)

# welcome info.

print "Cabinet Shell V1.0, Copyright i-MD 2013, All rights reserved."

print "For usage, invoke help().\n"

# main functions.

def help():

print "Main functions:"

print " get(db, id)"

print " set(db, id, val)"

print " delete(db, id)"

print " unpackDocInfo(docstr)"

print " packDocInfo(doc)"

def get(db, id):

return client.Get(db, id)

def set(db, id, val):

return client.Set(db, id, val)

def delete(db, id):

return client.delete(db, id)

def packDocInfo(doc):

transportOut = TTransport.TMemoryBuffer()

protocolOut = TCompactProtocol.TCompactProtocol(transportOut)

doc.write(protocolOut)

return transportOut.getvalue()

def unpackDocInfo(docstr):

transportIn = TTransport.TMemoryBuffer(docstr)

protocolIn = TCompactProtocol.TCompactProtocol(transportIn)

doc = Document()

doc.read(protocolIn)

return doc

不满意的地方是原生的python shell不保存history。