37 lines
901 B
Python
37 lines
901 B
Python
|
import sys
|
||
|
|
import csv
|
||
|
|
|
||
|
|
if len(sys.argv) < 4:
|
||
|
|
print('Syntax: ./alt.py $COUNTRY $CITY $VARIANCE')
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
country = sys.argv[1]
|
||
|
|
city = sys.argv[2]
|
||
|
|
var = float(sys.argv[3])
|
||
|
|
|
||
|
|
data_file = open('data.csv')
|
||
|
|
data = csv.reader(data_file, delimiter=';', quotechar='"')
|
||
|
|
|
||
|
|
my_location = []
|
||
|
|
|
||
|
|
for item in data:
|
||
|
|
if item[1] == country:
|
||
|
|
if item[2] == city:
|
||
|
|
my_location = item
|
||
|
|
break
|
||
|
|
|
||
|
|
if my_location:
|
||
|
|
print('%s, %s has an altitude of %s' % (my_location[2], my_location[1], my_location[5]))
|
||
|
|
else:
|
||
|
|
print('Location not found in data!')
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
my_altitude = float(my_location[5])
|
||
|
|
print("\n" + 'Here are the cities that share the same altitude with you:' + "\n")
|
||
|
|
|
||
|
|
for item in data:
|
||
|
|
if my_altitude - var < float(item[5]) and my_altitude + var > float(item[5]):
|
||
|
|
print('%s, %s has an altitude of %s' % (item[2], item[1], item[5]))
|
||
|
|
|
||
|
|
data_file.close()
|