Python更新财智理财行情数据库

财智家庭理财软件是在PC端的家庭记账和理财软件中市场占有率很高。这个软件在股票基金等理财记账方面做得还可以，行情数据可以通过他们公司的服务器在线更新。
但是近期发生的两件事让我萌生了自己想办法更新行情数据的想法。

1. 去年有几周行情数据都无法更新，据说原因是更换服务器和证书问题。
2. 我的某个基金（东方红睿满）无法获取行情数据，客服的答复是软件使用的行情数据源（http://jingzhi.funds.hexun.com/jz/kaifang.htm）没有这个基金的数据，公司也没有计划更换行情数据源。

实际上网上能够提供股票和基金行情的数据源有很多，用Python写个爬虫程序抓数据不难。剩下的问题就是如何把行情数据写入财智软件的数据库。我花了点时间研究了一下，总结如下：

1. 财智软件的行情数据库使用Access数据库，文件是Data目录的mhlink.mdb；
2. 数据库文件中的TBSecuPrice表格存放股票基金的行情数据；

3. 表格中的字段SecuCode/PriceDate/Price分别表示证券代码/价格日期/价格，CurrType表示货币类型，人民币对应1，ObjType表示证券的种类，股票对应3，基金对应4。

   
   mhlink.PNG

Python对Access数据库操作可以使用pyodbc库，另外需要安装Microsoft Access Database Engine (https://www.microsoft.com/en-us/download/details.aspx?id=13255)。
需要注意的是32位版本和64位版本的选择，如果Office是32位的，数据库引擎和Python都需要安装32位版本的。

范例代码如下：

# -*- coding:utf-8 -*-
import pyodbc


class MSAccessODBC(object):
    def __init__(self, db_file, db_table, password=''):
        self.file = db_file
        self.password = password
        self.table = db_table
        connect_str = 'Driver={Microsoft Access Driver (*.mdb, *.accdb)};PWD' + self.password + ";DBQ=" + self.file
        self.connect = pyodbc.connect(connect_str)
        self.cursor = self.connect.cursor()

    def close(self):
        self.cursor.close()
        self.connect.close()

    def exec_sql(self, sql):
        try:
            self.cursor.execute(sql)
            if sql.upper().startswith('SELECT'):
                ret = self.cursor.fetchall()
                return ret
            else:
                self.connect.commit()
                return True
        except pyodbc.Error:
            return False

    def sql_insert(self, column_list, values_list):
        sql = u'INSERT INTO TableName (ColumnList)'
        sql += '\r\nVALUES (ValuesList)'
        sql = sql.replace('TableName', self.table)
        sql = sql.replace('ColumnList', unicode(column_list).replace('[', '').replace(']', '').replace("'", ''))
        sql = sql.replace('ValuesList', unicode(values_list).replace('[', '').replace(']', ''))
        ret = self.exec_sql(sql)
        return ret

    def sql_update(self, column, value, conditions_dict=None):
        sql = u'UPDATE TableName'
        sql += '\r\nSET ColumnName=ColumnValue'
        sql = sql.replace('TableName', self.table)
        sql = sql.replace('ColumnName', column)
        sql = sql.replace('ColumnValue', value)
        if conditions_dict:
            sql += '\r\nWHERE '
            for condition in conditions_dict.items():
                sql += condition[0] + '=' + condition[1] + '\r\nAND '
            sql = sql[:-5]
        ret = self.exec_sql(sql)
        return ret

    def sql_select(self, column, conditions_dict=None):
        sql = u'SELECT ColumnName'
        sql += '\r\nFROM TableName'
        sql = sql.replace('ColumnName', column)
        sql = sql.replace('TableName', self.table)
        if conditions_dict:
            sql += '\r\nWHERE '
            for condition in conditions_dict.items():
                sql += condition[0] + '=' + condition[1] + '\r\nAND '
            sql = sql[:-5]
        ret = self.exec_sql(sql)
        return ret


def demo():
    db_file = 'D:\Program Files (x86)\MoneyWise\MoneyHome8\Data\mhlink.mdb'
    db_table = 'TBSecuPrice'
    mhlink = MSAccessODBC(db_file, db_table)
    record = {'SecuCode': '600036', 'PriceDate': '2018-02-22', 'Price': '31.50'}

    # 查询数据库中是否已经有这个股票这一天的价格
    conditions = {'SecuCode': "'{0}'".format(record['SecuCode']), 'PriceDate': '#{0}#'.format(record['PriceDate'])}
    if not mhlink.sql_select('*', conditions):
        # 更新股票价格
        mhlink.sql_update('Price', record['Price'], conditions)
    else:
        # 新增股票价格
        column_list = ['SecuCode',
                       'PriceDate',
                       'Price',
                       'ObjectQuant',
                       'CurrType',
                       'ObjType']
        values_list = [record['SecuCode'],
                       record['PriceDate'],
                       record['Price'],
                       '1.0000',
                       '1',
                       '4']
        mhlink.sql_insert(column_list, values_list)


if __name__ == "__main__":
    demo()