3 # Copyright 2016-2017 RIFT.IO Inc
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
21 from gi
.repository
import(
27 class CloudAccountNotFound(Exception):
31 class CloudAccountDtsOperdataHandler(object):
32 def __init__(self
, dts
, log
, loop
, project
):
36 self
._project
= project
40 self
.cloud_accounts
= {}
42 def add_cloud_account(self
, account
):
43 self
.cloud_accounts
[account
.name
] = account
44 account
.start_validate_credentials(self
._loop
)
46 def delete_cloud_account(self
, account_name
):
47 del self
.cloud_accounts
[account_name
]
49 def get_saved_cloud_accounts(self
, cloud_account_name
):
50 ''' Get Cloud Account corresponding to passed name, or all saved accounts if name is None'''
51 saved_cloud_accounts
= []
53 if cloud_account_name
is None or cloud_account_name
== "":
54 cloud_accounts
= list(self
.cloud_accounts
.values())
55 saved_cloud_accounts
.extend(cloud_accounts
)
56 elif cloud_account_name
in self
.cloud_accounts
:
57 account
= self
.cloud_accounts
[cloud_account_name
]
58 saved_cloud_accounts
.append(account
)
60 errstr
= "Cloud account {} does not exist".format(cloud_account_name
)
61 raise KeyError(errstr
)
63 return saved_cloud_accounts
66 def create_notification(self
, account
):
67 xpath
= "N,/rw-cloud:cloud-notif"
68 ac_status
= RwCloudYang
.YangNotif_RwCloud_CloudNotif()
69 ac_status
.name
= account
.name
70 ac_status
.message
= account
.connection_status
.details
72 yield from self
._dts
.query_create(xpath
, rwdts
.XactFlag
.ADVISE
, ac_status
)
73 self
._log
.info("Notification called by creating dts query: %s", ac_status
)
76 def _register_show_status(self
):
77 def get_xpath(cloud_name
=None):
78 return "D,/rw-cloud:cloud/account{}/connection-status".format(
79 "[name='%s']" % cloud_name
if cloud_name
is not None else ''
83 def on_prepare(xact_info
, action
, ks_path
, msg
):
84 path_entry
= RwCloudYang
.CloudAcc
.schema().keyspec_to_entry(ks_path
)
85 cloud_account_name
= path_entry
.key00
.name
86 self
._log
.debug("Got show cloud connection status request: %s", ks_path
.create_string())
89 saved_accounts
= self
.get_saved_cloud_accounts(cloud_account_name
)
90 for account
in saved_accounts
:
91 connection_status
= account
.connection_status
92 self
._log
.debug("Responding to cloud connection status request: %s", connection_status
)
93 xpath
= self
._project
.add_project(get_xpath(account
.name
))
94 xact_info
.respond_xpath(
95 rwdts
.XactRspCode
.MORE
,
97 msg
=account
.connection_status
,
100 self
._log
.warning(str(e
))
101 xact_info
.respond_xpath(rwdts
.XactRspCode
.NA
)
104 xact_info
.respond_xpath(rwdts
.XactRspCode
.ACK
)
106 xpath
= self
._project
.add_project(get_xpath())
107 self
._regh
= yield from self
._dts
.register(
109 handler
=rift
.tasklets
.DTS
.RegistrationHandler(
110 on_prepare
=on_prepare
),
111 flags
=rwdts
.Flag
.PUBLISHER
,
114 def _register_validate_rpc(self
):
116 return "/rw-cloud:update-cloud-status"
119 def on_prepare(xact_info
, action
, ks_path
, msg
):
120 if not msg
.has_field("cloud_account"):
121 raise CloudAccountNotFound("Cloud account name not provided")
122 cloud_account_name
= msg
.cloud_account
124 if not self
._project
.rpc_check(msg
, xact_info
=xact_info
):
128 account
= self
.cloud_accounts
[cloud_account_name
]
130 errmsg
= "Cloud account name {} not found in project {}". \
131 format(cloud_account_name
, self
._project
.name
)
132 xact_info
.send_error_xpath(RwTypes
.RwStatus
.FAILURE
,
135 raise CloudAccountNotFound(errmsg
)
137 account
.start_validate_credentials(self
._loop
)
139 yield from self
.create_notification(account
)
141 xact_info
.respond_xpath(rwdts
.XactRspCode
.ACK
)
143 self
._rpc
= yield from self
._dts
.register(
145 handler
=rift
.tasklets
.DTS
.RegistrationHandler(
146 on_prepare
=on_prepare
148 flags
=rwdts
.Flag
.PUBLISHER
,
153 yield from self
._register
_show
_status
()
154 yield from self
._register
_validate
_rpc
()
156 def deregister(self
):
157 yield from self
._rpc
.deregister()
158 yield from self
._regh
.deregister()