Skip to content

Commit efc7e31

Browse files
committed
refactor: tidy up update_blue_green_upgrade_settings function
1 parent 91e9939 commit efc7e31

1 file changed

Lines changed: 19 additions & 14 deletions

File tree

src/aks-preview/azext_aks_preview/agentpool_decorator.py

Lines changed: 19 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2050,25 +2050,30 @@ def update_blue_green_upgrade_settings(self, agentpool: AgentPool) -> AgentPool:
20502050
final_soak_duration = self.context.get_final_soak_duration()
20512051
drain_batch_size = self.context.get_drain_batch_size()
20522052

2053-
if any([drain_timeout_bg, batch_soak_duration, final_soak_duration, drain_batch_size]):
2054-
# Initialize blue-green upgrade settings with existing values as defaults
2055-
upgrade_settings_blue_green = agentpool.upgrade_settings_blue_green
2056-
if upgrade_settings_blue_green is None:
2057-
upgrade_settings_blue_green = self.models.AgentPoolBlueGreenUpgradeSettings() # pylint: disable=no-member
2053+
should_mutate = False
2054+
# Initialize blue-green upgrade settings with existing values as defaults
2055+
upgrade_settings_blue_green = agentpool.upgrade_settings_blue_green
2056+
if upgrade_settings_blue_green is None:
2057+
upgrade_settings_blue_green = self.models.AgentPoolBlueGreenUpgradeSettings() # pylint: disable=no-member
20582058

2059-
if drain_batch_size is not None:
2060-
upgrade_settings_blue_green.drain_batch_size = drain_batch_size
2059+
if drain_batch_size is not None:
2060+
upgrade_settings_blue_green.drain_batch_size = drain_batch_size
2061+
should_mutate = True
20612062

2062-
if drain_timeout_bg is not None:
2063-
upgrade_settings_blue_green.drain_timeout_in_minutes = drain_timeout_bg
2063+
if drain_timeout_bg is not None:
2064+
upgrade_settings_blue_green.drain_timeout_in_minutes = drain_timeout_bg
2065+
should_mutate = True
20642066

2065-
if batch_soak_duration is not None:
2066-
upgrade_settings_blue_green.batch_soak_duration_in_minutes = batch_soak_duration
2067+
if batch_soak_duration is not None:
2068+
upgrade_settings_blue_green.batch_soak_duration_in_minutes = batch_soak_duration
2069+
should_mutate = True
20672070

2068-
if final_soak_duration is not None:
2069-
upgrade_settings_blue_green.final_soak_duration_in_minutes = final_soak_duration
2071+
if final_soak_duration is not None:
2072+
upgrade_settings_blue_green.final_soak_duration_in_minutes = final_soak_duration
2073+
should_mutate = True
20702074

2071-
# Set the blue-green upgrade settings as a separate property on agentpool
2075+
# Set the blue-green upgrade settings as a separate property on agentpool
2076+
if should_mutate:
20722077
agentpool.upgrade_settings_blue_green = upgrade_settings_blue_green
20732078

20742079
return agentpool

0 commit comments

Comments
 (0)