26 lines
822 B
Python
26 lines
822 B
Python
import csv
|
|
import os
|
|
|
|
|
|
class ReadCsv:
|
|
def __init__(self, **kwargs):
|
|
self.filedir = kwargs.get('filedir')
|
|
self.filename = kwargs.get('filename')
|
|
self.delimiter = kwargs.get('delimiter',',')
|
|
self.lineterminator = kwargs.get('lineterminator','\n')
|
|
|
|
def readcsv(self, drop_header=False):
|
|
result=[]
|
|
with open(os.path.join(self.filedir, self.filename), newline='') as file:
|
|
reader = csv.reader(file, delimiter=self.delimiter)
|
|
if drop_header: next(reader)
|
|
for row in reader:
|
|
row = tuple(row)
|
|
result.append(row)
|
|
return result
|
|
|
|
def rm_csv(self):
|
|
file_2_delete = os.path.join(self.filedir, self.filename)
|
|
if os.path.exists(file_2_delete):
|
|
os.remove(file_2_delete)
|