forked from p34709852/monkey
31 lines
836 B
Python
31 lines
836 B
Python
from connectors.vcenter import VCenterJob, VCenterConnector
|
|
from connectors.demo import DemoJob, DemoConnector
|
|
|
|
available_jobs = [VCenterJob, DemoJob]
|
|
|
|
|
|
def get_connector_by_name(name):
|
|
for jobclass in available_jobs:
|
|
if name == jobclass.connector_type.__name__:
|
|
return jobclass.connector_type()
|
|
return None
|
|
|
|
|
|
def get_jobclass_by_name(name):
|
|
for jobclass in available_jobs:
|
|
if jobclass.__name__ == name:
|
|
return jobclass
|
|
|
|
|
|
def refresh_connector_config(mongo, connector):
|
|
properties = mongo.db.connector.find_one({"type": connector.__class__.__name__})
|
|
if properties:
|
|
connector.load_properties(properties)
|
|
|
|
|
|
def load_connector(mongo, name):
|
|
con = get_connector_by_name(name)
|
|
if not con:
|
|
return None
|
|
refresh_connector_config(mongo, con)
|
|
return con |