玩蛇网提供最新Python编程技术信息以及Python资源下载!

Python shelve方法完成信息录入小工具源码

python 培训

此段python代码是参考了《Python基础教程2》改写的,用了shelve,是关于信息录入的方法。我使用的python版本是2.7的,用的是Notepad++ 6.1.5。我感觉这本书很好,特别适合刚刚接触Python语言的伙伴们入门使用。

main()中的:database = shelve.open('D:\\python-test\\test.dat'),这里的路径请自行修改为自己可以使用的正确路径。

Python shelve方法完成信息录入小工具源码如下:

import sys, shelve
def store_person(db):
	'''
	Store your info.
	'''
	pid = raw_input('Enter your ID: ')
	person = {}
	person['name'] = raw_input('Enter your name: ')
	person['age'] = raw_input('Enter your age: ')
	person['phone'] = raw_input('Enter your phone number: ')
	db[pid] = person

#www.iplaypython.com

def lookup_person(db):
	'''
	Lookup your info.
	'''
	pid = raw_input('Enter the ID: ')
	if pid in db.keys():        #检查输入的ID是否存在
		field = raw_input('What do you want to lookup?(Name, Age, Phone)')
		field = field.strip()   #删除field中可能有的空格和换行符
		if field in ('Name', 'Age', 'Phone'):   #检查用户的输入       
			field = field.strip().lower()
                        print field.capitalize() + ':', db[pid][field]
		else:
			print 'The input is error!Please enter: Name, Age or Phone'
	else:
		print "The ID is not exist!"
		#lookup_person(db)



def print_help():
	print '''
	The available commands are:
	store, lookup,quit,?
	'''

def enter_command():
	cmd = raw_input('Enter your command("?" for help): ')
	cmd = cmd.strip().lower()
	return cmd

def main():
	database = shelve.open('D:\\python-test\\test.dat')
	try:
		while True:
			cmd = enter_command()
			if cmd == 'store':
				store_person(database)
			elif cmd == 'lookup':
				lookup_person(database)
			elif cmd == '?':
				print_help()
			elif cmd == 'quit':
				return
	finally:
		database.close()

if __name__ == '__main__': main()

玩蛇网原创,转载请注明文章出处和来源网址:http://www.iplaypython.com/code/base/b2168.html



微信公众号搜索"玩蛇网Python之家"加关注,每日最新的Python资讯、图文视频教程可以让你一手全掌握。强烈推荐关注!

微信扫描下图可直接关注

玩蛇网Python新手QQ群,欢迎加入: ① 240764603 玩蛇网Python新手群
文章发布日期:2016-06-22 14:07 玩蛇网 www.iplaypython.com

评论列表(网友评论仅供网友表达个人看法,并不表明本站同意其观点或证实其描述)
相关文章推荐
别人正在看
特别推荐
去顶部去底部