[google 日历读取工具 python] oauth2client 授权 获得日期指定日期段所有活动

GoogleAuth.py

#!/usr/bin/env python
# -*- coding: utf-8 -*
from __future__ import print_function
import httplib2
import os

from apiclient import discovery
from oauth2client import client
from oauth2client import tools
from oauth2client.file import Storage

try:
    import argparse
    args=argparse.ArgumentParser(parents=[tools.argparser])
    flags = args.parse_args()
except ImportError:
    flags = None

# If modifying these scopes, delete your previously saved credentials
# at ~/.credentials/calendar-python-quickstart.json

SCOPES = 'https://www.googleapis.com/auth/calendar.readonly'
CLIENT_SECRET_FILE = './data/client_secret.json'
APPLICATION_NAME = 'Google Calendar API'
HOME_DIR = "./data/"


def get_credentials(user_name):
    """Gets valid user credentials from storage.

    If nothing has been stored, or if the stored credentials are invalid,
    the OAuth2 flow is completed to obtain the new credentials.

    Returns:
        Credentials, the obtained credential.
    """
    credential_dir = os.path.join(HOME_DIR, '.credentials')
    if not os.path.exists(credential_dir):
        os.makedirs(credential_dir)
    credential_path = os.path.join(credential_dir,
                                   user_name+'_calendar.json')

    store = Storage(credential_path)
    credentials = store.get()
    if not credentials or credentials.invalid:
        flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
        flow.user_agent = APPLICATION_NAME
        http = httplib2.Http(proxy_info=httplib2.ProxyInfo(
            httplib2.socks.PROXY_TYPE_SOCKS5, '127.0.0.1', 1080))

        credentials = tools.run_flow(flow, store, flags, http)

        print('Storing credentials to ' + credential_path)
    return credentials


def get_service(user_name, credential_path=None, proxy=None):
    """
    获得日历的server
    :return:
    """
    # 获得授权文件
    if credential_path is None:
        credentials = get_credentials(user_name)
    else:
        store = Storage(credential_path)
        credentials = store.get()

    # 设置代理
    if proxy is None:
        http = httplib2.Http(proxy_info=httplib2.ProxyInfo(
            httplib2.socks.PROXY_TYPE_SOCKS5, '127.0.0.1', 1080))
        http = credentials.authorize(http)
        service = discovery.build('calendar', 'v3', http=http)
        return service
    else:
        http = httplib2.Http(proxy_info=proxy)
        http = credentials.authorize(http)
        service = discovery.build('calendar', 'v3', http=http)
        return service


def get_calender_id(credential_service, name=u"时间日志"):
    """
    获得 日历 ID
    :param credential_service:  日历服务
    :param name:  日历的名称
    :return: 此名称日历对应ID
    """
    page_token = None
    id = None
    while True:
        calendar_list = credential_service.calendarList().list(pageToken=page_token).execute()
        for calendar_list_entry in calendar_list['items']:
            if name == calendar_list_entry['summary']:
                id = calendar_list_entry['id']
        page_token = calendar_list.get('nextPageToken')
        if not page_token or not id:
            break
    return id


def get_calender_content(credential_service, calender_id, min_time, max_time):
    """
    获得 日历 的指定时间段所有的活动
    :param credential_service:
    :param calender_id:
    :param min_time:
    :param max_time:
    :return:
    """
    # calendar = credential_service.calendars().get(calendarId=calender_id).execute()
    calendar_request = credential_service.events().list(calendarId=calender_id,
                                                     timeMin=min_time,
                                                     timeMax=max_time,
                                                     singleEvents=True,
                                                     orderBy='startTime')
    events_response = calendar_request.execute()
    list_result = []
    while events_response is not None:
        # 1. 添加日历结果到list列表
        events = events_response.get('items', [])
        for event in events:
            start = event['start'].get('dateTime', event['start'].get('date'))
            end = event['end'].get('dateTime', event['end'].get('date'))
            title = event.setdefault('summary',"")
            listIndex = [start, end, title]
            list_result.append(listIndex)
        # 2. 获得下一页内容
        t_calendar_request = credential_service.events().list_next(calendar_request, events_response)
        if t_calendar_request is None:
            break
        events_response = t_calendar_request.execute()
    # print("final"+str(len(list_result)))
    return list_result


if __name__ == '__main__':
    credential_service = get_service("mm")
    calender_id = get_calender_id(credential_service)
    content = get_calender_content(credential_service, calender_id, "2019-01-15T0:0:0Z", "2019-01-18T0:0:0Z")
    print(content)