#!/usr/bin/env python
# encoding: utf-8
"""
@version: ??
@author: phpergao
@license: Apache Licence
@file: mssql.py
@time: 2016/7/18 10:18
"""
import pymssql,os
class MSSQL:
"""
对pymssql的简单封装
pymssql库,该库到这里下载:http://www.lfd.uci.edu/~gohlke/pythonlibs/#pymssql
使用该库时,需要在Sql Server Configuration Manager里面将TCP/IP协议开启
用法:
"""
def __init__(self,host,user,pwd,db):
self.host = host
self.user = user
self.pwd = pwd
self.db = db
def __GetConnect(self):
"""
得到连接信息
返回: conn.cursor()
"""
if not self.db:
raise(NameError,"没有设置数据库信息")
self.conn = pymssql.connect(host=self.host,user=self.user,password=self.pwd,database=self.db,charset="utf8")
cur = self.conn.cursor()
if not cur:
raise(NameError,"连接数据库失败")
else:
return cur
def ExecQuery(self,sql):
"""
执行查询语句
返回的是一个包含tuple的list,list的元素是记录行,tuple的元素是每行记录的字段
调用示例:
ms = MSSQL(host="localhost",user="sa",pwd="123456",db="PythonWeiboStatistics")
resList = ms.ExecQuery("SELECT id,NickName FROM WeiBoUser")
for (id,NickName) in resList:
print str(id),NickName
"""
cur = self.__GetConnect()
cur.execute(sql)
resList = cur.fetchall()
# 完毕后必须关闭连接
self.conn.close()
return resList
def ExecNonQuery(self,sql):
"""
执行非查询语句
调用示例:
cur = self.__GetConnect()
cur.execute(sql)
self.conn.commit()
self.conn.close()
"""
cur = self.__GetConnect()
cur.execute(sql)
self.conn.commit()
self.conn.close()
def select(hostname,ip,username,password,dbname,cmd):
if password==None:
password=''
cha = MSSQL(host=ip, user=username, pwd=password, db=dbname)
resList = cha.ExecQuery(cmd)
# print(resList)
for i in resList:
print(i)
def not_select(hostname,ip,username,password,dbname,cmd):
if password == None:
password = ''
ms = MSSQL(host=ip, user=username, pwd=password, db=dbname)
ms.ExecNonQuery(cmd)
print("主机{}执行:{}.成功".format(hostname,cmd))
def main(cmd):
## ms = MSSQL(host="localhost",user="sa",pwd="123456",db="PythonWeiboStatistics")
## #返回的是一个包含tuple的list,list的元素是记录行,tuple的元素是每行记录的字段
## ms.ExecNonQuery("insert into WeiBoUser values('2','3')")
ms = MSSQL(host="127.0.0.1",user="sa",pwd="123456",db="test")
resList = ms.ExecQuery("SELECT * FROM dbo.db_host")#从本地数据库获取到要操作的远程数据库信息
for i in resList:
#print (i)
ip_add=os.popen("ping -n 1 {}".format(i[1])).read().strip()#如果ping不通,就不操作了。
if 'TTL'not in ip_add:
continue
if "SELECT" in cmd or 'select' in cmd:
select(i[0],i[1],i[2],i[3],i[4],cmd)
print("主机{}执行查询命令的结果:".format(i[0]))
else:
not_select(i[0],i[1],i[2],i[3],i[4],cmd)
#dele =ms.ExecNonQuery("DELETE FROM waimai4.dbo.baidu_rueren")
#update=ms.ExecNonQuery("UPDATE dbo.GOODS SET CLASSID='19' WHERE GOODSNAME LIKE'%牛肉%'")
if __name__ == '__main__':
cmd=input("请输入SQL语句:").strip()
main(cmd)