DdZddlmZddlmZddlmZdZdZdZdZy ) z'Utilities for unit operations commands.)absolute_import)division)unicode_literalsc|jds|S|jds|jdr|Si|j_|S)aRequest hook to handle use of --provision flag. Args: unused_ref: A resource ref to the parsed resource. args: Parsed args namespace containing the flags. request: The request message to be modified. Returns: The modified request message. provisionprovision_releaseprovision_input_variables) IsSpecified unitOperationr unused_refargsrequests >lib/googlecloudsdk/command_lib/saas_runtime/unit_operations.pyHandleEmptyProvisionrsP   + & N )*d.>.>!/ N$&'! .c|jds|S|jds|jdr|Si|j_|S)aRequest hook to handle use of --upgrade flag. Args: unused_ref: A resource ref to the parsed resource. args: Parsed args namespace containing the flags. request: The request message to be modified. Returns: The modified request message. upgradeupgrade_releaseupgrade_input_variables)r r rr s rHandleEmptyUpgrader+sP   ) $ N '(D,<,<- N"$' .rcN|jds|Si|j_|S)a Request hook to handle use of --deprovision flag. Args: unused_ref: A resource ref to the parsed resource. args: Parsed args namespace containing the flags. request: The request message to be modified. Returns: The modified request message. deprovision)r r rr s rHandleEmptyDeprovisionr@s*   - ( N&('# .rcVgd}gd}dg}tfd|Dtfd|Dtfd|Dd}|jD]M\}}|r t|j|s t |j|7t |j|dO|S) aXRequest hook to handle updates to the operation type. The declarative framework does not fully support one_of fields in updates. For example, if the command 'saas-runtime unit-operations update --provision' is run, the request will have an empty upgrade field but that field should not be present at all. This hook will delete the unspecified one_of field from the request. Args: unused_ref: A resource ref to the parsed resource. args: Parsed args namespace containing the flags. request: The request message to be modified. Returns: The modified request message. )add_provision_input_variablesclear_provision_input_variablesclear_provision_releaserr r remove_provision_input_variables)add_upgrade_input_variablesclear_upgrade_input_variablesclear_upgrade_releaserrrremove_upgrade_input_variablesrc3@K|]}j|ywNr .0flagrs r z1HandleOneOfOperationTypeUpdate..wsJ/$t''-/c3@K|]}j|ywr%r&r's rr*z1HandleOneOfOperationTypeUpdate..xsF T%%d+ r+c3@K|]}j|ywr%r&r's rr*z1HandleOneOfOperationTypeUpdate..ysNr=s'.&'**"2r