OLD | NEW |
| (Empty) |
1 # Copyright 2012 Google Inc. All Rights Reserved. | |
2 # | |
3 # Licensed under the Apache License, Version 2.0 (the "License"); | |
4 # you may not use this file except in compliance with the License. | |
5 # You may obtain a copy of the License at | |
6 # | |
7 # http://www.apache.org/licenses/LICENSE-2.0 | |
8 # | |
9 # Unless required by applicable law or agreed to in writing, software | |
10 # distributed under the License is distributed on an "AS IS" BASIS, | |
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
12 # See the License for the specific language governing permissions and | |
13 # limitations under the License. | |
14 | |
15 """ | |
16 Class that holds state (bucket_storage_uri_class and debug) needed for | |
17 instantiating StorageUri objects. The StorageUri func defined in this class | |
18 uses that state plus gsutil default flag values to instantiate this frequently | |
19 constructed object with just one param for most cases. | |
20 """ | |
21 | |
22 import boto | |
23 from gslib.exception import CommandException | |
24 | |
25 | |
26 class StorageUriBuilder(object): | |
27 | |
28 def __init__(self, debug, bucket_storage_uri_class): | |
29 """ | |
30 Args: | |
31 debug: Debug level to pass in to boto connection (range 0..3). | |
32 bucket_storage_uri_class: Class to instantiate for cloud StorageUris. | |
33 Settable for testing/mocking. | |
34 """ | |
35 self.bucket_storage_uri_class = bucket_storage_uri_class | |
36 self.debug = debug | |
37 | |
38 def StorageUri(self, uri_str, is_latest=False): | |
39 """ | |
40 Instantiates StorageUri using class state and gsutil default flag values. | |
41 | |
42 Args: | |
43 uri_str: StorageUri naming bucket or object. | |
44 is_latest: boolean indicating whether this versioned object represents the | |
45 current version. | |
46 | |
47 Returns: | |
48 boto.StorageUri for given uri_str. | |
49 | |
50 Raises: | |
51 InvalidUriError: if uri_str not valid. | |
52 """ | |
53 return boto.storage_uri( | |
54 uri_str, 'file', debug=self.debug, validate=False, | |
55 bucket_storage_uri_class=self.bucket_storage_uri_class, | |
56 suppress_consec_slashes=False, is_latest=is_latest) | |
OLD | NEW |