X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fviews%2Falarm_view.py;h=83189c12fc18879da104bbac8d90f15ad0764338;hb=refs%2Fheads%2Fg-release;hp=f5e6b8b3387476f4c5f0f522f4e194ebe4640952;hpb=c532f7fbf1934c58e79c01dc36cd3eeaf9656b99;p=pti%2Fo2.git diff --git a/o2ims/views/alarm_view.py b/o2ims/views/alarm_view.py index f5e6b8b..83189c1 100644 --- a/o2ims/views/alarm_view.py +++ b/o2ims/views/alarm_view.py @@ -15,8 +15,11 @@ import uuid as uuid from o2common.service import unit_of_work +from o2common.views.view import gen_filter, check_filter from o2common.views.pagination_view import Pagination -from o2common.views.view import gen_filter +from o2common.views.route_exception import BadRequestException, \ + NotFoundException + from o2ims.views.alarm_dto import SubscriptionDTO from o2ims.domain.alarm_obj import AlarmSubscription, AlarmEventRecord @@ -59,23 +62,40 @@ def subscription_one(subscriptionId: str, return first.serialize() if first is not None else None -def subscription_create(subscriptionDto: SubscriptionDTO.subscription, +def subscription_create(subscriptionDto: SubscriptionDTO.subscription_create, uow: unit_of_work.AbstractUnitOfWork): + filter = subscriptionDto.get('filter', '') + consumer_subs_id = subscriptionDto.get('consumerSubscriptionId', '') + + check_filter(AlarmEventRecord, filter) sub_uuid = str(uuid.uuid4()) subscription = AlarmSubscription( sub_uuid, subscriptionDto['callback'], - subscriptionDto['consumerSubscriptionId'], - subscriptionDto['filter']) + consumer_subs_id, filter) with uow: + args = list() + args.append(getattr(AlarmSubscription, 'callback') + == subscriptionDto['callback']) + args.append(getattr(AlarmSubscription, 'filter') == filter) + args.append(getattr(AlarmSubscription, + 'consumerSubscriptionId') == consumer_subs_id) + count, _ = uow.alarm_subscriptions.list_with_count(*args) + if count > 0: + raise BadRequestException("The value of parameters is duplicated") uow.alarm_subscriptions.add(subscription) uow.commit() - return {"alarmSubscriptionId": sub_uuid} + first = uow.alarm_subscriptions.get(sub_uuid) + return first.serialize() def subscription_delete(subscriptionId: str, uow: unit_of_work.AbstractUnitOfWork): with uow: + first = uow.alarm_subscriptions.get(subscriptionId) + if not first: + raise NotFoundException( + "Alarm Subscription {} not found.".format(subscriptionId)) uow.alarm_subscriptions.delete(subscriptionId) uow.commit() return True