#! /usr/bin/env python

import argparse
import logging

from okra.logging_utils import enable_log, enable_cloud_log
from okra.worker import redis_worker, redis_loader

logger = logging.getLogger(__name__)

parser = argparse.ArgumentParser()
parser.add_argument("--redis", help="job name for redis loader/worker")
parser.add_argument("--load_redis", help="file prefix to load redis queue")
parser.add_argument("--logfile", help="filepath to log file")

args = parser.parse_args()

# Configure logging

if args.logfile:
    enable_log(args.logfile)

else:
    enable_cloud_log()

# Configure redis job

if args.load_redis:
    logger.info("STARTED redis loader")
    redis_loader(job=args.redis, prefix=args.load_redis)
    logger.info("FINISHED redis loader")

elif args.redis:
    logger.info("STARTED redis worker")
    redis_worker(job=args.redis)
    logger.info("FINISHED redis worker")

# Return help menu otherwise

else:
    parser.print_help()
    
