Python 讀取 支付寶帳單並存儲到 Access 中

我有一個不少年前本身寫的C#+Access的記帳程序，用了不少年，如今花錢的機會多了，而且大部分走的支付寶，因而就想把帳單從支付寶網站上下載下來，直接寫入到Access，這樣就很省心了。

記帳程序是長這個樣子的：

 

還有報表彙總模塊兒：

 

Access 主要表結構以下：

 

支付寶支付流水下載下來以下：

 

具體代碼以下：

table="收支記錄表"
path="C:\\Users\\user\\Desktop\\tmp\\alipay_record_20190114.xlsx"

#test_connect()
read_excel(path)

 

#-*-coding:utf-8 -*-
import pypyodbc
import xlrd
from  ClassDef import *
import CommonUtil
import traceback

 

 

 

def read_excel(path):
    workbook=xlrd.open_workbook(path)
    sheet=workbook.sheet_by_index(0)
    nrows=sheet.nrows
    ncols=sheet.ncols
    print(nrows, ncols)

    j=0

    for i in range(1, nrows):
        # pay_time=sheet.row_values(i,0)#付款時間
        # print(pay_time)
        #print(sheet.cell(i, 0).value)  # 指定行和列獲取數據

        goodsName = sheet.cell(i, 5).value  #商品名稱
        inOrOut = sheet.cell(i, 7).value  ## 收、支
        status = sheet.cell(i, 8).value  ## 交易狀態
        # if (reason.find("收益發放") > -1):
        #     continue
        if inOrOut is not None and len(inOrOut.strip()) >0 and goodsName.find("收益發放") == -1\
                and goodsName.find("賺錢紅包推薦獎勵")==-1\
                and (status.find("交易成功")>-1 or status.find("等待確認收貨")>-1):
            #print(i,inOrOut)
            ioRecord = InOutRecord()

            ioRecord.reason=goodsName################################## reason
            ioRecord.inOrOut=inOrOut[0:1]########################### 收/支

            payTimeStr = sheet.cell(i, 0).value
            if payTimeStr is not None :
                # print (i, payTimeStr)
                payTimeStr = payTimeStr[0:8]
                dateStr = payTimeStr[0:4]
                monthStr = payTimeStr[4:6]
                dayStr = payTimeStr[6:8]
                # value = datetime.datetime(payTimeStr) XXXX
                # print(payTimeStr)
                # print(dateStr)
                # print(monthStr)
                # print(dayStr)

                #自動生成編號：先查當天有幾條記錄了
                filterDate = dateStr[2:4]+monthStr+dayStr
                numberStr,connStr=genAccessKeyAndTradeDate(filterDate, dateStr, monthStr, dayStr)
                # print("編號===" + numberStr)
                if len(numberStr) > 0:
                    ioRecord.numberStr = numberStr  #########################  編號
                else:
                    print("eeeeeeeeeeeeeeeeeeeeeeee  " + "編號自動生成錯誤" + "eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee")


                ioRecord.payDate=connStr###################################  日期
                ##########################################################################
                amount = sheet.cell(i, 6).value#金額
                ioRecord.amount=amount
                # print("金額==="+str(amount))

                #交易來源地
                tradeSource = sheet.cell(i, 3).value  # 交易來源地

                #獲取交易對方，來確認收支類型
                tradeObject = sheet.cell(i, 4).value  # 交易對方

                #獲取收支具體類型，如衣、食、住、用、行等
                ioRecord.type=getType(tradeObject)
                #修正 type
                if ioRecord.reason is not None:
                    if ioRecord.reason.find('打車')>-1:
                        ioRecord.type='行'
                #完善
                ioRecord.reason=ioRecord.reason.strip()+"【"+tradeObject.strip()+"】"+"【"+tradeSource.strip()+"】"
                # if len(ioRecord.reason)>15:#商品名稱只取前15個字符
                #     ioRecord.reason=ioRecord.reason[0:15]
                ioRecord.comment=tradeObject.strip()


                ret=insertRecord(ioRecord)#存儲
                #ret=insertTest();
                #ret=test_insertRecord(ioRecord)
                if ret==1:
                    print(str(j + 1) + "--" + str(
                        i) + "  " + ioRecord.numberStr + "  " + ioRecord.payDate + "  " + ioRecord.inOrOut + "  "
                          + str(ioRecord.amount) + "  " + ioRecord.type + "  " + ioRecord.reason + "---存儲成功")

                print("==============================================================================================")

                j=j+1

 

#記錄當前已經記了幾條帳目了
def getRecordCountByDate(date):
    conn,cur=connect_access()
    #sql="SELECT * FROM " + table+" where 編號 like"+date+""
    sql = "select * from 收支記錄表  where 編號 like "+"'"+"%"+date+"%"+"'"
    #sql = "select * from 收支記錄表  where 編號 =  "+"'"+date+"'"
    #sql = "select * from 收支記錄表  where 編號 like "  + date
    #print(sql)
    cur.execute(sql)
    #cur.execute(sql,date)
    alldata = cur.fetchall()
    total_rows = len(alldata)
    # total_cols = len(alldata[0])
    # print("總行數 = %d" % ( total_rows))
    return total_rows

 

