From 6a99d25018d9ead5cf4d99620d143d7904025e75 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9my=20HUBSCHER?= Date: Fri, 22 Jan 2016 13:38:42 +0100 Subject: [PATCH] setup_remote_client should return the remote_client rather than setting it. --- kinto_client/importer.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/kinto_client/importer.py b/kinto_client/importer.py index 635dc81..acdc276 100644 --- a/kinto_client/importer.py +++ b/kinto_client/importer.py @@ -200,27 +200,29 @@ def setup_remote_client(self, remote_client=None): self.args['bucket'], self.args['collection'])) - self.remote_client = Client(server_url=self.args['host'], - auth=self.args['auth'], - bucket=self.args['bucket'], - collection=self.args['collection']) + remote_client = Client(server_url=self.args['host'], + auth=self.args['auth'], + bucket=self.args['bucket'], + collection=self.args['collection']) # Create bucket # XXX: Move this to a configure # XXX: Add a create if not exist functionality try: - self.remote_client.create_bucket( + remote_client.create_bucket( permissions=self.bucket_permissions) except KintoException as e: if not hasattr(e, 'response') or e.response.status_code != 412: raise e try: - self.remote_client.create_collection( + remote_client.create_collection( permissions=self.collection_permissions) except KintoException as e: if e.response.status_code != 412: raise e + return remote_client + def get_remote_records(self): self.logger.log(COMMAND_LOG_LEVEL, 'Working on %r' % self.args['host']) return list(self.remote_client.get_records())