#!/usr/bin/python
import yaml

def __run__(params):
    REDIS_HOST       = params.get('host','localhost')
    REDIS_PORT       = int(params.get('port',6379))
    REDIS_DB         = int(params.get('db',0))
    try:
        import redis
    except ImportError:
        return [1, "WARNING: Need to install redis python library."]

    try:
        r = redis.Redis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB)
        if r.set('foo', 'bar'):
            r.delete('foo')
            return [0, "OK - OK: Redis writing"]

        else:
            return [2, "CRITICAL - Error: %s" % repr(e)]


    except Exception, e:
        return [2, "CRITICAL - Error: %s" % repr(e)]