# coding=utf-8
import os

from base import init_env


os.environ["DJANGO_SETTINGS_MODULE"] = "configs.dev_zjl"


init_env(False)

import csv

from apps.web.merchant.models import WxMerchantAddress


def load_city_code():
    with open("cityCode.csv", "r") as f:
        r = csv.DictReader(f, fieldnames=["code", "name"])
        for row in r:
            code, name = row["code"], row["name"]
            name = name.replace(u"中国,", "")
            if name.startswith(","):
                name = "{}{}".format(name[1:3], name)

            nameSplit = name.split(",")
            if len(nameSplit) != 3:
                continue

            name = "".join(nameSplit)
            WxMerchantAddress(code = code, name = name).save()