"""
Usage:
        binlog_server.py --user=<username> --password=<password> --host=<remote_host> --port=<remote_port> --backup-dir=<backup_dir> --log=<log> [--last-file=<last-file>]
        binlog_server.py -h | --help
        binlog_server.py --version
        binlog_server.py --config=<config_file> --dbname=<database_name> [--last-file=<last-file>]
Options:
        -h --help                      Show help information.
        --version                      Show version.
        --user=<username>              The user name used to connect to the remote server.
        --password=<password>          The password used to connect to the remote server.
        --host=<remote_host>            The remote host IP address.
        --port=<remote_port>            The remote MySQL server port.
        --backup-dir=<backup_dir>      The dest to store binlog.
        --log=<log>                    The log.
        --last-file=<last-file>        Specify the starting binlog.
        --config=<config_file>          Config file.
        --dbname=<database_name>        Section name in config file.
"""
from docopt import docopt
import subprocess
import logging
import time
import ConfigParser
import os
arguments = docopt(__doc__, version='Binlog server 1.0')
if arguments['--config']:
    cf=ConfigParser.ConfigParser()
    cf.read(arguments['--config'])
    section_name = arguments['--dbname']
    db_host = cf.get(section_name, "db_host")
    db_port = cf.get(section_name, "db_port")
    db_user = cf.get(section_name, "db_user")
    db_passwd = cf.get(section_name, "db_passwd")
    backup_dir = cf.get(section_name, "backup_dir")
    log = cf.get(section_name, "log")
    logging.basicConfig(level=logging.DEBUG,
                format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                datefmt='%a, %d %b %Y %H:%M:%S',
                filename=log,
                filemode='a')
logging.basicConfig(level=logging.DEBUG,
                format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                datefmt='%a, %d %b %Y %H:%M:%S',
                filename=arguments['--log'],
                filemode='a')
def dumpBinlog(user,password,host,port,backup_dir,log,last_file=''):
        LOCAL_BACKUP_DIR=backup_dir
        if backup_dir[-1]!= '/':
            os.exit()
        #BACKUP_LOG='/data4/binlog_backup/120.27.136.247/BB.log'
        BACKUP_LOG=log[log.rfind('/')+1:]
        while True:
            if not last_file:
                    cmd="ls -A {LOCAL_BACKUP_DIR} | grep -v {BACKUP_LOG} | grep -v nohup.out |wc -l".format(LOCAL_BACKUP_DIR=LOCAL_BACKUP_DIR,BACKUP_LOG=BACKUP_LOG)
                    child=subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)
                    child.wait()
                    wc_l=int(child.communicate()[0].strip())
                    if wc_l != 0:
                            cmd="ls -l %s | grep -v %s | grep -v nohup.out |tail -n 1 |awk '{print $9}'" % (LOCAL_BACKUP_DIR,BACKUP_LOG)
                            child=subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)
                            child.wait()
                            LAST_FILE=child.communicate()[0].strip()
            else:
                    LAST_FILE=last_file
            logging.info('Last File is %s' % (LAST_FILE))
            mysqlbinlog='mysqlbinlog --raw --read-from-remote-server --stop-never --host={REMOTE_HOST} --port={REMOTE_PORT} --user={REMOTE_USER} --password={REMOTE_PASS} --result-file={RESULT_FILE} {LAST_FILE}'.format(REMOTE_HOST=host,REMOTE_PORT=port,REMOTE_USER=user,REMOTE_PASS=password,RESULT_FILE=LOCAL_BACKUP_DIR,LAST_FILE=LAST_FILE)
            subprocess.call(mysqlbinlog,shell=True)
            logging.info('Binlog server stop!!!,reconnect after 10 seconds')
            time.sleep(10)
if __name__ == '__main__':
    if arguments['--config']:
        lock_file=db_host+"_binlog_server.lock"
    else:
        lock_file=arguments['--host']+"_binlog_server.lock"
    child=subprocess.Popen('ls /tmp|grep %s' % (lock_file),shell=True,stdout=subprocess.PIPE)
    child.wait()
    lock=child.communicate()[0].strip()
    if not lock:
        subprocess.call('touch /tmp/%s' % (lock_file),shell=True)
        logging.info('Get lock,Binlog server start!!!')
        if not arguments['--config']:
          dumpBinlog(arguments['--user'],arguments['--password'],arguments['--host'],arguments['--port'],arguments['--backup-dir'],arguments['--log'],arguments['--last-file'])
        else:
          dumpBinlog(db_user,db_passwd,db_host,db_port,backup_dir,log,arguments['--last-file'])
    else:
        logging.info('Binlog server already running!!!')
        print('Binlog server already running!!!,please check or reomove the lock file')