#!/usr/bin/python
#-*- coding:UTF-8 -*-#@author Jane.Hoo
#@date2016/11/29
from__future__ import division
import os
import MySQLdb
import time
import commands
import logging
import re
import math
print'*******************************'timestamp=time.strftime("%Y%m%d%H%M%S", time.localtime())
logfile='/tmp/myloaddataout.log%s'%timestamp
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='%s'%logfile,
filemode='w')
print'导出日志记录在:%s'%logfile
#################################################################################################
#定义一个StreamHandler,将INFO级别或更高的日志信息打印到标准错误,并将其添加到当前的日志处理对象#
console=logging.StreamHandler()
console.setLevel(logging.INFO)
formatter= logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
console.setFormatter(formatter)
logging.getLogger('').addHandler(console)
#################################################################################################
#python/cygdrive/c/Users/Jane.Hoo/PycharmProjects/loaddata/MySQLloaddataout.py
db_host='127.0.0.1'db_user='test'db_pwd='test'db='test'db_port=int('3306')classToolloadout:
def __init__(self,pathfile=0):
self.pathfile=pathfile
#得到数据库连接
def getdbconn(self,DB_HOST=db_host,DB_USER='routeload',DB_PWD='routeload',DB='test01',DB_PORT='3306'):
logging.debug('%s:%s:%s:%s:%s'%(DB_HOST,DB_USER,DB_PWD,DB,DB_PORT))
conn='unlink'
try:
conn=MySQLdb.connect(host=DB_HOST,user=DB_USER,passwd=DB_PWD,port=int(DB_PORT),db=DB)
logging.info('数据库连接成功')
except MySQLdb.OperationalError,e:
logging.warning('数据库连接失败!%s'%e)returnconn
#关闭数据库连接
def closedbconn(self,conn=0):
logging.debug('闭数据库连接')
conn.close()
#检查导出文件
def chekpathfile(self):
pathfile=self.pathfile
check_flag=0iffile_exists=os.path.exists(pathfile)if iffile_exists==1:
check_flag=1logging.info('导出文件已存在!')else:
path=os.path.split(pathfile)[0]
ifpath_exists=os.path.isdir(path)if ifpath_exists==1:
logging.info('%s是有效的路径'%path)else:
check_flag=1logging.info('%s是无效的路径'%path)returncheck_flagif __name__=='__main__':
print'准备导出...'pathfile=raw_input("请输入导出文件路径:").strip()
cmd_sql=raw_input('请输入要查询的语句:').strip()
c=Toolloadout(pathfile)
pf_check=c.chekpathfile()if pf_check==0:
logging.info('文件路径校验通过')
ifinputdbmsg=raw_input('是否需要自定义数据库连接信息(Y|N)?').strip()if ifinputdbmsg=='Y':
db_host=raw_input('请输入数据地址:').strip()
db_user=raw_input('请输入用户名:').strip()
db_pwd=raw_input('请输入密码:').strip()
db=raw_input('请输入数据库名:').strip()
db_port=raw_input('请输入数据库端口:').strip()if db_host=='' or db_user=='' or db_pwd=='' or db=='':
logging.info('自定义数据库输入信息有为空.')
check_flag=1if_port=re.match(r"[0-9]", db_port)ifif_port:
db_port=int(db_port)else:
db_port=int('3306')
logging.info('端口格式输入有误,将使用默认端口%s'%db_port)
conn=c.getdbconn(db_host,db_user,db_pwd,db,db_port)if str(conn)!='unlink':
selectsql="mysql -A %s -h %s -u%s -p%s -P%s -ss -e '%s;' | sed 's/\\t/,/g;s/^//;s/$//;s/\\n//g' >%s"%(db,db_host,db_user,db_pwd,db_port,cmd_sql,pathfile)
print'querysql:',selectsqltry:
os.system('%s'%selectsql)
except BaseException,e:
logging.info('导出数据过程中报错!%s'%e)
c.closedbconn(conn)else:
logging.info('失败')else:
logging.info('文件路径校验不通过,导出结束')