class FileHandler(object):
def __init__(self):
self.Validator = Validator()
self.uspsFields = ['Last','Delivery','Second','Recipient','Phone']
def ValidAttr(self, attrName, attr):
""" Returns Empty String if Invalid, else returns sanitized attr"""
if self.Validator.validateAttr(attrName, attr) :
return self.Validator.sanitize(attr)
return ''
def splitLast(self,last):
city =''
state = ''
zipcode = ''
row = last.split(' ')
if len(row)==1:
if self.Validator.isValidState(row[0]):
state = row[0]
elif row[0].isdigit():
zipcode = row[0]
else:
city = row[0]
else:
stateIndex = None
for index, el in enumerate(row):
if self.Validator.isValidState(el):
stateIndex = index
if stateIndex:
state = row[stateIndex]
if stateIndex+1 == len(row):
city = ' '.join(row[0:stateIndex])
elif stateIndex == 0:
zipcode = row[stateIndex+1]
else:
city = ' '.join(row[0:stateIndex])
state = row[stateIndex]
zipcode = row[stateIndex+1]
else:
if row[len(row)-1].isdigit():
zipcode = row[len(row)-1]
city = ' '.join(row[0:len(row)-1])
else:
city = ' '.join(row)
return [city,state,zipcode]
def readUSPS(self,filepath):
tsv = open(filepath,'r')
reader = csv.DictReader(tsv, delimiter='\t',restval='')
attributes =[]
for row in reader:
attr ={}
last = self.splitLast(row['Last'])
attr['name'] = row['Recipient']
attr['phone'] = self.ValidAttr('phone',row['Phone'])
attr['address'] = row['Delivery']
attr['address2'] = row['Second']
attr['city'] = self.ValidAttr('city',last[0])
attr['state'] = self.ValidAttr('state',last[1])
attr['zipcode'] = self.ValidAttr('zipcode',last[2])
if attr['name']:
attributes.append(attr)
return attributes
def writeUSPS(self,contacts, filepath):
tsv = open(filepath,'w')
writer = csv.DictWriter(tsv, delimiter='\t', fieldnames=self.uspsFields)
writer.writeheader()
for contact in contacts:
attributes ={}
attributes['Recipient'] = contact.getAttr('name')
last = [contact.getAttr('city'),contact.getAttr('state'),contact.getAttr('zipcode')]
attributes['Second'] = contact.getAttr('address2')
attributes['Last'] = ' '.join(last)
attributes['Delivery'] = contact.getAttr('address')
attributes['Phone'] = contact.getAttr('phone')
writer.writerow(attributes)
def testSanitize(self):
v = Validator()
bad = 'Hello\tRoger\n Rabbit'
good = "Hello Roger Rabbit"
self.assertEqual(good,v.sanitize(bad))
