Implement CSV reader

This commit is contained in:
coolneng 2020-10-31 14:04:52 +01:00
parent 1504ccc3fe
commit 5a3afd0bfa
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
2 changed files with 16 additions and 0 deletions

1
.gitignore vendored
View File

@ -1 +1,2 @@
src/constants.py src/constants.py
data

15
src/database.py Normal file
View File

@ -0,0 +1,15 @@
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