1
|
|
|
""" |
2
|
|
|
Copyright George Sibble 2018 |
3
|
|
|
""" |
4
|
1 |
|
from .file_list import B2FileList |
5
|
1 |
|
from ..b2_exceptions import B2Exception |
6
|
1 |
|
from ..api import API |
7
|
|
|
|
8
|
1 |
|
class B2Bucket(object): |
|
|
|
|
9
|
|
|
""" |
10
|
|
|
|
11
|
|
|
""" |
12
|
1 |
|
def __init__(self, connector, parent_list, bucketId, bucketName, bucketType, bucketInfo, lifecycleRules, revision, |
|
|
|
|
13
|
|
|
corsRules, *args, **kwargs): |
14
|
|
|
""" |
15
|
|
|
|
16
|
|
|
:param connector: |
17
|
|
|
:param parent_list: |
18
|
|
|
:param bucketId: |
19
|
|
|
:param bucketName: |
20
|
|
|
:param bucketType: |
21
|
|
|
:param bucketInfo: |
22
|
|
|
:param lifecycleRules: |
23
|
|
|
:param revision: |
24
|
|
|
:param corsRules: |
25
|
|
|
:param args: |
26
|
|
|
:param kwargs: |
27
|
|
|
""" |
28
|
1 |
|
self.bucket_id = bucketId |
29
|
1 |
|
self.bucket_name = bucketName |
30
|
1 |
|
self.bucket_type = bucketType |
31
|
1 |
|
self.bucket_info = bucketInfo |
32
|
1 |
|
self.lifecycle_rules = lifecycleRules |
33
|
1 |
|
self.revision = revision |
34
|
1 |
|
self.cors_rules = corsRules |
35
|
1 |
|
self.connector = connector |
36
|
1 |
|
self.parent_list = parent_list |
37
|
1 |
|
self.deleted = False |
38
|
|
|
|
39
|
1 |
|
def delete(self, delete_files=False, confirm_non_empty=False): |
40
|
|
|
""" Delete a bucket. |
41
|
|
|
|
42
|
|
|
Params: |
43
|
|
|
delete_files: (bool) Delete all files first. |
44
|
|
|
confirm_non_empty: (bool) Confirm deleting on bucket not empty. |
45
|
|
|
""" |
46
|
1 |
|
path = API.delete_bucket |
47
|
1 |
|
files = self.files.all(include_hidden=True) |
|
|
|
|
48
|
1 |
|
if delete_files: |
49
|
|
|
if not confirm_non_empty: |
50
|
|
|
raise B2Exception('Bucket is not empty! Must confirm deletion of all files with confirm_non_empty=True') |
|
|
|
|
51
|
|
|
else: |
52
|
|
|
print("Deleting all files from bucket. Beware API limits!") |
|
|
|
|
53
|
|
|
self.files.delete_all(confirm=True) |
54
|
|
|
|
|
|
|
|
55
|
|
|
|
56
|
1 |
|
params = { |
57
|
|
|
'bucketId': self.bucket_id |
58
|
|
|
} |
59
|
1 |
|
response = self.connector.make_request(path=path, method='post', params=params, account_id_required=True) |
|
|
|
|
60
|
1 |
|
if response.status_code == 200: |
61
|
1 |
|
self.deleted = True |
62
|
1 |
|
del self.parent_list._buckets_by_name[self.bucket_name] |
|
|
|
|
63
|
1 |
|
del self.parent_list._buckets_by_id[self.bucket_id] |
|
|
|
|
64
|
|
|
else: |
65
|
1 |
|
raise B2Exception.parse(response) |
66
|
|
|
|
67
|
1 |
|
def edit(self): |
|
|
|
|
68
|
|
|
#TODO: Edit details |
|
|
|
|
69
|
|
|
pass |
70
|
|
|
|
71
|
1 |
|
@property |
72
|
|
|
def files(self): |
73
|
|
|
""" List of files in the bucket. B2FileList instance. """ |
74
|
|
|
return B2FileList(connector=self.connector, bucket=self) |
|
|
|
|