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
)
77 def _register_show_status(self
):
78 def get_xpath(cloud_name
=None):
79 return "D,/rw-cloud:cloud/account{}/connection-status".format(
80 "[name='%s']" % cloud_name
if cloud_name
is not None else ''
84 def on_prepare(xact_info
, action
, ks_path
, msg
):
85 path_entry
= RwCloudYang
.CloudAcc
.schema().keyspec_to_entry(ks_path
)
86 cloud_account_name
= path_entry
.key00
.name
87 self
._log
.debug("Got show cloud connection status request (action: %s): %s",
88 xact_info
.query_action
, ks_path
.create_string())
91 saved_accounts
= self
.get_saved_cloud_accounts(cloud_account_name
)
92 for account
in saved_accounts
:
93 connection_status
= account
.connection_status
94 self
._log
.debug("Responding to cloud connection status request: %s", connection_status
)
95 xpath
= self
._project
.add_project(get_xpath(account
.name
))
96 xact_info
.respond_xpath(
97 rwdts
.XactRspCode
.MORE
,
99 msg
=account
.connection_status
,
101 except KeyError as e
:
102 self
._log
.warning(str(e
))
103 xact_info
.respond_xpath(rwdts
.XactRspCode
.NA
)
106 xact_info
.respond_xpath(rwdts
.XactRspCode
.ACK
)
108 xpath
= self
._project
.add_project(get_xpath())
109 self
._regh
= yield from self
._dts
.register(
111 handler
=rift
.tasklets
.DTS
.RegistrationHandler(
112 on_prepare
=on_prepare
),
113 flags
=rwdts
.Flag
.PUBLISHER
,
117 def _register_validate_rpc(self
):
119 return "/rw-cloud:update-cloud-status"
122 def on_prepare(xact_info
, action
, ks_path
, msg
):
123 if not msg
.has_field("cloud_account"):
124 raise CloudAccountNotFound("Cloud account name not provided")
125 cloud_account_name
= msg
.cloud_account
127 if not self
._project
.rpc_check(msg
, xact_info
=xact_info
):
131 account
= self
.cloud_accounts
[cloud_account_name
]
133 errmsg
= "Cloud account name {} not found in project {}". \
134 format(cloud_account_name
, self
._project
.name
)
135 xact_info
.send_error_xpath(RwTypes
.RwStatus
.FAILURE
,
138 raise CloudAccountNotFound(errmsg
)
140 account
.start_validate_credentials(self
._loop
)
142 yield from self
.create_notification(account
)
144 xact_info
.respond_xpath(rwdts
.XactRspCode
.ACK
)
146 self
._rpc
= yield from self
._dts
.register(
148 handler
=rift
.tasklets
.DTS
.RegistrationHandler(
149 on_prepare
=on_prepare
151 flags
=rwdts
.Flag
.PUBLISHER
,
156 self
._log
.debug("Register cloud account for project %s", self
._project
.name
)
157 yield from self
._register
_show
_status
()
158 yield from self
._register
_validate
_rpc
()
160 def deregister(self
):
161 self
._log
.debug("De-register cloud account for project %s", self
._project
.name
)
162 self
._rpc
.deregister()
163 self
._regh
.deregister()