-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtry_fetch_locations.py
37 lines (32 loc) · 1.41 KB
/
try_fetch_locations.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import requests
import csv
def get_coordinates(place_name):
url = f"https://nominatim.openstreetmap.org/search?q={place_name}&format=json"
response = requests.get(url)
if response.status_code == 200:
data = response.json()
if data:
return float(data[0]['lat']), float(data[0]['lon'])
return None, None
def update_csv(input_file, output_file):
with open(input_file, 'r', encoding='utf-8') as f:
reader = csv.reader(f, delimiter=';')
next(reader) # Skip header
rows = list(reader)
for row in rows:
if row[1] == '0' and row[2] == '0': # If lat lon are missing
lat, lon = get_coordinates(row[5] + ", " + row[7].replace(',', '.'))
if lat == None: #If precise location cannot be found, skip
continue;
print(f"Fetched missing coordinates for {row[5]}, {row[7]} - {lat}, {lon}")
row[1] = str(lat)
row[2] = str(lon)
with open(output_file, 'w', encoding='utf-8', newline='') as f:
writer = csv.writer(f, delimiter=';')
writer.writerow(['Cislo zastavky', 'Zemepisna sirka', 'Zemepisna dlzka', 'Stat', 'Okres', 'Obec', 'Cast obce', 'Nazov zastavky'])
writer.writerows(rows)
if __name__ == "__main__":
input_file = "data/stops.csv"
output_file = "updated_bus_stops.csv"
update_csv(input_file, output_file)
print("CSV file updated successfully.")