python - 使用列表存储折扣代码，并在兑换后将其从列表中删除

Question

我将具有不同前缀和末尾唯一数字的折扣代码( 10OFF<abc> 、 25OFF<abc> 、 50OFF<abc> 等)存储在文件中，然后将该文件加载到列表中。

我正在尝试创建一个函数，以便在兑换它们时，将它们从列表中删除，并覆盖文件。现在我正在做的事情看起来像这样:

for x in range(0, 5):
    total += codes[0] + '\n'
    codes.remove(codes[0])
with open('codes.txt', 'w') as f:
    for code in codes:
        f.write(code+'\n')

一方面，我不认为这是一种非常 Pythonic 的做事方式，而且感觉很脏。另一方面，我实际上没有办法指定要选择和删除哪个折扣代码 - 这样做我必须为 10OFF 制作单独的文件。 , 25OFF ，和50OFF代码。

有人有什么建议吗？

Answer 1

这应该做你想要的

list_of_codes=open('codes.txt','rt').read().split('\n')

while True:
    code=input('enter code to remove:')
    if code in list_of_codes:
       break
    else:
       print('code you entered is not in the list')
       continue

list_of_codes.pop(list_of_codes.index(code))

with open('codes.txt','wt') as f:
    [f.write(item+'\n') for item in list_of_codes]