task.py

import json
import time
import random
import requests
from concurrent.futures import ThreadPoolExecutor
from util import multip_accounts_task

MAX_INVOKE_TIMES = 80
EXECUTOR_POOL_SIZE = 10


def config(path, data=None):
    with open(path, mode="r+") as conf:
        if not data:
            return json.load(conf)
        json.dump(data, conf, sort_keys=True, indent=4)


def get_access_token(app):
    return requests.post(
        'https://login.microsoftonline.com/common/oauth2/v2.0/token',
        data={
            'grant_type': 'refresh_token',
            'refresh_token': app['refresh_token'],
            'client_id': app['client_id'],
            'client_secret': app['client_secret'],
            'redirect_uri': app['redirect_uri']
        }
    ).json()


def invoke_api(path):
    app = config(path)
    tokens = get_access_token(app)
    access_token = tokens.get('access_token')
    refresh_token = tokens.get('refresh_token')
    username = app['username']

    if len(access_token) < 50 or len(refresh_token) < 50:
        return f'✘ 帳號 [{username}] API 呼叫失敗'

    apis = [
        'https://graph.microsoft.com/v1.0/groups',
        'https://graph.microsoft.com/v1.0/sites/root',
        'https://graph.microsoft.com/v1.0/sites/root/sites',
        'https://graph.microsoft.com/v1.0/sites/root/drives',
        'https://graph.microsoft.com/v1.0/sites/root/columns',
        'https://graph.microsoft.com/v1.0/me/',
        'https://graph.microsoft.com/v1.0/me/events',
        'https://graph.microsoft.com/v1.0/me/people',
        'https://graph.microsoft.com/v1.0/me/contacts',
        'https://graph.microsoft.com/v1.0/me/calendars',
        'https://graph.microsoft.com/v1.0/me/drive',
        'https://graph.microsoft.com/v1.0/me/drive/root',
        'https://graph.microsoft.com/v1.0/me/drive/root/children',
        'https://graph.microsoft.com/v1.0/me/drive/recent',
        'https://graph.microsoft.com/v1.0/me/drive/sharedWithMe',
        'https://graph.microsoft.com/v1.0/me/onenote/pages',
        'https://graph.microsoft.com/v1.0/me/onenote/sections',
        'https://graph.microsoft.com/v1.0/me/onenote/notebooks',
        'https://graph.microsoft.com/v1.0/me/outlook/masterCategories',
        'https://graph.microsoft.com/v1.0/me/mailFolders',
        'https://graph.microsoft.com/v1.0/me/mailFolders/Inbox/messages/delta',
        'https://graph.microsoft.com/v1.0/me/mailFolders/inbox/messageRules',
        'https://graph.microsoft.com/v1.0/me/messages',
        "https://graph.microsoft.com/v1.0/me/messages?$filter=importance eq 'high'",
        'https://graph.microsoft.com/v1.0/me/messages?$search="hello world"',
        'https://graph.microsoft.com/beta/me/messages?$select=internetMessageHeaders&$top',
    ]
    headers = {'Authorization': f'Bearer {access_token}'}

    def single_period(period):
        result = '==========================================================================================\n'
        random.shuffle(apis)
        for api in apis:
            try:
                if requests.get(api, headers=headers).status_code == 200:
                    result += '{:>20s} | {:>6s} | {:<50s}\n'.format(
                        f'帳號 : {username}',
                        f'週期 : {period}',
                        f'成功 : {api}'
                    )
            except Exception:
                time.sleep(random.random() * 3)
                pass
        return result

    periods = random.randint(50, MAX_INVOKE_TIMES)
    futures, pool = [], ThreadPoolExecutor(EXECUTOR_POOL_SIZE)
    for period in range(1, periods):
        futures.append(pool.submit(single_period, period))

    result = ''
    for future in futures:
        result += future.result()

    app['refresh_token'] = refresh_token
    config(path, app)
    return f'{result}✔ 帳號 [{username}] API 呼叫成功'


if __name__ == '__main__':
    multip_accounts_task(invoke_api)