【Python】bitflyer の Ticker を MySQL に突っ込む

2018年4月6日2020年10月11日Python,開発

おはようございます。

昨日に引き続き、今回はTicker のデータをMySQLに登録してみます。

基本的に SQLite でやったことなんでサクッと。

プログラムは以前の記事を流用します。

【Python】BitflyerのTickerをSQLiteに突っ込む

1. 新規クラス作成
2. プログラムの修正
3. 起動してデータを確認
4. まとめ

新規クラス作成

SQLiteUtil.py をコピーしてクラス名を変更し、処理の内容を MySQL ように書き換える。

MySQLUtil.py

import mysql.connector
import logging
from contextlib import closing


class MySQLUtil:
    """
    MySQL 操作用クラス
    """

    def __init__(self, host="localhost", port="3306", user="USER01", password="USER01", database="DB01"):
        self.config = {
            "host": host,
            "port": port,
            "user": user,
            "password": password,
            "database": database
        }
        self.create_db()

    def create_db(self):
        """
        データベース、及び必要なテーブルを作成します.
        :return:
        """

        logging.info("create database")
        with closing(mysql.connector.connect(**self.config)) as conn:

            c = conn.cursor()

            # ティッカーテーブル
            sql = "CREATE TABLE IF NOT EXISTS TBL_TICKER ("
            sql += "  PRODUCT_CODE VARCHAR(20)"
            sql += ", TIME_STAMP DATETIME"
            sql += ", TICK_ID VARCHAR(10)"
            sql += ", BEST_BID VARCHAR(20)"
            sql += ", BEST_ASK VARCHAR(20)"
            sql += ", BEST_BID_SIZE VARCHAR(20)"
            sql += ", BEST_ASK_SIZE VARCHAR(20)"
            sql += ", TOTAL_BID_DEPTH VARCHAR(20)"
            sql += ", TOTAL_ASK_DEPTH VARCHAR(20)"
            sql += ", LTP VARCHAR(20)"
            sql += ", VOLUME VARCHAR(20)"
            sql += ", VOLUME_BY_PRODUCT VARCHAR(20)"
            sql += ", PRIMARY KEY (TICK_ID)"
            sql += ")"
            c.execute(sql)

            c.close()
            conn.commit()

    def delete_data(self):
        """
        データを削除します
        :return:
        """

        logging.info("delete_data")
        with closing(mysql.connector.connect(**self.config)) as conn:

            c = conn.cursor()

            # データクリア
            sql = "DELETE FROM TBL_TICKER"
            c.execute(sql)
            c.close()
            conn.commit()

    def insert_data(self, ticker):
        """
        データを登録します
        :param ticker:
        :return:
        """

        with closing(mysql.connector.connect(**self.config)) as conn:

            c = conn.cursor()
            # データ登録
            sql = "INSERT INTO TBL_TICKER VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
            c.execute(sql, ticker)

            c.close()
            conn.commit()

importmysql.connector

importlogging

fromcontextlibimportclosing

classMySQLUtil:

"""

MySQL 操作用クラス

"""

def__init__(self,host="localhost",port="3306",user="USER01",password="USER01",database="DB01"):

self.config={

"host":host,

"port":port,

"user":user,

"password":password,

"database":database

}

self.create_db()

defcreate_db(self):

"""

データベース、及び必要なテーブルを作成します.

:return:

"""

logging.info("create database")

withclosing(mysql.connector.connect(**self.config))asconn:

c=conn.cursor()

# ティッカーテーブル

sql="CREATE TABLE IF NOT EXISTS TBL_TICKER ("

sql+=" PRODUCT_CODE VARCHAR(20)"

sql+=", TIME_STAMP DATETIME"

sql+=", TICK_ID VARCHAR(10)"

sql+=", BEST_BID VARCHAR(20)"

sql+=", BEST_ASK VARCHAR(20)"

sql+=", BEST_BID_SIZE VARCHAR(20)"

sql+=", BEST_ASK_SIZE VARCHAR(20)"

sql+=", TOTAL_BID_DEPTH VARCHAR(20)"

sql+=", TOTAL_ASK_DEPTH VARCHAR(20)"

sql+=", LTP VARCHAR(20)"

sql+=", VOLUME VARCHAR(20)"

sql+=", VOLUME_BY_PRODUCT VARCHAR(20)"

sql+=", PRIMARY KEY (TICK_ID)"

sql+=")"

c.execute(sql)

c.close()

conn.commit()

defdelete_data(self):

"""

データを削除します

:return:

"""

logging.info("delete_data")

withclosing(mysql.connector.connect(**self.config))asconn:

c=conn.cursor()

# データクリア

sql="DELETE FROM TBL_TICKER"

c.execute(sql)

c.close()

conn.commit()

definsert_data(self,ticker):

"""

データを登録します

:param ticker:

:return:

"""

withclosing(mysql.connector.connect(**self.config))asconn:

c=conn.cursor()

# データ登録

sql="INSERT INTO TBL_TICKER VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"

c.execute(sql,ticker)

c.close()

conn.commit()

プログラムの修正

Ticker 保存処理の修正

BfApi.py

def save_ticker(self, message):
    """
    ティッカー情報をデータベースに保存
    :param message:
    :return:
    """
    mysql = MySQLUtil()
    mysql.insert_data(self.convert_ticker_tuple(message.message))