import MySQLdb

DB_HOST = ""
DB_USER = ""
DB_PASS = ""
DB_NAME = ""

db = MySQLdb.connect(DB_HOST, DB_USER, DB_PASS, DB_NAME)


def connect():
    cursor = db.cursor()
    cursor.execute("SELECT VERSION()")
    data = cursor.fetchone()

    if not data:
        return 0
    elif data:
        return 1
    else:
        return "database null"


def read():
    if connect() == 1:
        print "OK"
    else:
        print "NO"




read()
db.close()