#根據日期自動生成編號，並生成交易日期
def genAccessKeyAndTradeDate(filterDate, dateStr, monthStr, dayStr):
    recordCount = getRecordCountByDate(filterDate)
    print("filterDate==="+filterDate)
    print("recordCount==="+str(recordCount))
    numberStr = ""
    if recordCount == 0:
        numberStr = filterDate + "01"
    elif recordCount > 0 and recordCount < 9:
        numberStr = filterDate + "0" + str(recordCount + 1)
    elif recordCount >= 9:
        numberStr = filterDate + str(recordCount + 1)

    connStr = dateStr + "/" + monthStr + "/" + dayStr

    return numberStr, connStr

 

def getType(tradeObject):
    type=""
    if tradeObject is not None:
        if tradeObject.find('超市') > -1 or tradeObject.find('匯悅') > -1 or tradeObject.find(
                '大王') > -1 or tradeObject.find('便利') > -1: \
                type = '食'

        if tradeObject.find('商品') > -1 or tradeObject.find('蔬菜') > -1 or tradeObject.find('奶吧') > -1:
            type = "食"

        if tradeObject.find('條碼支付-總部') or tradeObject.find('食品') > -1 or tradeObject.find(
                '餐吧') > -1 or tradeObject.find('切面') > -1:
            type = "食"

        if tradeObject.find('德青源') > -1 or tradeObject.find('食') > -1:
            type = "食"

        if tradeObject.find('童') > -1 or tradeObject.find('母') > -1:
            type = '寶寶'

        if tradeObject.find('移動') > -1:
            type = "通訊費用"

        if tradeObject.find('地鐵') > -1:
            type = "行"

        if tradeObject.find('市婦幼') > -1 or tradeObject.find('御和堂') > -1 or tradeObject.find('醫院') > -1:
            type = '醫療健康'

        if tradeObject.find('七匹狼') > -1 or tradeObject.find('南極人') > -1:
            type = '衣'

        if tradeObject.find('圖書') > -1 or tradeObject.find('噹噹') > -1:
            type = '自我完善'

        if tradeObject.find('供電') > -1:
            type = '電費'

    return type

 

#存儲一條數據
def insertRecord(ioRecord):
    if (ioRecord is not None):
        #printItemAllInfo(ioRecord)
        conn, cursor = connect_access()
        #sql = "insert into 收支記錄表 (編號, 日期, 收支緣由, 收支, 金額, 類型) values( %s, %s, %s, %s, %s, %s)"
        # sql = "insert into 收支記錄表 (編號, 日期, 收支緣由, 收支, 金額, 類型) " \
        #       "values(%s, %s, %s, %s, %s, %s)"
        createDate = CommonUtil.getCurrentDateStr()
        sql = "insert into 收支記錄表 (編號, 日期, 收支緣由, 收支, 金額, 類型,備註) values("\
              + "'"+ str(ioRecord.numberStr) + "'" +","\
              + "'" + str(ioRecord.payDate) + "'" + ','\
              + "'" + str(ioRecord.reason) + "'" + ','\
              + "'" + str(ioRecord.inOrOut) + "'" + ','\
              + "'" + str(ioRecord.amount) + "'" + ','\
              + "'" + str(ioRecord.type) + "'"+ ',' \
              + "'" + str(ioRecord.comment+"_"+createDate) + "'" + ')'

        #sql = 'insert into 收支記錄表 (編號, 日期, 收支緣由, 收支, 金額, 類型) values('+str(ioRecord.numberStr)+','
        # +ioRecord.payDate+','+ioRecord.reason+','+ioRecord.inOrOut+','+float(ioRecord.amount)+','+ioRecord.type+');'

        # print("=======================================================================")
        # print(dataRow[16], dataRow[17], dataRow[18])
        #print(sql)
        #print(ioRecord.numberStr,ioRecord.payDate,ioRecord.reason,ioRecord.inOrOut,ioRecord.amount,ioRecord.type)


        #print(createDate)
        try:
            # cursor.execute(sql, (
            #     str(ioRecord.numberStr),str(ioRecord.payDate),str(ioRecord.reason),
            #     str(ioRecord.inOrOut),str(ioRecord.amount),str(ioRecord.type)))
            cursor.execute(sql)
            conn.commit()
            cursor.close()
            conn.close()
            return 1
        except Exception as e:
            traceback.print_exc()
            print("insertOnRecord() 出錯，具體記錄內容以下XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX")
            printItemAllInfo(ioRecord)
            cursor.close()
            conn.close()
            return -1

 

#獲取數據庫鏈接
def connect_access():
    mdb = 'Driver={Microsoft Access Driver (*.mdb,*.accdb)};' \
          'DBQ=D:\\我的\\個人軟件\\PersonalAsistance_2012_04_13\\PersonalAsistance\\PersonalAsistance\\bin\\Debug\\PADB.mdb'
    conn = pypyodbc.win_connect_mdb(mdb)
    cur = conn.cursor()
    return conn,cur

總結:

這裏面有幾個關鍵點：

一、讀Excel（這個不算，比較淘常見）

二、生成Access表的主鍵，這個主鍵的規則是這樣的：取年的最後兩位，月的兩位，日的兩位，而後從01開始日後編碼，有幾條編到幾；

三、鏈接Access，讀、寫Access；

 

python 代碼已經放出來了，記帳程序有須要的可QQ我：861712499