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))