covot/src/database.py

16 lines
391 B
Python
Raw Normal View History

2020-10-31 14:04:52 +01:00
from csv import DictReader
def find_user(reader, id):
for row in reader:
if row["numero_de_documento"] == id or row["correo_institucional"] == id:
return row
raise Exception("Error: User not found")
def search_csv(filepath, id):
with open(filepath) as csvfile:
reader = DictReader(csvfile)
user = find_user(reader, id)
return user