def getCountries():
defaultParam = {"lazyDays": [], "holidays": [], "enabled": True}
res = mongo.config().find_one(MEDIATION_DOCUMENT)["countries"]
countries = {}
for countryName, country in res.items():
countries[countryName] = {**defaultParam, **country}
return countries
def deleteFlow(flow):
dataPath = "lobs." + flow["country"] + "." + flow[
"lobName"] + "." + flow["type"] + "." + flow["name"]
return mongo.config().update_one(MEDIATION_DOCUMENT,
{"$unset": {
dataPath: {}
}})
def getLobs(country, enabledOnly=False):
"""
returns config for all inputs and forwards. Config is derived from parent object.
:return:
"""
res = mongo.config().find_one(MEDIATION_DOCUMENT)
countryEnabled = MediationConfig.getCountryByName(country)["enabled"]
defaultConfig = {
"granularity": 480,
"hardAlarmLevel": 0.5,
"softAlarmLevel": 0.7,
"difference": "day",
"enabled": countryEnabled,
"minimalExpectation": 1,
"lazyDayDifference": 0.7
}
if country not in res["lobs"]:
return {}
for lobName, config in res["lobs"][country].items():
config["country"] = country
config["name"] = lobName
if "options" not in config:
config["options"] = {}
config["options"] = {**defaultConfig, **config["options"]}
config["options"]["enabled"] = defaultConfig["enabled"] and config[
"options"]["enabled"]
if "inputs" not in config:
config["inputs"] = {}
if "forwards" not in config:
config["forwards"] = {}
config["flows"] = {}
inputs = {}
for flowName, flowOptions in config["inputs"].items():
flow = {
"options":
setFlowDefaultOptions(flowOptions,
parentObj=config["options"])
}
flow["name"] = flowName
flow["type"] = "inputs"
flow["lobName"] = lobName
flow[
"dataPath"] = country + "." + lobName + ".inputs." + flowName
flow["gName"] = country + "_" + lobName + "_" + flowName
flow["country"] = country
inputs[flowName] = flow
config["flows"][flowName] = flow
config["inputs"] = inputs
forwards = {}
for flowName, flowOptions in config["forwards"].items():
inputName = flowName.split(":")[0]
if inputName in config["inputs"]:
flow = {
"options":
setFlowDefaultOptions(
flowOptions,
parentObj=inputs[inputName]["options"])
}
else:
flow = {
"options":
setFlowDefaultOptions(flowOptions,
parentObj=config["options"])
}
flow["name"] = flowName
flow["type"] = "forwards"
flow["lobName"] = lobName
flow["country"] = country
flow[
"dataPath"] = country + "." + lobName + ".forwards." + flowName
flow["gName"] = country + "_" + lobName + "_" + flowName
forwards[flowName] = flow
config["flows"][flowName] = flow
config["forwards"] = forwards
lobs = res["lobs"][country]
if enabledOnly:
for lobName, lob in lobs.items():
for flowName, flow in lob["flows"].copy().items():
if flow["options"]["enabled"] == False:
del lob["flows"][flowName]
del lob[flow["type"]][flowName]
return lobs
def deleteLob(lob):
dataPath = "lobs." + lob["country"] + "." + lob["name"]
return mongo.config().update_one(MEDIATION_DOCUMENT,
{"$unset": {
dataPath: {}
}})
def addLob(country, name):
dataPath = "lobs." + country + "." + name
return mongo.config().update_one(MEDIATION_DOCUMENT,
{"$set": {
dataPath: {}
}})
def addFlow(flow):
dataPath = "lobs." + flow["country"] + "." + flow[
"lobName"] + "." + flow["type"] + "." + flow["name"]
mongo.config().update_one(MEDIATION_DOCUMENT, {"$set": {dataPath: {}}})
pass
def getCollection():
return mongo.config()