85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
import time
|
||
import requests
|
||
import pymysql
|
||
from datetime import datetime
|
||
|
||
# MySQL配置
|
||
DB_CONFIG = {
|
||
"host": "127.0.0.1",
|
||
"user": "root",
|
||
"password": "2GS@bPYcgiMyL14A",
|
||
"database": "btcdb",
|
||
"port": 4423
|
||
}
|
||
|
||
# 获取当前时间戳
|
||
def get_current_timestamp():
|
||
return int(time.time())
|
||
|
||
# 获取API1的BTC价格（示例：币安）
|
||
def get_binance_price():
|
||
url = "https://api.binance.com/api/v3/ticker/price?symbol=BTCUSDT"
|
||
resp = requests.get(url, timeout=5)
|
||
resp.raise_for_status()
|
||
data = resp.json()
|
||
return float(data["price"])
|
||
|
||
# 获取API2的BTC价格（示例：Coinbase）
|
||
def get_coinbase_price():
|
||
url = "https://api.coinbase.com/v2/prices/spot?currency=USD"
|
||
resp = requests.get(url, timeout=5)
|
||
resp.raise_for_status()
|
||
data = resp.json()
|
||
return float(data["data"]["amount"])
|
||
|
||
# 更新或插入价格
|
||
def upsert_price(source, price, timestamp):
|
||
connection = pymysql.connect(**DB_CONFIG)
|
||
try:
|
||
with connection.cursor() as cursor:
|
||
# 先判断该 source 是否已存在
|
||
sql_check = "SELECT id FROM btc_realtime_prices WHERE source = %s"
|
||
cursor.execute(sql_check, (source,))
|
||
result = cursor.fetchone()
|
||
|
||
if result:
|
||
# 已存在，执行更新
|
||
sql_update = """
|
||
UPDATE btc_realtime_prices
|
||
SET price = %s, timestamp = %s
|
||
WHERE source = %s
|
||
"""
|
||
cursor.execute(sql_update, (price, timestamp, source))
|
||
else:
|
||
# 不存在，执行插入
|
||
sql_insert = """
|
||
INSERT INTO btc_realtime_prices (timestamp, source, price)
|
||
VALUES (%s, %s, %s)
|
||
"""
|
||
cursor.execute(sql_insert, (timestamp, source, price))
|
||
connection.commit()
|
||
finally:
|
||
connection.close()
|
||
|
||
def main():
|
||
while True:
|
||
now_ts = get_current_timestamp()
|
||
|
||
try:
|
||
binance_price = get_binance_price()
|
||
print(f"Binance BTC Price: {binance_price}")
|
||
upsert_price("binance", binance_price, now_ts)
|
||
except Exception as e:
|
||
print(f"获取Binance价格失败: {e}")
|
||
|
||
try:
|
||
coinbase_price = get_coinbase_price()
|
||
print(f"Coinbase BTC Price: {coinbase_price}")
|
||
upsert_price("coinbase", coinbase_price, now_ts)
|
||
except Exception as e:
|
||
print(f"获取Coinbase价格失败: {e}")
|
||
|
||
time.sleep(60) # 每分钟执行一次
|
||
|
||
if __name__ == "__main__":
|
||
main() |