#!/usr/bin/env python3
# Copyright (c) 2019 Arista Networks, Inc.  All rights reserved.
# Arista Networks, Inc. Confidential and Proprietary.
'''
Upgrades power supply firmware.
'''
import PowerFirmware
import sys
from time import asctime

if __name__ == '__main__':
   try:
      args = sys.argv[ 1 : ]
      force = '--force' in args
      if force:
         args.remove( '--force' )
      slot = int( args[ 0 ] ) if len( args ) >= 1 else None
      firmwareFileName = args[ 1 ] if len( args ) >= 2 else None
      PowerFirmware.upgradeFirmware( slot, firmwareFileName, force )
   except: # pylint: disable-msg=W0702
      print( asctime(), "Firmware upgrade failed", file=sys.stderr )
