Python自定義類連線資料庫

利用之前unixODBC配置的資料來源，自定義一個類然後連線資料庫

import pyodbc
class DB18:
    def __init__(self):
        self.conn = self.open()
        if self.conn:
            self.cr = self.conn.cursor()


    def open(self):
        try:
            cnxn=pyodbc.connect("DSN=200;PWD=test1234")
            cnxn.setdecoding(pyodbc.SQL_WCHAR, encoding='GBK')
            cnxn.setdecoding(pyodbc.SQL_CHAR, encoding='GBK')
            cnxn.setencoding(encoding='GBK')
        except:
            print("資料庫連線失敗!請聯絡路保輝.")
        else:
            return cnxn

    def close(self):
        try:
            self.conn.close()
        except:
            raise("關閉異常")
 

使用pandas read_sql讀取資料到DataFrame

import pandas as pd
db = DB18()
mycon = db.conn
stmt = 'select first 10 * from bftj where date=today-1'
df = pd.read_sql(stmt,mycon,index_col=['date'])