#!/usr/bin/env python3
# Copyright (c) 2012 Arista Networks, Inc.  All rights reserved.
# Arista Networks, Inc. Confidential and Proprietary.

from PowerDiagLib import OptionParser
import PyClient, Tac

def main():
   parser = OptionParser()
   parser.add_option( "", "--powerSupply", action="store",
                      help="The power supply to test " )
   parser.add_option( "", "--dumpRegistersAlways", action="store_true",
                      help="Let log always dump registers " )
   parser.add_option( "", "--dumpRegistersOnEvents", action="store_true",
                      help="Let log only dump registers on events" )
   ( options, args ) = parser.parse_args()

   if args:
      parser.error( "unexpected arguments" )
   if not options.powerSupply:
      parser.error( "you must specify a power supply to test" )


         
   root = PyClient.PyClient( sysname='ar', agent='Pmbus' ).agentRoot()
   driver = root['Pmbus'].reactor.powerSupplyDriver[ int( options.powerSupply ) ]
   if options.dumpRegistersAlways and options.dumpRegistersOnEvents:
      print( "Shoudl only choose one of 'dumpRegistersAlways' and "
             "'dumpRegistersOnEvents'" )

   if options.dumpRegistersAlways:
      print( "Dump registers always set to true for PS#", options.powerSupply )
      driver.dumpRegistersAlways = True
   elif options.dumpRegistersOnEvents:
      print( "Dump registers always set to false for PS#", options.powerSupply )
      driver.dumpRegistersAlways = False
   else:
      print( "Did not do anything" )

   print( "DONE!" )

if __name__ == "__main__":
   main()

