def test_lifecycle_translation(self):
"""Tests lifecycle translation for various formats."""
json_text = self.lifecycle_doc
entries_list = LifecycleTranslation.JsonLifecycleToMessage(json_text)
boto_lifecycle = LifecycleTranslation.BotoLifecycleFromMessage(
entries_list)
converted_entries_list = LifecycleTranslation.BotoLifecycleToMessage(
boto_lifecycle)
converted_json_text = LifecycleTranslation.JsonLifecycleFromMessage(
converted_entries_list)
self.assertEqual(json.loads(json_text),
json.loads(converted_json_text))
def test_lifecycle_translation(self):
"""Tests lifecycle translation for various formats."""
# TODO: Use lifecycle_doc again once Boto is updated to support new fields.
# json_text = self.lifecycle_doc
json_text = self.lifecycle_doc_without_storage_class_fields
entries_list = LifecycleTranslation.JsonLifecycleToMessage(json_text)
boto_lifecycle = LifecycleTranslation.BotoLifecycleFromMessage(entries_list)
converted_entries_list = LifecycleTranslation.BotoLifecycleToMessage(
boto_lifecycle)
converted_json_text = LifecycleTranslation.JsonLifecycleFromMessage(
converted_entries_list)
self.assertEqual(json.loads(json_text), json.loads(converted_json_text))
def _SetLifecycleConfig(self):
"""Sets lifecycle configuration for a Google Cloud Storage bucket."""
lifecycle_arg = self.args[0]
url_args = self.args[1:]
# Disallow multi-provider 'lifecycle set' requests.
if not UrlsAreForSingleProvider(url_args):
raise CommandException('"%s" command spanning providers not allowed.' %
self.command_name)
# Open, read and parse file containing JSON document.
lifecycle_file = open(lifecycle_arg, 'r')
lifecycle_txt = lifecycle_file.read()
lifecycle_file.close()
# Iterate over URLs, expanding wildcards and setting the lifecycle on each.
some_matched = False
for url_str in url_args:
bucket_iter = self.GetBucketUrlIterFromArg(url_str,
bucket_fields=['lifecycle'])
for blr in bucket_iter:
url = blr.storage_url
some_matched = True
self.logger.info('Setting lifecycle configuration on %s...', blr)
if url.scheme == 's3':
self.gsutil_api.XmlPassThroughSetLifecycle(
lifecycle_txt, url, provider=url.scheme)
else:
lifecycle = LifecycleTranslation.JsonLifecycleToMessage(lifecycle_txt)
bucket_metadata = apitools_messages.Bucket(lifecycle=lifecycle)
self.gsutil_api.PatchBucket(url.bucket_name, bucket_metadata,
provider=url.scheme, fields=['id'])
if not some_matched:
raise CommandException('No URLs matched')
return 0
def _GetLifecycleConfig(self):
"""Gets lifecycle configuration for a Google Cloud Storage bucket."""
bucket_url, bucket_metadata = self.GetSingleBucketUrlFromArg(
self.args[0], bucket_fields=['lifecycle'])
if bucket_url.scheme == 's3':
sys.stdout.write(self.gsutil_api.XmlPassThroughGetLifecycle(
bucket_url, provider=bucket_url.scheme))
else:
if bucket_metadata.lifecycle and bucket_metadata.lifecycle.rule:
sys.stdout.write(LifecycleTranslation.JsonLifecycleFromMessage(
bucket_metadata.lifecycle))
else:
sys.stdout.write('%s has no lifecycle configuration.\n' % bucket_url)
return 0