Question

我正在编写简单的API管理器，但在方法中使用字典存在问题，这是我到目前为止编写的内容：

class BnManager():
    def __init__(self, api_key, api_secret):

        self.api_key = api_key
        self.api_secret = api_secret
        self.client = Client(api_key, api_secret)

    def get_candles(self, symbol, interval):
        self.symbol = symbol
        self.interval = interval
        choice = {
            '1m' : Client.KLINE_INTERVAL_1MINUTE,
            '3m' : Client.KLINE_INTERVAL_3MINUTE,
            '5m' : Client.KLINE_INTERVAL_5MINUTE,
            '15m' : Client.KLINE_INTERVAL_15MINUTE,
            '30m' : Client.KLINE_INTERVAL_30MINUTE,
            '1h' : Client.KLINE_INTERVAL_1HOUR,
            '2h' : Client.KLINE_INTERVAL_2HOUR,
            '4h' : Client.KLINE_INTERVAL_4HOUR,
            '6h' : Client.KLINE_INTERVAL_6HOUR,
            '8h' : Client.KLINE_INTERVAL_8HOUR,
            '12h' : Client.KLINE_INTERVAL_12HOUR,
            '1d' : Client.KLINE_INTERVAL_1DAY,
            '3d' : Client.KLINE_INTERVAL_3DAY,
            '1w' : Client.KLINE_INTERVAL_1WEEK,
            '1m' : Client.KLINE_INTERVAL_1MONTH,
            }
        self.klines = self.client.get_klines(
            self.symbol, choice[self.interval])
        self.df = pd.DataFrame(self.klines, columns=[
            'Date', 'Open', 'High', 'Low', 'Close', 'Volume',
            'x', 'x1', 'x2', 'x3', 'x4', 'x5'])
        self.df.drop(labels=['x', 'x1', 'x2', 'x3', 'x4', 'x5'],
                     axis=1, inplace=True).astype(float)
        self.df['Date'] = date2num(pd.to_datetime(df.Date, unit='ms'))
        self.df['Change'] = df['Close'].diff()

当我尝试执行get_candles方法时出现问题。例如，当我写manager.get_candles('BTCUSDT', '1m')时，我得到：

self.symbol，choice [self.interval] TypeError：get_candles（）需要1 位置参数，但给出了3个

我知道这可能是个琐碎的问题，但我真的不知道问题出在哪里。我的第二个问题是：如何不使用dict来编写它。我的意思是我想实现以下目标：

self.klines = self.client.get_klines(
            self.symbol, Client.KLINE_INTERVAL_+interval)

Answer 1

对于重写问题，您可以研究Programmatic access to enumeration members and their attributes。基本上，文档说明您可以将字符串用作枚举的键。

from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

print(Color['RED'])  # output: <Color.RED: 1>

如何用字典选项编写Python方法？

1 个答案: