# -*- coding: utf-8 -*-
#!/usr/bin/python

import os
import sys
import time
import json
import redis
import pymysql
import datetime
import pprint
from string import Template
from optparse import OptionParser

import _utils

global_redis_conf = _utils.getGlobalRedisConf()
mysql_conf = _utils.getMysqlConf()

optparser = OptionParser(usage="%prog [options]")
optparser.add_option("-d",
                  "--db",
                  dest = "db",
                  help = "guns,global,serverid1,serverid2,serverid8-serveid20")
(options, args) = optparser.parse_args()
_utils.checkDBOption(options.db)
_utils.printDBOption(options.db)
dbs = options.db.split(',')

def _importGunsDB():
    _utils.importMysql(mysql_conf, 'guns', 'guns.sql')

def _importGlobalRedis():
    _utils.importRedis(global_redis_conf, 'global redis', 'global.rdb')

def _importGameDBAndRedis(server_id):
    conf = _utils.getGameConf(server_id)
    _utils.importMysql(conf, 'legend_' + str(server_id), 'legend_%d.sql' % (server_id))
    _utils.importMysql(conf, 'logger_' + str(server_id), 'logger_%d.sql' % (server_id))
    _utils.importRedis(conf, 'game redis', 'dump_%d.rdb' % (server_id))

for dbname in dbs:
    if dbname == 'guns':
        _importGunsDB()
    elif dbname == 'global':
        _importGlobalRedis()
    else:
        _importGameDBAndRedis(int(dbname))