def update_province_area_state(res):
"""
更新各省级数据
:return:
"""
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
data = session.query(Count).filter(Count.area_type == 'province').all()
map_data = {}
for count in data:
map_data[count.area_name] = count
for o in res:
comment = o['comment']
key = o['provinceShortName']
count = map_data.get(key)
if not count:
count = Count()
count.from_source = 'dxy'
count.area_name = key
count.area_type = 'province'
count.province = key
session.add(count)
json_to_model(count, o)
if comment != '':
num_data = format_data(comment)
filter_data(num_data, count)
count.update_time = now
session.commit()
def update_city_area_state(res):
data = session.query(Count).filter(Count.area_type == 'city').all()
map_data = {}
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
for count in data:
map_data["city_" + count.province + "_" + count.area_name] = count
for province in res:
cities = province['cities']
province = province['provinceShortName']
for city in cities:
city_key = city['cityName']
city_count = map_data.get("city_" + province + "_" + city_key)
if not city_count:
city_count = Count()
city_count.from_source = 'dxy'
city_count.area_name = city_key
city_count.area_type = 'city'
city_count.province = province
session.add(city_count)
json_to_model(city_count, city)
city_count.update_time = now
session.commit()
