import requests

from flights_mongodb import mongo_con_parse
from settings import FLIGHTS_CITY_AIRPORT_CODE_TAB


def insert_codes():

    """
        网站没有城市代码,得自己找 根据航段找...
        采集航段中
            # 能查到的城市码就这几个　
                城市码和机场码不一致: 'NRT', KIX'
                存疑 HKO "北海道 (所有机场)", EKK "四國 (所有機場)" 应该是网站为了区分地区自定义的
                城市机场码一致的 'OKA', 'CTS', 'TAK', 'KMJ', 'KOJ', 'KMI', 'AKJ', 'OIT', 'MYJ', 'NGO', 'FUK', 'NGS' KCZ
            # 没有的城市/机场码: OSA SPK TYO SHI
            这里直接手动定义
    """
    airport_code_li = [
        {"city_code": "TYO", "country_code": "JP", "airport_code": "NRT"},
        {"city_code": "OSA", "country_code": "JP", "airport_code": "KIX"},

        {"city_code": "HKO", "country_code": "JP", "airport_code": "HKO"},
        {"city_code": "EKK", "country_code": "JP", "airport_code": "EKK"},

        {"country_code": "JP", "city_code": "OKA", "airport_code": "OKA"},
        {"country_code": "JP", "city_code": "CTS", "airport_code": "CTS"},
        {"country_code": "JP", "city_code": "TAK", "airport_code": "TAK"},
        {"country_code": "JP", "city_code": "KMJ", "airport_code": "KMJ"},
        {"country_code": "JP", "city_code": "KOJ", "airport_code": "KOJ"},
        {"country_code": "JP", "city_code": "KMI", "airport_code": "KMI"},
        {"country_code": "JP", "city_code": "AKJ", "airport_code": "AKJ"},
        {"country_code": "JP", "city_code": "OIT", "airport_code": "OIT"},
        {"country_code": "JP", "city_code": "MYJ", "airport_code": "MYJ"},
        {"country_code": "JP", "city_code": "NGO", "airport_code": "NGO"},
        {"country_code": "JP", "city_code": "FUK", "airport_code": "FUK"},
        {"country_code": "JP", "city_code": "NGS", "airport_code": "NGS"},
        {"country_code": "JP", "city_code": "KCZ", "airport_code": "KCZ"},


    ]
    print(len(airport_code_li))

    db = mongo_con_parse()
    website = "gk"
    db.get_collection(FLIGHTS_CITY_AIRPORT_CODE_TAB).delete_many({"website": website})  # 清空原来属于该航司的数据
    for item in airport_code_li:
        item["website"] = website
        db.get_collection(FLIGHTS_CITY_AIRPORT_CODE_TAB).insert_one(item)
    print('insert finish...')


if __name__ == "__main__":
    insert_codes()