import pymssql

class MSSQL:
    """
    對pymssql的簡單封裝
    pymssql庫，該庫到這里下載：http://www.lfd.uci.edu/~gohlke/pythonlibs/#pymssql
    使用該庫時(shí)，需要在Sql Server Configuration Manager里面將TCP/IP協(xié)議開啟

    用法：

    """

    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,"沒有設(shè)置數(shù)據(jù)庫信息")
        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,"連接數(shù)據(jù)庫失敗")
        else:
            return cur

    def ExecQuery(self,sql):
        """
        執(zhí)行查詢語句
        返回的是一個(gè)包含tuple的list，list的元素是記錄行，tuple的元素是每行記錄的字段

        調(diào)用示例：
                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()

        #查詢完畢后必須關(guān)閉連接
        self.conn.close()
        return resList

    def ExecNonQuery(self,sql):
        """
        執(zhí)行非查詢語句

        調(diào)用示例：
            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 main():
## ms = MSSQL(host="localhost",user="sa",pwd="123456",db="PythonWeiboStatistics")
## #返回的是一個(gè)包含tuple的list，list的元素是記錄行，tuple的元素是每行記錄的字段
## ms.ExecNonQuery("insert into WeiBoUser values('2','3')")

    ms = MSSQL(host="localhost",user="sa",pwd="123456",db="PythonWeiboStatistics")
    resList = ms.ExecQuery("SELECT id,weibocontent FROM WeiBo")
    for (id,weibocontent) in resList:
        print str(weibocontent).decode("utf8")

if __name__ == '__main__':
    main()