对关闭的 csv 文件的 I/O 操作

Question

我在.csv 文件中有一些信息。我希望能够检查文件里面是否有特定的代码。

下面是我的代码：

import csv
with open('bezoekerscode.csv', 'r') as f:
    reader = csv.reader(f)


def code_control():
    code = str(input('Type hier uw code in die u wilt controleren:\n'))
    if code in reader:
        print('code komt overheen, en wordt nu gecheckt of de aanbieder wel correct is.')
        f.close()
    else:
        print('deze code komt niet overheen.')
        f.close()

code_control()

结果应该是：

• 代码匹配

  或

• 代码不匹配

Answer 1

您可以利用 Python 中的空列表 [] 在 if 子句中被评估为 False 的事实：

if [line for line in reader.readlines() if code in line.strip()]:
    # if code doesn't match a line, it will return [] ergo false
    # and else is going to get executed.
else: 
    # Do else stuff.

您通常也希望将您的 with open 语句移动到函数中。 with 之后会自动关闭你的文件。您的代码看起来像这样会更好：

import csv

def code_control():
    with open('bezoekerscode.csv', 'r') as f:
         reader = csv.reader(f)
         code = str(input('Type hier uw code in die u wilt controleren:\n'))
        if [line for line in reader.readlines() if code in line.strip()]::
            print('code komt overheen, en wordt nu gecheckt of de aanbieder wel correct is.')
    else:
        print('deze code komt niet overheen.')

code_control()

---

请注意，您不能对已关闭的文件执行操作。要评估文件是否仍然打开，您可以检查其file.closed 属性。

Answer 2

您的代码的问题是您在该 code_control 函数中使用了 reader 变量，但该变量在 context 中是本地的>。

您可以全局声明它，也可以将其作为参数发送给 code_control 函数：

import csv

def code_control(reader):
    code = str(input('Type hier uw code in die u wilt controleren:\n'))
    if code in reader:
        print('code komt overheen, en wordt nu gecheckt of de aanbieder wel correct is.')
    else:
        print('deze code komt niet overheen.')

with open('bezoekerscode.csv', 'r') as f:
   reader = csv.reader(f)
   code_control(reader)

你也不需要 f.close() 因为你通过 with 使用的上下文管理器已经处理了。