update_scrip.py

import sys
from updateopera import Updater
import importlib
from utils.mysqlopera import COMPANY_PN_LIST
if __name__ == '__main__':
    plat_name = sys.argv[1]
    module_name = 'plat.{}.cate_stock'.format(plat_name)
    module = importlib.import_module(module_name)

    updater = Updater(plat_name)
    if plat_name in COMPANY_PN_LIST:
        updater.up_once(module.levels)
    else:
        updater.up(module.levels)