5 from .client
import client
6 from .client
import connection
7 from .client
import watcher
8 from .model
import Model
10 log
= logging
.getLogger(__name__
)
13 class Controller(object):
14 def __init__(self
, loop
=None):
15 """Instantiate a new Controller.
17 One of the connect_* methods will need to be called before this
18 object can be used for anything interesting.
20 :param loop: an asyncio event loop
23 self
.loop
= loop
or asyncio
.get_event_loop()
24 self
.connection
= None
27 self
, endpoint
, username
, password
, cacert
=None, macaroons
=None):
28 """Connect to an arbitrary Juju controller.
31 self
.connection
= await connection
.Connection
.connect(
32 endpoint
, None, username
, password
, cacert
, macaroons
)
34 async def connect_current(self
):
35 """Connect to the current Juju controller.
39 await connection
.Connection
.connect_current_controller())
41 async def connect_controller(self
, controller_name
):
42 """Connect to a Juju controller by name.
46 await connection
.Connection
.connect_controller(controller_name
))
48 async def disconnect(self
):
49 """Shut down the watcher task and close websockets.
52 if self
.connection
and self
.connection
.is_open
:
53 log
.debug('Closing controller connection')
54 await self
.connection
.close()
55 self
.connection
= None
58 self
, model_name
, cloud_name
=None, credential_name
=None,
59 owner
=None, config
=None, region
=None):
60 """Add a model to this controller.
62 :param str model_name: Name to give the new model.
63 :param str cloud_name: Name of the cloud in which to create the
64 model, e.g. 'aws'. Defaults to same cloud as controller.
65 :param str credential_name: Name of the credential to use when
66 creating the model. Defaults to current credential. If you
67 pass a credential_name, you must also pass a cloud_name,
68 even if it's the default cloud.
69 :param str owner: Username that will own the model. Defaults to
71 :param dict config: Model configuration.
72 :param str region: Region in which to create the model.
75 model_facade
= client
.ModelManagerFacade()
76 model_facade
.connect(self
.connection
)
78 owner
= owner
or self
.connection
.info
['user-info']['identity']
80 # XXX: We should be able to accept a credential_name without
81 # a cloud_name, and just get the cloud_name from the controller.
82 if credential_name
and cloud_name
:
83 credential
= tag
.credential(
85 tag
.untag('user-', owner
),
92 log
.debug('Creating model %s', model_name
)
94 model_info
= await model_facade
.CreateModel(
95 tag
.cloud(cloud_name
),
105 self
.connection
.endpoint
,
107 self
.connection
.username
,
108 self
.connection
.password
,
109 self
.connection
.cacert
,
110 self
.connection
.macaroons
,
115 async def destroy_models(self
, *uuids
):
116 """Destroy one or more models.
118 :param str \*uuids: UUIDs of models to destroy
121 model_facade
= client
.ModelManagerFacade()
122 model_facade
.connect(self
.connection
)
125 'Destroying model%s %s',
126 '' if len(uuids
) == 1 else 's',
130 await model_facade
.DestroyModels([
131 client
.Entity(tag
.model(uuid
))
134 destroy_model
= destroy_models
136 def add_user(self
, username
, display_name
=None, acl
=None, models
=None):
137 """Add a user to this controller.
139 :param str username: Username
140 :param str display_name: Display name
141 :param str acl: Access control, e.g. 'read'
142 :param list models: Models to which the user is granted access
147 def change_user_password(self
, username
, password
):
148 """Change the password for a user in this controller.
150 :param str username: Username
151 :param str password: New password
156 def destroy(self
, destroy_all_models
=False):
157 """Destroy this controller.
159 :param bool destroy_all_models: Destroy all hosted models in the
165 def disable_user(self
, username
):
168 :param str username: Username
173 def enable_user(self
):
174 """Re-enable a previously disabled user.
180 """Forcibly terminate all machines and other associated resources for
186 def get_models(self
, all_
=False, username
=None):
187 """Return list of available models on this controller.
189 :param bool all_: List all models, regardless of user accessibilty
191 :param str username: User for which to list models (admin use only)
196 def get_payloads(self
, *patterns
):
197 """Return list of known payloads.
199 :param str \*patterns: Patterns to match against
201 Each pattern will be checked against the following info in Juju::
214 def get_users(self
, all_
=False):
215 """Return list of users that can connect to this controller.
217 :param bool all_: Include disabled users
223 """Log in to this controller.
228 def logout(self
, force
=False):
229 """Log out of this controller.
231 :param bool force: Don't fail even if user not previously logged in
237 def get_model(self
, name
):
238 """Get a model by name.
240 :param str name: Model name
245 def get_user(self
, username
):
246 """Get a user by name.
248 :param str username: Username