mirror of https://github.com/midoks/mdserver-web
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.6 KiB
91 lines
2.6 KiB
# coding: utf-8
|
|
|
|
import re
|
|
import os
|
|
import sys
|
|
|
|
sys.path.append("/usr/local/lib/python2.7/site-packages")
|
|
|
|
|
|
class mysql:
|
|
__DB_PASS = None
|
|
__DB_USER = 'root'
|
|
__DB_PORT = 3306
|
|
__DB_HOST = 'localhost'
|
|
__DB_NAME = 'test'
|
|
__DB_CONN = None
|
|
__DB_CUR = None
|
|
__DB_ERR = None
|
|
__DB_CNF = '/etc/my.cnf'
|
|
|
|
def __Conn(self):
|
|
'''连接MYSQL数据库'''
|
|
try:
|
|
socket = '/tmp/mysql.sock'
|
|
try:
|
|
import MySQLdb
|
|
except Exception, ex:
|
|
self.__DB_ERR = ex
|
|
return False
|
|
try:
|
|
self.__DB_CONN = MySQLdb.connect(host=self.__DB_HOST, user=self.__DB_USER, passwd=self.__DB_PASS,
|
|
port=self.__DB_PORT, db=self.__DB_NAME, charset="utf8", connect_timeout=10, unix_socket=socket)
|
|
except MySQLdb.Error, e:
|
|
self.__DB_HOST = '127.0.0.1'
|
|
self.__DB_CONN = MySQLdb.connect(host=self.__DB_HOST, user=self.__DB_USER, passwd=self.__DB_PASS,
|
|
port=self.__DB_PORT, db=self.__DB_NAME, charset="utf8", connect_timeout=10, unix_socket=socket)
|
|
self.__DB_CUR = self.__DB_CONN.cursor()
|
|
return True
|
|
|
|
except MySQLdb.Error, e:
|
|
self.__DB_ERR = e
|
|
return False
|
|
|
|
def setHost(self, host):
|
|
self.__DB_HOST = host
|
|
|
|
def setPwd(self, pwd):
|
|
self.__DB_PASS = pwd
|
|
|
|
def setUser(self, user):
|
|
self.__DB_USER = user
|
|
|
|
def setPort(self, port):
|
|
self.__DB_PORT = port
|
|
|
|
def setDb(self, name):
|
|
self.__DB_NAME = name
|
|
|
|
def getPwd(self):
|
|
return self.__DB_PASS
|
|
|
|
def execute(self, sql):
|
|
# 执行SQL语句返回受影响行
|
|
if not self.__Conn():
|
|
return self.__DB_ERR
|
|
try:
|
|
result = self.__DB_CUR.execute(sql)
|
|
self.__DB_CONN.commit()
|
|
self.__Close()
|
|
return result
|
|
except Exception, ex:
|
|
return ex
|
|
|
|
def query(self, sql):
|
|
# 执行SQL语句返回数据集
|
|
if not self.__Conn():
|
|
return self.__DB_ERR
|
|
try:
|
|
self.__DB_CUR.execute(sql)
|
|
result = self.__DB_CUR.fetchall()
|
|
# 将元组转换成列表
|
|
data = map(list, result)
|
|
self.__Close()
|
|
return data
|
|
except Exception, ex:
|
|
return ex
|
|
|
|
# 关闭连接
|
|
def __Close(self):
|
|
self.__DB_CUR.close()
|
|
self.__DB_CONN.close()
|
|
|