-
Notifications
You must be signed in to change notification settings - Fork 53
/
city_scripts.py
68 lines (50 loc) · 1.64 KB
/
city_scripts.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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#same_street scripts
import math
def get_Col(db,repo):
'''input: string representing database name db
output: returns list with all key,values from database db
'''
col_list = []
for elem in repo['ckarjadi_johnnyg7.' + db].find({}):
col_list.append(elem)
return col_list
def count_city(col,name):
'''name = area; area is the city where the commGarden is in.'''
fin =[]
d={}
check=[]
#every x will have a dictionary {area: "city"}
for x in col:
city = x[name]
if city not in check:
d[city] = 1
check+=[city]
else:
d[city]+=1
#every x in d is a city name; every d[x] is the # of commGardens in that city
for x in d:
fin+=[{'city':x, 'num_comm_gardens':d[x]}]
return [fin,check]
def count_food(col,name,compare,data, final):
'''name = num_fast_food; compare=count_cities; data = fast_food;
final = count_gardens.'''
fin=[]
d={}
check=[]
for x in col:
try:
business=x['businessname']
city = x['city']
if city in compare and business in data and city not in check:
d[city]=1
check+=[city]
elif city in compare and business in data and city in check:
d[city]+=1
except KeyError:
continue
#every x in final is a dictionary, with city: "..."
for x in final:
city = x['city']
if city in d:
x[name]=d[city]
return final