mirror of https://gitlab.com/akasroua/covot
Implement CSV reader
This commit is contained in:
parent
1504ccc3fe
commit
5a3afd0bfa
|
@ -1 +1,2 @@
|
||||||
src/constants.py
|
src/constants.py
|
||||||
|
data
|
||||||
|
|
|
@ -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
|
Loading…
Reference in New Issue