#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
Date: 2025/6/16 18:00
Desc: 唯爱期货-期权保证金
https://www.iweiai.com/qihuo/yuanyou
"""

import requests
import pandas as pd
from io import StringIO

from bs4 import BeautifulSoup
from functools import lru_cache


@lru_cache()
def option_margin_symbol() -> pd.DataFrame:
    """
    获取商品期权品种代码和名称
    :return: 商品期权品种代码和名称
    :rtype: pandas.DataFrame
    """
    url = "https://www.iweiai.com/qiquan/yuanyou"
    r = requests.get(url)
    soup = BeautifulSoup(r.content, features="lxml")
    symbol_text = [
        item.get_text() for item in soup.find_all("a") if "qiquan" in item["href"]
    ]
    symbol_url = [
        item["href"] for item in soup.find_all("a") if "qiquan" in item["href"]
    ]
    symbol_df = pd.DataFrame([symbol_text, symbol_url]).T
    symbol_df.columns = ["symbol", "url"]
    return symbol_df


def option_margin(symbol: str = "原油期权") -> pd.DataFrame:
    """
    获取商品期权保证金
    :param symbol: 商品期权品种名称, 如 "原油期权"，可以通过 ak.option_margin_symbol() 获取所有商品期权品种代码和名称
    :type symbol: str
    :return: 商品期权保证金
    :rtype: pandas.DataFrame
    """
    option_margin_symbol_df = option_margin_symbol()
    url = option_margin_symbol_df[option_margin_symbol_df["symbol"] == symbol][
        "url"
    ].values[0]
    r = requests.get(url)
    soup = BeautifulSoup(r.content, features="lxml")
    updated_time = soup.find_all("small")[0].get_text().strip("最近更新：")
    temp_df = pd.read_html(StringIO(r.text))[0]
    temp_df["更新时间"] = updated_time

    temp_df["结算价"] = pd.to_numeric(temp_df["结算价"], errors="coerce")
    temp_df["交易乘数"] = pd.to_numeric(temp_df["交易乘数"], errors="coerce")
    temp_df["买方权利金"] = pd.to_numeric(temp_df["买方权利金"], errors="coerce")
    temp_df["卖方保证金"] = pd.to_numeric(temp_df["卖方保证金"], errors="coerce")
    temp_df["开仓手续费"] = pd.to_numeric(temp_df["开仓手续费"], errors="coerce")
    temp_df["平今手续费"] = pd.to_numeric(temp_df["平今手续费"], errors="coerce")
    temp_df["平昨手续费"] = pd.to_numeric(temp_df["平昨手续费"], errors="coerce")
    temp_df["手续费(开+平今)"] = pd.to_numeric(
        temp_df["手续费(开+平今)"], errors="coerce"
    )
    return temp_df


if __name__ == "__main__":
    option_margin_df = option_margin(symbol="原油期权")
    print(option_margin_df)