import pymysql

config_dev = {
    "mysql": {
        "host": "192.168.1.238",
        "port": 3306,
        "user": "liexin_data_distribution",
        "password": "liexin_data_distribution#zsyM",
        "database": "liexin_data_distribution"
    },
    "redis": {
        "host": "192.168.1.235",
        "port": 6379,
        "password": "icDb29mLy2s"
    }
}

config_prod = {
    "mysql": {
        "host": "bigdata2.ichunt.db",
        "port": 3306,
        "user": "DsbuUx",
        "password": "sfj09JjsfpQx0",
        "database": "liexin_data_distribution"
    },
    "redis": {
        "host": "172.18.137.38",
        "port": 6379,
        "password": "icDb29mLy2s"
    }
}

"""
为新增列 sku_sn字段赋值，当前为platform = 1的数据，sku_sn为 sku_id字段的字符串形式。
"""


def main(conf: dict):
    host = conf["mysql"]["host"]
    port = conf["mysql"]["port"]
    user = conf["mysql"]["user"]
    password = conf["mysql"]["password"]
    database = conf["mysql"]["database"]
    db = pymysql.connect(host=host, port=port, user=user, password=password, database=database)
    cursor = db.cursor()
    sql = "select id, lx_sku_id from lie_shop_sku where id >= %s and shop_id = 2 order by id asc limit 10000"
    index = 1
    max_id = 100000
    while index <= max_id:
        cursor.execute(query=sql, args=(index,))
        results = cursor.fetchall()
        for i in range(len(results)):
            id = results[i][0]
            lx_sku_id = results[i][1]
            sku_sn = str(lx_sku_id)
            update_sql = "update lie_shop_sku set sku_sn = %s where id = %s"
            cursor.execute(query=update_sql, args=(sku_sn, id,))
            db.commit()
            print(update_sql % (sku_sn, id))
        index += 10000


if __name__ == '__main__':
    main(config_prod)