manyi/script/upgrade/import_adapter_and_eventer.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

"""

"""

import os
import sys

#: current_dir - 2


PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
sys.path.insert(0, PROJECT_ROOT)

from script.base import init_env, get_logger

logger = get_logger(__name__)

init_env(interactive=True)

from apps.web.core.models import DriverCode, DriverAdapter, DriverEventer

for item in DriverCode.objects().all(): # type: DriverCode
    adapter = DriverAdapter.objects(adapterFile = item.adapterFile).first()
    if not adapter:
        DriverAdapter(adapterFile = item.adapterFile, adapterVer = item.adapterVer, adapter = item.adapter).save()
    else:
        if adapter.adapter != item.adapter:
            print('adapter {} exists.'.format(item.adapterFile))

    if not DriverEventer.objects(eventerFile = item.eventerFile).first():
        DriverEventer(eventerFile = item.eventerFile, eventerVer = item.eventerVer).save()
    else:
        pass

try:
    DriverAdapter(adapterFile = 'commonPulse', adapter = 'CommonPulseAdapter').save()
    DriverAdapter(adapterFile = 'offline', adapter = 'OfflineSmartBox').save()

    DriverEventer(eventerFile = 'commonPulse').save()
    DriverEventer(eventerFile = 'offline').save()
except Exception as e:
    pass