使用oauth2和Google API

时间:2017-10-13 20:24:24

标签: python python-3.x google-api google-api-python-client oauth2client

我在某些脚本中使用Google API服务并遇到一些问题。这个错误很奇怪,但我们走了。 我有一个列出我的Google云端硬盘文件的脚本。

from apiclient import discovery
from httplib2 import Http
from oauth2client import file, client, tools

SCOPES = 'https://www.googleapis.com/auth/drive.readonly.metadata'
store = file.Storage('storage.json')
creds = store.get()
if not creds or creds.invalid:
    flow = client.flow_from_clientsecrets('client_secret.json', SCOPES)
    creds = tools.run_flow(flow, store)
DRIVE = discovery.build('drive', 'v3', http=creds.authorize(Http()))

files = DRIVE.files().list().execute().get('files', [])
for f in files:
    print(f['name'], f['mimeType'],f['id'])

效果很好,我从Google API下载client_secret.json并将其保存在同一个文件夹中,然后我启动脚本来检查每件事情是否合适。

然后我开始编辑我的文件以更改执行方式并且不读取文件,而是调用脚本并发送 client_id client_secret < / em>字符串到脚本,最终版本是这样的:

import sys
from apiclient import discovery
from httplib2 import Http
from oauth2client import file, client, tools

# ID and SECRET arguments
client_id = sys.argv[1]
client_secret = sys.argv[2]

SCOPES = 'https://www.googleapis.com/auth/drive.readonly.metadata'

def listFiles(drive):
    """Receive the service and list the files"""
    files = drive.files().list().execute().get('files', [])
    for f in files:
        print(f['name'], f['mimeType'],f['id'])

def main():
    store = file.Storage('storage.json')
    creds = store.get()
    if not creds or creds.invalid:
        flow = client.OAuth2WebServerFlow(client_id, client_secret, SCOPES)
        creds = tools.run_flow(flow, store, tools.argparser.parse_args())
    DRIVE = discovery.build('drive', 'v3', http=creds.authorize(Http()))
    listFiles(DRIVE)


if __name__ == "__main__":
    main()

我第一次启动这个新版本的脚本时,因为旧版本的脚本创建了storage.json文件。 然后我将我的新版本脚本移动到另一个文件夹或机器(storage.json文件不存在)以检查它是否有效,然后我得到了这个:

$ python3 drive_list.py asdasdasdsa jijfkljflksdjflksdj

/usr/local/lib/python3.4/dist-packages/oauth2client/_helpers.py:255: UserWarning: Cannot access storage.json: No such file or directory
  warnings.warn(_MISSING_FILE_MESSAGE.format(filename))
usage: drive_list.py [--auth_host_name AUTH_HOST_NAME]
                     [--noauth_local_webserver]
                     [--auth_host_port [AUTH_HOST_PORT [AUTH_HOST_PORT ...]]]
                     [--logging_level {DEBUG,INFO,WARNING,ERROR,CRITICAL}]
drive_list.py: error: unrecognized arguments: asdasdasdsa jijfkljflksdjflksdj

有关storage.json文件的警告是正常的,并且出现在两个脚本版本中,是oauth2client的一部分。

这是古玩部分,为什么在storage.json文件存在时才识别参数(仅创建阅读client_secret.json)?如果脚本第一次启动它会创建文件。 这个错误确实很奇怪,而我只是想找到确切发生的事情。 如果有人可以帮助我,我会非常感激。

1 个答案:

答案 0 :(得分:2)

之所以会这样,是因为您要导入oauth2client.tools模块。

要正常工作,此模块依赖于标准argparse模块。如果您不知道,该标准模块用于编写用户友好的命令行界面,并轻松管理命令行参数。这与您使用sys.argv[1]sys.argv[2]参数的情况不符。

要解决此问题,可以像下面的示例一样在命令行中添加新参数。进行此修改后,您将可以像这样运行该工具

python3 drive_list.py -ci "your_client_id" -cs "your_client_secret"

以下是您的代码经过略微修改以添加新的命令行参数的情况:

import argparse
from apiclient import discovery
from httplib2 import Http
from oauth2client import file, client, tools

# ID and SECRET arguments as new command line parameters
# Here is where you extend the oauth2client.tools startnd arguments
tools.argparser.add_argument('-ci', '--client-id', type=str, required=True, help='The client ID of your GCP project')
tools.argparser.add_argument('-cs', '--client-secret', type=str, required=True,
                             help='The client Secret of your GCP project')

SCOPES = 'https://www.googleapis.com/auth/drive.readonly.metadata'


def list_files(drive):
    """Receive the service and list the files"""
    files = drive.files().list().execute().get('files', [])
    for f in files:
        print(f['name'], f['mimeType'], f['id'])


def main():
    store = file.Storage('storage.json')
    creds = store.get()
    if not creds or creds.invalid:
        # You want to be sure to parse the args to take in consideration the new parameters
        args = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
        flow = client.OAuth2WebServerFlow(args.client_id, args.client_secret, SCOPES)
        creds = tools.run_flow(flow, store, tools.argparser.parse_args())
    drive_sdk = discovery.build('drive', 'v3', http=creds.authorize(Http()))
    list_files(drive_sdk)


if __name__ == "__main__":
    main()