#!/opt/vmware/bin/python

import logging
import os
import platform
import sys

if platform.system() == 'Linux':
   sys.path.append('/usr/lib/vmware/site-packages')
   logDir = '/var/log/restore'
   pattern = '/usr/lib/restore/*'
else:
   sys.path.append(os.path.join(os.environ['VMWARE_CIS_HOME'], 'python-modules'))
   logDir = os.path.join(os.environ['VMWARE_LOG_DIR'], 'restore')
   pattern = os.path.join(os.environ['VMWARE_CIS_HOME'], 'visl-integration',
                          'usr', 'lib', 'restore', '*')

import backupUtils

backupUtils.initLogging(logDir, 'restore.log')

if backupUtils.isMarkerSet():
   timeout = 300
   isRestorable = True
   try:
      # Note: For infra nodes, there is no vchaLib. The import
      # is needed only for non-infrastructure nodes.
      import vchaLib
      isRestorable = vchaLib.isRestorable()
   except:
      pass
   if isRestorable:
      backupUtils.runScripts(logDir, pattern, timeout, 'restore')
   else:
      logging.info('Skipping restore as backup marker was removed.')
   backupUtils.removeBackupMarker()
else:
   logging.info('Skipping restore as backup marker is not set')
