X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2common%2Fadapter%2Fnotifications.py;h=d8eace262ae2d6b047ea38282c9aa02187b6bb00;hb=58994b7d851b47456eed1820d36cc06803777e3b;hp=4c49c1b7aa7ac4767f499590fc05bd7f905b014d;hpb=5e02e76308e1677fb106572b885a366eb6c0fbec;p=pti%2Fo2.git

diff --git a/o2common/adapter/notifications.py b/o2common/adapter/notifications.py
index 4c49c1b..d8eace2 100644
--- a/o2common/adapter/notifications.py
+++ b/o2common/adapter/notifications.py
@@ -1,20 +1,66 @@
+# Copyright (C) 2021-2023 Wind River Systems, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
 # pylint: disable=too-few-public-methods
 import abc
-from o2ims import config
 
+from o2common.config import config, conf
+from o2common.service.command.handler import SMOClient
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
 
 SMO_O2_ENDPOINT = config.get_smo_o2endpoint()
 
 
 class AbstractNotifications(abc.ABC):
     @abc.abstractmethod
-    def send(self, message):
+    def send(self, url, message):
         raise NotImplementedError
 
 
+class NoneNotifications(AbstractNotifications):
+    def __init__(self):
+        pass
+
+    def send(self, url, message):
+        pass
+
+
 class SmoO2Notifications(AbstractNotifications):
     def __init__(self, smoO2Endpoint=SMO_O2_ENDPOINT):
         self.smoO2Endpoint = smoO2Endpoint
 
-    def send(self, message):
+    def send(self, url, message):
         pass
+
+
+class SmoNotifications(AbstractNotifications):
+    def __init__(self):
+        logger.debug('In SmoNotifications')
+        if conf.PUBSUB.SMO_AUTH_URL is not None \
+                and conf.PUBSUB.SMO_AUTH_URL != '':
+            logger.debug(f'SMO_AUTH_URL is {conf.PUBSUB.SMO_AUTH_URL}')
+            self.smo_client = SMOClient(
+                conf.PUBSUB.SMO_CLIENT_ID, conf.PUBSUB.SMO_AUTH_URL,
+                conf.PUBSUB.SMO_USERNAME, conf.PUBSUB.SMO_PASSWORD,
+                use_oauth=True)
+        else:
+            self.smo_client = SMOClient()
+
+    def send(self, url, message):
+        try:
+            return self.smo_client.post(url, message)
+        except Exception as e:
+            logger.critical('Notify except: {}'.format(e))
+            return False