import cx_Oracle
import os
import datetime
from time import localtime, strftime, strptime
import ConfigParser

# check-oracle-asm-disk-missing.py
# 
# to cx_Oracle work, you need:
# 1. set LIBRARY_PATH correctly
# 2. configure yaml file
# 3. echo /u01/app/oracle/product/11.2.0.3/db/lib > /etc/ld.so.conf.d/oracle.conf 
#  ldconfig


def __run__(params):
  conn = None
  cf = ConfigParser.ConfigParser()
  config_file = params.get('config_file')
  cf.readfp(open(config_file))
  user = cf.get('asm', 'user')
  password = cf.get('asm', 'passwd')

  lock_critical = cf.getint('oracle', 'lock_critical')
  lock_warning = cf.getint('oracle', 'lock_warning')
  instances = cf.get('asm','instances')

  oracle_home = cf.get('asm', 'oracle_home')
  os.environ['ORACLE_HOME'] = oracle_home
  try:
    instances = instances.split();
    check_message = ''
    nerros = 0
    for instance in instances:
      os.environ['ORACLE_SID'] = instance
      conn = cx_Oracle.connect(user, password, mode=cx_Oracle.SYSDBA)
      cursor = conn.cursor()
      sql = """select disk_number, group_number, mount_status, state from v$asm_disk where mount_status = 'MISSING' or state in ('DROPPED', 'FORCING')"""
      cursor.execute(sql)
      result = cursor.fetchall()
      rowcount = cursor.rowcount
      cursor.close()
      if rowcount <> 0:
        for row in result:
          dskn = row[0]
          grpn = row[1]
          msta = row[2]
          stat = row[3]
          check_message = check_message + '(!!) Asm disks error found on database : ' + instance + ' : DskNumber: %s - GrpNumber: %s - MountStatus: %s - State: %s ' % (dskn, grpn, msta, stat)
          nerros += 1
      else:
        check_message = check_message + ' ' + 'No asm disks error found on database : ' + instance
    if nerros <> 0:
      return [2, str(check_message) ]
    else:
      return [0, str(check_message) ]
  except Exception, e:
    return [2, '{0}'.format(str(e))]