Configure Maintenance Mode During Upgrades

(computer_id, duration, api, configuration, api_version, api_exception):
    """ Turns on maintenance mode on a computer.
    :param computer_id: The ID of the computer.
    :param duration: The maintenance mode duration.
    :param api: The Deep Security API modules.
    :param configuration: Configuration object to pass to the api client.
    :param api_version: The version of the API to use.
    :param api_exception: The Deep Security API exception module.
    :return: A Computer object.
    """

    # Create and configure an ApplicationControlComputerExtesnion object
    application_control = api.ApplicationControlComputerExtension()
    application_control.maintenance_mode_status = "on"
    application_control.maintenance_mode_duration = duration

    # Add the ApplicationControlComputerExtension to a Computer object
    computer = api.Computer()
    computer.application_control = application_control

    try:
        # Update the computer
        computers_api = api.ComputersApi(api.ApiClient(configuration))
        return computers_api.modify_computer(computer_id, computer, api_version)

    except api_exception as e:
        return "Exception: " + str(e)

/*
 * Turn on maintenance mode on a computer.
 * @param {Number} computerID The ID of the computer.
 * @param {Number} duration The maintenance mode duration in milliseconds.
 * @param {ApiClient} api The Deep Security API exports.
 * @param {String} apiVersion The api version to use.
 * @returns {Promise} A promise object that resolves to a Computer object. 
 */
exports.turnOnMaintenanceMode = function (computerID, duration, api, apiVersion) {
  //  Create and configure an ApplicationControlComputerExtension
  const applicationControl = new api.ApplicationControlComputerExtension();
  applicationControl.maintenanceModeStatus = api.ApplicationControlComputerExtension.MaintenanceModeStatusEnum.on;
  applicationControl.maintenanceModeDuration = duration;

  //  Add the ApplicationControlComputerExtension to a computer
  const computer = new api.Computer();
  computer.applicationControl = applicationControl;

  //  Update the computer
  const computersApi = new api.ComputersApi();
  return computersApi.modifyComputer(computerID, computer, apiVersion);
};

 /*
 * Turn on maintenance mode on a computer.
 * @param computerId The ID number of the computer.
 * @param duration The duration of the maintenance mode in milliseconds.
 * @param apiVersion The api version.
 */
public static Computer turnOnMaintenanceMode(Integer computerId, Integer duration, String apiVersion) {
    //  Create and configure an ApplicationControlComputerExtension
    ApplicationControlComputerExtension applicationControl = new ApplicationControlComputerExtension();
    applicationControl.setMaintenanceModeStatus(MaintenanceModeStatusEnum.ON);
    applicationControl.setMaintenanceModeDuration(duration);

    //  Create a computer object and add the ApplicationControlComputerExtension
    Computer computer = new Computer();
    computer.setApplicationControl(applicationControl);

    //  Update the computer
    ComputersApi computersApi = new ComputersApi();
    try {
        return computersApi.modifyComputer(computerId, computer, Boolean.FALSE, apiVersion);
    } catch (ApiException e) {
        e.printStackTrace();
    }
    return null;
}