python - 从 Python 3.6 中的文件导入字典

Question

我正在尝试用 Python 制作一个机器人，以使用 tweepy 管理 Twitter 上的一些任务。

我将身份验证凭据保存在多个文件中，但我只需要一个文件。这是一个例子:

for x in range(0, 3):

  if x == 0 : from keysaccount1 import keys
  if x == 1 : from keysaccount2 import keys
  if x == 2 : from keysaccount3 import keys
  if x == 3 : from keysaccount4 import keys

  CONSUMER_KEY = keys['consumer_key']
  CONSUMER_SECRET = keys['consumer_secret']
  ACCESS_TOKEN = keys['access_token']
  ACCESS_TOKEN_SECRET = keys['access_token_secret']

  auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
  auth.set_access_token(ACCESS_TOKEN, ACCESS_TOKEN_SECRET)
  api = tweepy.API(auth, wait_on_rate_limit=True)

存储数据的文件采用这种格式，例如 keysaccount1 :

keys = dict(
    consumer_key =          'xxxxxxxxxxxxx',
    consumer_secret =       'xxxxxxxxxxxxx',
    access_token =          'xxxxxxxxxxxxx',
    access_token_secret =   'xxxxxxxxxxxxx',
)

有没有办法将所有这些保存在同一个文件中，因为我在循环中有 x 变量，我可以使用它来选择我正在管理的帐户？

Answer 1

您可以使用单个字典:

auth_data = {
    0: {
        'consumer_key': 'xxxxxxxxxx',
        'consumer_secret': 'xxxxxxxxxxxxx',
        'access_token': 'xxxxxxxxxx',
        'access_token_secret': 'xxxxxxxxxxxxxx',
       }
    ...

然后您可以像这样访问特定字段:

for x in range(0, 3):
    keyset = auth_data[x]

    auth = tweepy.OAuthHandler(keyset['consumer_key'], keyset['consumer_secret'])
    auth.set_access_token(keyset['access_token'], keyset['access_token_secret'])
    api = tweepy.API(auth, wait_on_rate_limit=True)

要保存数据，你可以使用 python 模块 pickle:

import pickle

def read():
    with open('data/keys.p', 'r') as f:
        return pickle.load(f)

def write(data): 
    with open('data/keys.p', 'w+') as f:
        return pickle.dump(f, data)

# Read data

auth_data = read()

# Write data

write(auth_data)