2

我是 python 新手（和一般编程），正在为一个典型的类创建一个数据库/注册。我希望用户能够从数据库中添加和删除学生，我主要为此使用列表，但遇到了困难。

每当我重新启动程序时，用户已修改的列表返回到我在代码中指定的默认列表。我环顾互联网并尝试将列表保存到单独的 txt 文件中。但是，每次我重新启动程序时，txt 文件也会回到默认值。我希望您能给我一种方法来保存对列表所做的更改并保持这种状态。这是代码（不是很好）：

def menu():
    print "*****************CLASS REGISTER*****************"
    print "Press 1 See The List Of Pupils"
    print "Press 2 To Add New Pupils"
    print "Press 3 To Remove Pupils"
    print "Press 0 To Quit \n"

filename = open('pupil.txt','r')

pupil = ["James Steele", "Blain Krontick", "Leeroy Jenkins", "Tanvir Choudrey"]

def see_list(x):
    print x

def add_pupil(x):
    print "You have chosen to add a new pupil.\n"
    option = raw_input("Please type the childs name.")
    x.append(option)
    filename = open('pupil.txt','w') 
    filename.write('\n'.join(pupil)) 
    filename.close() 
    print option, "has been added to the system."
    return x

def delete_pupil(x):
    print "You have chosen to remove a pupil.\n"
    option = raw_input("Please type the childs name.")
    if option in x:
        x.remove(option)
        filename = open('pupil.txt','w') 
        filename.write('\n'.join(pupil)) 
        filename.close() 
        print option, "has been removed from the system."
    else:
        print "That person is not in the system."
    return x


one = 1
while one != 0:
    menu()
    option = input() 
    if option == 1:
        see_list(pupil)
    elif option == 2:
        add_pupil(pupil)
    elif option == 3:
        delete_pupil(pupil)
    elif option == 0:
        break
    else:
        print "That is not a valible choice."

filename = open('pupil.txt','w') 
filename.write('\n'.join(pupil)) 
filename.close() 


if option == 0:
    quit
4

3 回答 3

4

好吧，您只是打开pupil.txt文件，但从不读回其内容。你需要这样的东西：

filename = open('pupil.txt', 'r')
contents = filename.read()
filename.close()

pupil = [name for name in contents.split('\n') if name]

此外，您需要处理pupil.txt文件不存在的情况；这可以通过try..except围绕 IO 调用的块来完成。

最后，正如上面提到的评论之一，看看这个pickle模块，它可以让你将一个 Python 对象以 Python 的内部格式存储在一个文件中（这不是真正可读的，但可以为你省去很多麻烦）。

于 2012-05-18T19:48:08.377 回答
1

与您的问题没有直接关系，但是：

one = 1
while one != 0:
    ...

很傻。所有你需要的是：

while True:
    ...
于 2012-05-18T19:50:22.283 回答
0

这就是数据库的用途。使用sqlite- 一个简单的基于文件的数据库，其库与 python 捆绑在一起。

于 2012-05-18T19:53:10.433 回答