Repository navigation
Expand file tree
/
Copy pathsseconfig.py
More file actions
118 lines (95 loc) · 3.53 KB
/
Copy pathsseconfig.py
File metadata and controls
118 lines (95 loc) · 3.53 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# -*- coding: utf-8 -*-
# MinIO Python Library for Amazon S3 Compatible Cloud Storage, (C)
# 2020 MinIO, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Request/response of PutBucketEncryption and GetBucketEncryption APIs."""
from __future__ import absolute_import, annotations
from abc import ABCMeta
from typing import Type, TypeVar, cast
from xml.etree import ElementTree as ET
from .xml import Element, SubElement, find, findtext
AES256 = "AES256"
AWS_KMS = "aws:kms"
A = TypeVar("A", bound="Rule")
class Rule:
"""Server-side encryption rule. """
__metaclass__ = ABCMeta
def __init__(
self,
sse_algorithm: str,
kms_master_key_id: str | None = None,
):
self._sse_algorithm = sse_algorithm
self._kms_master_key_id = kms_master_key_id
@property
def sse_algorithm(self) -> str:
"""Get SSE algorithm."""
return self._sse_algorithm
@property
def kms_master_key_id(self) -> str | None:
"""Get KMS master key ID."""
return self._kms_master_key_id
@classmethod
def new_sse_s3_rule(cls: Type[A]) -> A:
"""Create SSE-S3 rule."""
return cls(AES256)
@classmethod
def new_sse_kms_rule(
cls: Type[A],
kms_master_key_id: str | None = None,
) -> A:
"""Create new SSE-KMS rule."""
return cls(AWS_KMS, kms_master_key_id)
@classmethod
def fromxml(cls: Type[A], element: ET.Element) -> A:
"""Create new object with values from XML element."""
element = cast(
ET.Element,
find(element, "ApplyServerSideEncryptionByDefault", True),
)
return cls(
cast(str, findtext(element, "SSEAlgorithm", True)),
findtext(element, "KMSMasterKeyID"),
)
def toxml(self, element: ET.Element | None) -> ET.Element:
"""Convert to XML."""
if element is None:
raise ValueError("element must be provided")
element = SubElement(element, "Rule")
tag = SubElement(element, "ApplyServerSideEncryptionByDefault")
SubElement(tag, "SSEAlgorithm", self._sse_algorithm)
if self._kms_master_key_id is not None:
SubElement(tag, "KMSMasterKeyID", self._kms_master_key_id)
return element
B = TypeVar("B", bound="SSEConfig")
class SSEConfig:
"""server-side encryption configuration."""
def __init__(self, rule: Rule):
if not rule:
raise ValueError("rule must be provided")
self._rule = rule
@property
def rule(self) -> Rule:
"""Get rule."""
return self._rule
@classmethod
def fromxml(cls: Type[B], element: ET.Element) -> B:
"""Create new object with values from XML element."""
element = cast(ET.Element, find(element, "Rule", True))
return cls(Rule.fromxml(element))
def toxml(self, element: ET.Element | None) -> ET.Element:
"""Convert to XML."""
element = Element("ServerSideEncryptionConfiguration")
self._rule.toxml(element)
return element