chinapnr / fishbase

自主开发、整理的一套 Python 基础函数库,涵盖 system 系统增强包、logger 日志记录增强包、file 文件处理增强包、 date 日期处理函数包、data 数据信息处理函数包、csv 处理增强函数包、crypt 加密/编码增强包等,可减少程序开发工作量、降低引用复杂度。
MIT License
94 stars 29 forks source link

fish_data, 抽象 sqlite 查询访问的函数 #158

Closed wingfish closed 5 years ago

wingfish commented 5 years ago

功能:将目前 sqlite 查询的代码抽象出来,

函数:sqlite_query()

    dir_path = os.path.dirname(os.path.abspath(__file__))

    conn = sqlite3.connect(os.path.join(dir_path, 'db', 'fish_data.sqlite'))

    cursor = conn.cursor()

    cursor.execute('select bin,bank,card_type,length from cn_cardbin where bank=:bank and card_type=:card_type',
                   {"bank": bank, "card_type": card_type})

    values = cursor.fetchall()

    cursor.close()
    conn.close()

    return values

将其中 sqlite 文件名和 sql 抽象出来

wingfish commented 5 years ago

修改如下,以后再加入更多异常判断

def sqlite_query(db, sql, params):

    dir_path = os.path.dirname(os.path.abspath(__file__))

    conn = sqlite3.connect(os.path.join(dir_path, 'db', db))

    cursor = conn.cursor()

    cursor.execute(sql, params)

    values = cursor.fetchall()

    cursor.close()
    conn.close()

    return values