def create_specific_rule(self, tenantId, serverId, rule):
"""Creates a new specific rule for a server
:param str tenantId: The id of the tenant
:param str serverId: The id of the server
:param str rule: The rule description in json format
"""
try:
entity = Entity.objects.get(serverId__exact=serverId)
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)
entity.save()
try:
condition = self.getContition(rule)
action = self.getAction(rule)
name = self.getName(rule)
except Exception as err:
raise ValueError(str(err) + " is missing")
self.checkRule(name, condition, action)
# Its necesary modify action to get subscriptionId
modifiedAction = self.pimp_rule_action(action, name, serverId)
modifiedCondition = self.pimp_rule_condition(condition, name, serverId)
createdAt = datetime.datetime.now(tz=timezone.get_default_timezone())
ruleId = uuid.uuid1()
rule = SpecificRule(
specificRule_Id=ruleId,
tenantId=tenantId,
name=name,
condition=condition,
action=action,
clips_condition=modifiedCondition,
clips_action=modifiedAction,
createdAt=createdAt,
)
"""try:
rule = SpecificRule.objects.get(serverId__exact=serverId, )
raise ValueError("rule name already exists")
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)"""
rule.save()
entity.specificrules.add(rule)
rule.save()
ruleResult = RuleModel()
ruleResult.ruleId = str(ruleId)
logger.info("RuleId %s was created for server %s" % (str(ruleId), serverId))
return ruleResult
def create_specific_rule(self, tenantId, serverId, rule):
"""Creates a new specific rule for a server
:param str tenantId: The id of the tenant
:param str serverId: The id of the server
:param str rule: The rule description in json format
"""
try:
entity = Entity.objects.get(serverId__exact=serverId)
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)
entity.save()
try:
condition = self.getContition(rule)
action = self.getAction(rule)
name = self.getName(rule)
except Exception as err:
raise ValueError(str(err) + " is missing")
self.checkRule(name, condition, action)
#Its necesary modify action to get subscriptionId
modifiedAction = self.pimp_rule_action(action, name, serverId)
modifiedCondition = self.pimp_rule_condition(condition, name, serverId)
createdAt = datetime.datetime.now(tz=timezone.get_default_timezone())
ruleId = uuid.uuid1()
rule = SpecificRule(specificRule_Id=ruleId,
tenantId=tenantId,
name=name,
condition=condition,
action=action,
clips_condition=modifiedCondition,
clips_action=modifiedAction,
createdAt=createdAt)
"""try:
rule = SpecificRule.objects.get(serverId__exact=serverId, )
raise ValueError("rule name already exists")
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)"""
rule.save()
entity.specificrules.add(rule)
rule.save()
ruleResult = RuleModel()
ruleResult.ruleId = str(ruleId)
logger.info("RuleId %s was created for server %s" %
(str(ruleId), serverId))
return ruleResult
def subscribe_to_rule(self, tenantId, serverId, subscription):
"""Creates a server subscription to a rule.
:param str tenantId: The id of the tenant
:param str serverId: The id of the server
:param str subscription: The subscription description in json format
"""
context_broker_subscription = False
try:
entity = Entity.objects.get(serverId__exact=serverId)
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)
entity.save()
ruleId = json.loads(subscription)["ruleId"]
SpecificRule.objects.get(specificRule_Id__exact=ruleId, entity__exact=serverId)
url = json.loads(subscription)["url"]
# Verify that there is no more subscriptions to the rule for that server
it = entity.subscription.iterator()
for sub in it:
if sub.serverId == serverId:
context_broker_subscription = sub.cbSubscriptionId
if sub.ruleId == ruleId:
raise Conflict("Subscription already exists")
self.verify_url(url)
if not context_broker_subscription:
cbSubscriptionId = self.orionClient.contextBrokerSubscription(tenantId, serverId)
else:
cbSubscriptionId = context_broker_subscription
subscription_Id = uuid.uuid1()
subscr = Subscription(
subscription_Id=subscription_Id,
ruleId=ruleId,
url=url,
serverId=serverId,
cbSubscriptionId=cbSubscriptionId,
)
subscr.save()
entity.subscription.add(subscr)
entity.save()
logger.info(
"Server %s was subscribed to rule %s: cbSubscriptionId is %s and internal subscription %s"
% (serverId, ruleId, cbSubscriptionId, str(subscription_Id))
)
return subscription_Id
def subscribe_to_rule(self, tenantId, serverId, subscription):
"""Creates a server subscription to a rule.
:param str tenantId: The id of the tenant
:param str serverId: The id of the server
:param str subscription: The subscription description in json format
"""
context_broker_subscription = False
try:
entity = Entity.objects.get(serverId__exact=serverId)
except Entity.DoesNotExist as err:
entity = Entity(serverId=serverId, tenantId=tenantId)
entity.save()
ruleId = json.loads(subscription)['ruleId']
SpecificRule.objects.get(specificRule_Id__exact=ruleId,
entity__exact=serverId)
url = json.loads(subscription)['url']
#Verify that there is no more subscriptions to the rule for that server
it = entity.subscription.iterator()
for sub in it:
if sub.serverId == serverId:
context_broker_subscription = sub.cbSubscriptionId
if sub.ruleId == ruleId:
raise Conflict("Subscription already exists")
self.verify_url(url)
if not context_broker_subscription:
cbSubscriptionId = self.orionClient.contextBrokerSubscription(
tenantId, serverId)
else:
cbSubscriptionId = context_broker_subscription
subscription_Id = uuid.uuid1()
subscr = Subscription(subscription_Id=subscription_Id,
ruleId=ruleId,
url=url,
serverId=serverId,
cbSubscriptionId=cbSubscriptionId)
subscr.save()
entity.subscription.add(subscr)
entity.save()
logger.info(
"Server %s was subscribed to rule %s: cbSubscriptionId is %s and internal subscription %s"
% (serverId, ruleId, cbSubscriptionId, str(subscription_Id)))
return subscription_Id
