89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
### Util library for interacting with csv files
|
|
|
|
import csv
|
|
|
|
|
|
def readerWithFunction(filepath, function, arg=None):
|
|
"""Read from a csv and execute a function using each row. Return a sequential list of all values returned by the function.
|
|
|
|
filepath -- the path of the csv we're reading\n
|
|
function -- the function we're executing once per row\n
|
|
arg -- if the function takes an additional argument, we pass arg (default None)\n
|
|
"""
|
|
with open(filepath, newline="") as file:
|
|
reader = csv.reader(file, delimiter=",")
|
|
value = []
|
|
for row in reader:
|
|
if arg == None:
|
|
value.append(function(row))
|
|
else:
|
|
value.append(function(row, arg))
|
|
file.close()
|
|
return value
|
|
|
|
|
|
def dictWriter(filepath, dict, flag="w"):
|
|
"""Completely rewrite a csv using the values in a single dictionary.
|
|
|
|
filepath -- the path of the csv, whether or not it exists\n
|
|
dict -- the dictionary to pull all values from\n
|
|
flag -- flag for open() (default "w")\n
|
|
"w" will also activate writeheader()\n
|
|
"""
|
|
with open(filepath, flag, newline="") as file:
|
|
|
|
dictWriter = csv.DictWriter(file, fieldnames=dict.keys(), delimiter=",")
|
|
if flag == "w":
|
|
dictWriter.writeheader()
|
|
dictWriter.writerow(dict)
|
|
file.close()
|
|
|
|
|
|
def dictReaderFirstRow(filepath):
|
|
"""Get a dictionary of the first (non-header) row of values in a csv
|
|
|
|
filepath -- the path of the csv, whether or not it exists\n
|
|
"""
|
|
with open(filepath, newline="") as file:
|
|
dict = {}
|
|
dictReader = csv.DictReader(file)
|
|
for row in dictReader:
|
|
firstRow = True
|
|
if firstRow:
|
|
dict = row
|
|
firstRow = False
|
|
file.close()
|
|
return dict
|
|
|
|
|
|
def dictReaderMultiRow(filepath, idName):
|
|
"""Get a dictionary of dictionaries of the (non-header) row of values in a csv
|
|
|
|
filepath -- the path of the csv, whether or not it exists\n
|
|
idName -- the property in the header of the csv to use as each key of the resulting dictionary\n
|
|
"""
|
|
with open(filepath, newline="") as file:
|
|
dict = {}
|
|
dictReader = csv.DictReader(file)
|
|
for row in dictReader:
|
|
dict[row[idName]] = row
|
|
file.close()
|
|
return dict
|
|
|
|
|
|
def getIndexOfField(filepath, field):
|
|
"""Get the index (column number) of a field in a csv.
|
|
|
|
filepath -- the path of the csv we're searching\n
|
|
field -- the field string we're looking for\n
|
|
"""
|
|
return readerWithFunction(filepath, sub_getIndexOfField, field)[0]
|
|
|
|
|
|
def sub_getIndexOfField(row, field):
|
|
"""Helper function for getIndexOfField() that can be passed to readerWithFunction().
|
|
|
|
row - the array representation of a csv row (presumably the fields row)\n
|
|
field - the field string we're looking for\n
|
|
"""
|
|
return row.index(field)
|