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 save_country_data(res): """ 保存全国的数据 :return: """ count = session.query(Count).filter(Count.area_name == '全国').first() if not count: count = Count() count.area_type = 'country' count.area_name = '全国' count.from_source = 'dxy' session.add(count) filter_data(res, count) now = datetime.now().strftime("%Y-%m-%d %H:%M:%S") 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()