old_contents: # =============================================================================
  # cmd_capture
  #
  # Copyright (c)  2016, Cisco Systems
  # All rights reserved.
  #
  # # Author: Klaudiusz Staniek
  #
  # Redistribution and use in source and binary forms, with or without
  # modification, are permitted provided that the following conditions are met:
  #
  # Redistributions of source code must retain the above copyright notice,
  # this list of conditions and the following disclaimer.
  # Redistributions in binary form must reproduce the above copyright notice,
  # this list of conditions and the following disclaimer in the documentation
  # and/or other materials provided with the distribution.
  # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  # THE POSSIBILITY OF SUCH DAMAGE.
  # =============================================================================
  
  
  from horizon.plugin import Plugin
  
  
  class CmdCapturePlugin(Plugin):
      """
      Pre-upgrade check
      This plugin checks and record active packages
      """
      @staticmethod
      def start(manager, device, *args, **kwargs):
          command_list, timestamp = manager.load_data("custom-commands")
          if command_list:
              for cmd in command_list:
                  output = device.send(cmd, timeout=2200)
                  file_name = manager.file_name_from_cmd(cmd)
                  full_name = manager.save_to_file(file_name, output)
                  if full_name:
                      manager.save_data(cmd, full_name)
                  manager.log("Output of '{}' command saved to {}".format(cmd, file_name))
          else:
              manager.log("No custom commands provided")
new_contents: # =============================================================================
  # cmd_capture
  #
  # Copyright (c)  2016, Cisco Systems
  # All rights reserved.
  #
  # # Author: Klaudiusz Staniek
  #
  # Redistribution and use in source and binary forms, with or without
  # modification, are permitted provided that the following conditions are met:
  #
  # Redistributions of source code must retain the above copyright notice,
  # this list of conditions and the following disclaimer.
  # Redistributions in binary form must reproduce the above copyright notice,
  # this list of conditions and the following disclaimer in the documentation
  # and/or other materials provided with the distribution.
  # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  # THE POSSIBILITY OF SUCH DAMAGE.
  # =============================================================================
  
  
  from horizon.plugin import Plugin
  
  
  class CmdCapturePlugin(Plugin):
      """
      Pre-upgrade check
      This plugin checks and record active packages
      """
      @staticmethod
      def start(manager, device, *args, **kwargs):
          #command_list, timestamp = manager.load_data("custom-commands")
          command_list = manager.csm.custom_commands
          print command_list
          if command_list:
              for cmd in command_list:
                  output = device.send(cmd, timeout=2200)
                  file_name = manager.file_name_from_cmd(cmd)
                  full_name = manager.save_to_file(file_name, output)
                  if full_name:
                      manager.save_data(cmd, full_name)
                  manager.log("Output of '{}' command saved to {}".format(cmd, file_name))
          else:
              manager.log("No custom commands provided")
subject: Change the custom_commands to context property