zipcodes.py
# -*- coding :UTF-8 -*-
from datetime import datetime,timedelta,timezone
import csv
class Zipcode:
def __init__(self,dic):
self.utctime = datetime.utcnow().replace(tzinfo=timezone.utc)
self.zipcode = dic['zipcode']
self.timezone = dic['timezoneDiff']
self.time_zone = int(self.timezone)
self.location = Location(dic['city'],dic['state'])
self.geolocation = Geolocation(dic['latitude'],dic['longitude'])
self.daylightSaving = dic['observeDaylightSavings']
self.corresponding_time = self.utctime.astimezone(timezone(timedelta(hours=self.time_zone)))
# override
def __str__(self):
return 'zipcode:{} city:{} state:{} latitude:{} longitude:{} timezoneDiff:{} observeDaylightSavings:{} Corresponding_time:{}'.format(
self.zipcode,self.location.city,self.location.state,self.geolocation.latitude,self.geolocation.longitude,
self.timezone,self.daylightSaving,self.corresponding_time)
class Geolocation:
def __init__(self,latitude,longitude):
self.latitude = latitude
self.longitude = longitude
class Location:
def __init__(self,city,state):
self.city = city
self.state = state
allDic = {}
def readCSV(path):
try:
with open(path, 'r') as f:
read = csv.reader(f)
csv_keys = next(read)
csv_reader = csv.DictReader(f, fieldnames=csv_keys)
for row in csv_reader:
d = {}
for k, v in row.items():
d[k] = v
allDic[d['zipcode']]=Zipcode(d)
#print(allDic[d['zipcode']])
return allDic
except FileNotFoundError:
print("Didn't find")
except:
raise
finally:
if f:
f.close()
#Plz read the csv file first!
#path = r'S:\temp\test.csv'
#all_csv_info =readCSV(path)
findBy.py
# -*- coding :UTF-8 -*-
import zipcodes
#Plz read csv file first
path =r'S:\temp\test.csv'
all_csv_info =zipcodes.readCSV(path)
def findByZipcode(input_zipcode):
str_key = str(input_zipcode)
result = zipcodes.allDic.get(str_key,0)
return result
def findInSameState(zipcode,records):
corr_info = []
sublist =[]
for x in zipcode:
for y in records:
temp_state = y.location.state
for t in records:
temp_state2 = t.location.state
if temp_state == temp_state2:
sublist.append(t)
corr_info.append(sublist)
return corr_info
def findByState(state):
for x in state:
temp = x
for key in zipcodes.allDic:
value = zipcodes.allDic[key]
if temp == value.location.state:
print(value)
'''
#Testing for findByState:
input_state = list(map(str,input('State(Plz separated by spaces):').split()))
TheCorrespondingInfo= findByState(input_state)
'''
'''
#Testing for findByZipcode：
input_zipcode = input('ZipCode:')
TheCorrespondingInfo=findByZipcode(input_zipcode)
print(TheCorrespondingInfo)
'''
