def fixDate(date):
return "{0}/{1}/{2}".format(date[0:4], date[4:6], date[6:8])
iso2to3 = {}
with open("countrycodes.txt", "r") as f:
lines = f.readlines()
for line in lines:
line = line.split(",")
iso2to3[line[0]] = line[1].strip()
for DATA in ["CC", "DD", "FG", "FX", "HU", "PP", "RR", "SD", "SS", "TG", "TN", "TX"]:
#for DATA in ["TG"]:
rewrite = False
RR = DataSelector(DATA)
RR.getStations()
# Total number of stations
print "Number of stations: ", len(RR.stations.data["STAID"])
# Rewrite if not done yet.
if not os.path.isdir("rewritten/{0}".format(DATA)):
os.makedirs("rewritten/{0}".format(DATA))
print 'Rewriting'
for station in RR.stations.data["STAID"]:
f = RR.getSingleStation(station, mindate=20080101, maxdate=20090101)
f.rewrite()
RR.wipe_files()
# Select the rewritten files
RR_rewritten = DataSelector(DATA, "rewritten/")
