def update(provider, city, predict):
''' Update the data for a city. '''
# Get the current formatted data for a city
try:
stations = eval(provider).stations(city)
except:
return
# Update the database if the city can be predicted
if predict == 'Yes':
insert.city(city, stations)
# Save the data for the map
geojson = tb.json_to_geojson(stations)
tb.write_json(geojson, '{0}/{1}.geojson'.format(geojsonFolder, city))
# Refresh the latest update time
updates = tb.read_json('{}/updates.json'.format(informationFolder))
updates[city] = datetime.now().isoformat()
tb.write_json(updates, '{}/updates.json'.format(informationFolder))
try:
del namesFile[city]
except:
None
try:
if country not in citiesFile:
del namesFile[country]
except:
None
# Predictions file
try:
del predictionsFile[city]
except:
None
# Updates file
try:
del updatesFile[city]
except:
None
# Notification
print('{} has been removed.'.format(city))
# Save the information files
tb.write_json(stationsFile, '{}/stations.json'.format(informationFolder))
tb.write_json(providersFile, '{}/providers.json'.format(informationFolder))
tb.write_json(centersFile, '{}/centers.json'.format(informationFolder))
tb.write_json(citiesFile, '{}/cities.json'.format(informationFolder))
tb.write_json(namesFile, '{}/names.json'.format(informationFolder))
tb.write_json(predictionsFile, '{}/predictions.json'.format(informationFolder))
tb.write_json(updatesFile, '{}/updates.json'.format(informationFolder))
providersFile[provider] = []
providersFile[provider].append(city)
else:
if city not in providersFile[provider]:
providersFile[provider].append(city)
# City/Center file
center = [np.mean(latitudes), np.mean(longitudes)]
centersFile[city] = center
# Country/City file
if country not in citiesFile:
citiesFile[country] = []
citiesFile[country].append(city)
else:
if city not in citiesFile[country]:
citiesFile[country].append(city)
# Name/RealName file
namesFile[city] = cityRealName
namesFile[country] = countryRealName
# Predictions file
predictionsFile[city] = predict
# Notification
print('{} has been added.'.format(city))
# Save the information files
tb.write_json(stationsFile, '{}/stations.json'.format(informationFolder))
tb.write_json(providersFile, '{}/providers.json'.format(informationFolder))
tb.write_json(centersFile, '{}/centers.json'.format(informationFolder))
tb.write_json(citiesFile, '{}/cities.json'.format(informationFolder))
tb.write_json(namesFile, '{}/names.json'.format(informationFolder))
tb.write_json(predictionsFile, '{}/predictions.json'.format(informationFolder))
