Python shelve简单数据存储源码片段

小灰灰 2022-05-01 13:54 90阅读 0赞

下面内容是关于Python shelve简单数据存储片段的内容，应该能对大家有较大用途。

#!/usr/bin/python
    
    import sys, shelve
    
    def store_person(db):
        """ Query user for data and store it in the shelf object """
        pid = raw_input('Enter unique ID number:')
        person = {}
        person['name'] = raw_input('Enter name:')
        person['age'] = raw_input('Enter age:')
        person['phone'] = raw_input('Enter phone number:')
    
        db[pid] = person
    
    def lookup_person(db):
        """ Query user for ID and desired field, and fetch the corresponding data from the shelf object """
        pid = raw_input('Enter unique ID number:')
        temp = db[pid]
        field = raw_input('Please enter name, age or phone:')
        field.strip().lower()
        print field.capitalize() + ': ', temp[field]
    
    def print_help():
        print 'The avaliable commands are:'
        print 'store :Stores infomation about a person'
        print 'lookup :Looks up a person form ID number'
        print 'quit :Save changes and exit'
        print '? :Prints this message'
    
    def enter_command():
        cmd = raw_input('Enter command(? for help):')
        cmd = cmd.strip().lower()
        return cmd
    
    def main():
        database = shelve.open('database')  # database stores in current directory
        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()    # Close database in any condition
    
    if __name__ == '__main__':
        main()        复制代码

转载于:https://juejin.im/post/5cae9500e51d456e442ff2f5