python操作postgresql數據庫

　　之前在公司學習用python操作mysql數據庫，但是實際上我們公司一直用的是postgresql數據庫，所以我尋思着照葫蘆畫瓢也寫一個python操作postgresql的工具類吧！這里用到了psycopg2模塊

直接上代碼：

import psycopg2
class PostGreSQL:
    # 初始化
    def __init__(self, dataBaseName, userName, password, host, port):
        self.dataBaseName = dataBaseName
        self.userName = userName
        self.password = password
        self.host = host
        self.port = port

        self._conn = self.GetConnect()
        if self._conn:
            self._cur = self._conn.cursor()

    # 獲取數據庫連接對象
    def GetConnect(self):
        conn = False
        try:
            conn = psycopg2.connect(
                database=self.dataBaseName,
                user=self.userName,
                password=self.password,
                host=self.host,
                port=self.port
            )
        except Exception as err:
            print("連接數據庫失敗，%s" % err)
        return conn
    # 執行查詢sql
    def ExecQuery(self, sql):
        res = ""
        try:
            self._cur.execute(sql)
            res = self._cur.fetchall()
        except Exception as err:
            print("查詢失敗, %s" % err)
        else:
            return res

    # 執行增刪改sql
    def ExceNonQuery(self, sql):
        flag = False
        try:
            self._cur.execute(sql)
            self._conn.commit()
            flag = True
        except Exception as err:
            flag = False
            self._conn.rollback()
            print("執行失敗, %s" % err)
        else:
            return flag

    def GetConnectInfo(self):
        print("連接信息：")
        print("服務器:%s , 用戶名:%s , 數據庫:%s " % (self.host, self.userName, self.dataBaseName))