Skip to content

Commit c7932eb

Browse files
committed
Updated ecs-compose-x dependency. Fixed test files
1 parent e56bf8e commit c7932eb

File tree

7 files changed

+424
-443
lines changed

7 files changed

+424
-443
lines changed

.pre-commit-config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,11 @@ repos:
1414
- id: pyupgrade
1515
args: [ "--py38-plus" ]
1616
- repo: https://github.com/psf/black
17-
rev: 22.12.0
17+
rev: 23.1.0
1818
hooks:
1919
- id: black
2020
- repo: https://github.com/pycqa/isort
21-
rev: 5.11.4
21+
rev: 5.12.0
2222
hooks:
2323
- id: isort
2424
args: [ "--profile", "black", "--filter-files" ]

ecs_composex_msk_cluster/msk_cluster_logging.py

Lines changed: 31 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -25,41 +25,43 @@ def add_log_group(cluster: MskCluster, template: Template) -> None:
2525
"""
2626
Creates a new Log Group for the cluster and configures the property for it.
2727
"""
28-
param, prop, value = get_nested_property(
28+
resource_props_to_update: list = get_nested_property(
2929
cluster.cfn_resource, "LoggingInfo.BrokerLogs.CloudWatchLogs"
3030
)
31-
if isinstance(value, CloudWatchLogs):
32-
LOG.warning(
33-
f"{cluster.module.res_key}.{cluster.name} - "
34-
"Properties for LoggingInfo.BrokerLogs.CloudWatchLogs already set. Skipping"
35-
)
36-
return
37-
log_group = LogGroup(f"{cluster.logical_name}BrokersLogGroup")
38-
add_resource(template, log_group)
39-
if value and value is not None:
40-
setattr(param, prop, CloudWatchLogs(Enabled=True, LogGroup=Ref(log_group)))
41-
else:
42-
if not hasattr(cluster.cfn_resource, "LoggingInfo"):
43-
setattr(
44-
cluster.cfn_resource,
45-
"LoggingInfo",
46-
LoggingInfo(
47-
BrokerLogs=BrokerLogs(
48-
CloudWatchLogs=CloudWatchLogs(
49-
Enabled=True, LogGroup=Ref(log_group)
50-
)
51-
)
52-
),
31+
for _resource_prop_to_update in resource_props_to_update:
32+
param, prop, value = _resource_prop_to_update
33+
if isinstance(value, CloudWatchLogs):
34+
LOG.warning(
35+
f"{cluster.module.res_key}.{cluster.name} - "
36+
"Properties for LoggingInfo.BrokerLogs.CloudWatchLogs already set. Skipping"
5337
)
38+
return
39+
log_group = LogGroup(f"{cluster.logical_name}BrokersLogGroup")
40+
add_resource(template, log_group)
41+
if value and value is not None:
42+
setattr(param, prop, CloudWatchLogs(Enabled=True, LogGroup=Ref(log_group)))
5443
else:
55-
if not hasattr(
56-
cluster.cfn_resource.LoggingInfo.BrokerLogs, "CloudWatchLogs"
57-
):
44+
if not hasattr(cluster.cfn_resource, "LoggingInfo"):
5845
setattr(
59-
cluster.cfn_resource.LoggingInfo,
60-
"CloudWatchLogs",
61-
CloudWatchLogs(Enabled=True, LogGroup=Ref(log_group)),
46+
cluster.cfn_resource,
47+
"LoggingInfo",
48+
LoggingInfo(
49+
BrokerLogs=BrokerLogs(
50+
CloudWatchLogs=CloudWatchLogs(
51+
Enabled=True, LogGroup=Ref(log_group)
52+
)
53+
)
54+
),
6255
)
56+
else:
57+
if not hasattr(
58+
cluster.cfn_resource.LoggingInfo.BrokerLogs, "CloudWatchLogs"
59+
):
60+
setattr(
61+
cluster.cfn_resource.LoggingInfo,
62+
"CloudWatchLogs",
63+
CloudWatchLogs(Enabled=True, LogGroup=Ref(log_group)),
64+
)
6365

6466

6567
def handle_logging(

ecs_composex_msk_cluster/msk_sg_ingress.py

Lines changed: 16 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -197,15 +197,17 @@ def handle_msk_auth_parameters(cluster: MskCluster) -> None:
197197
),
198198
]
199199
for auth_property in properties:
200-
resource, prop, value = get_nested_property(
200+
res_props_to_update = get_nested_property(
201201
cluster.cfn_resource,
202202
auth_property[0],
203203
)
204-
if prop and value:
205-
cluster.stack.Parameters.update(
206-
{auth_property[2].title: str(auth_property[1](value))}
207-
)
208-
setattr(resource, prop, Ref(auth_property[2]))
204+
for _prop_to_update in res_props_to_update:
205+
resource, prop, value = _prop_to_update
206+
if prop and value:
207+
cluster.stack.Parameters.update(
208+
{auth_property[2].title: str(auth_property[1](value))}
209+
)
210+
setattr(resource, prop, Ref(auth_property[2]))
209211

210212

211213
def handle_msk_auth_settings(cluster: MskCluster):
@@ -214,10 +216,14 @@ def handle_msk_auth_settings(cluster: MskCluster):
214216
"SecurityGroupIngress",
215217
allow_clients_sg_ingress(cluster),
216218
)
217-
resource, prop, value = get_nested_property(
219+
resource_props_to_update = get_nested_property(
218220
cluster.cfn_resource,
219221
"BrokerNodeGroupInfo.ConnectivityInfo.PublicAccess.Type",
220222
)
221-
if value and value == "SERVICE_PROVIDED_EIPS":
222-
cluster.stack.Parameters.update({MSK_CLUSTER_ADDRESSING_TYPE.title: "PUBLIC"})
223-
handle_msk_auth_parameters(cluster)
223+
for _to_update in resource_props_to_update:
224+
resource, prop, value = _to_update
225+
if value and value == "SERVICE_PROVIDED_EIPS":
226+
cluster.stack.Parameters.update(
227+
{MSK_CLUSTER_ADDRESSING_TYPE.title: "PUBLIC"}
228+
)
229+
handle_msk_auth_parameters(cluster)

0 commit comments

Comments
 (0)