开发环境配置

import json
import time

import requests

lock_meta = []


def write_lock_meta(file_name: str = "-"):
    if file_name == "-":
        file_name = "plugins.lock." + \
            time.strftime('%Y%m%d-%H%M%S', time.localtime(time.time()))
    with open(file_name, 'w+') as write_f:
        write_f.write(json.dumps(lock_meta, indent=4, ensure_ascii=False))


def download_plugin(plugin_meta: dict):
    content = requests.get(plugin_meta['resource_url']).content
    with open("./files/" + plugin_meta['pack_name'], 'wb+') as write_f:
        write_f.write(content)


def get_plugin_meta(plugin_name: str) -> dict:
    print(f"get metadata for plugin {plugin_name}")
    url = "https://marketplace.visualstudio.com/_apis/public/gallery/extensionquery"
    headers = {
        'authority': 'marketplace.visualstudio.com',
        'accept': 'application/json;api-version=7.1-preview.1;excludeUrls=true',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36',
    }
    payload = json.dumps({
        "assetTypes": None,
        "filters": [
            {
                "criteria": [
                    {
                        "filterType": 7,
                        "value": plugin_name,
                    }
                ],
                "direction": 2,
                "pageSize": 100,
                "pageNumber": 1,
                "sortBy": 0,
                "sortOrder": 0,
                "pagingToken": None
            }
        ],
        "flags": 2151
    })
    response = requests.request("POST", url, headers=headers, data=payload)
    resp = json.loads(response.text)
    latest_version_meta = resp['results'][0]['extensions'][0]['versions'][0]
    version = latest_version_meta['version']
    pack_name = f'{plugin_name.lower()}-{version}.vsix'
    files = latest_version_meta['files']
    resource_url = ''
    for f in files:
        if f['assetType'] == "Microsoft.VisualStudio.Services.VSIXPackage":
            resource_url = f['source']
    return {
        'version': version,
        'pack_name': pack_name,
        'resource_url': resource_url,
    }


def handle_plugins(file_name: str = "./plugins.list") -> None:
    file = open(file_name)
    while True:
        line = file.readline()
        line = line.rstrip()
        if not line:
            break
        meta = get_plugin_meta(line)
        print(f'get meta {meta}')
        lock_meta.append(meta)
        print(f'download plugin {line}')
        download_plugin(meta)
    write_lock_meta()


def main():
    handle_plugins()


if __name__ == "__main__":
    main()

import json
import time
from typing import Union

import requests

lock_meta = []


def write_lock_meta(file_name: str = "-"):
    if file_name == "-":
        file_name = "plugins.lock." + \
            time.strftime('%Y%m%d-%H%M%S', time.localtime(time.time()))
    with open(file_name, 'w+') as write_f:
        write_f.write(json.dumps(lock_meta, indent=4, ensure_ascii=False))


def download_file(url: str, file_name: str):
    content = requests.get(url).content
    with open(f"./files/{file_name}", 'wb+') as write_f:
        write_f.write(content)


def get_meta_from_open_vsx(plugin_name: str) -> Union[dict, None]:
    print(f"get plugin {plugin_name} metadata from open-vsx")
    url = f"https://open-vsx.org/api/{plugin_name.split('.')[0]}/{plugin_name.split('.')[1]}"
    payload = {}
    headers = {
        'Accept': 'application/json',
        'Accept-Language': 'zh-CN,zh;q=0.9',
        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36',
    }
    response = requests.request("GET", url, headers=headers, data=payload)
    resp = json.loads(response.text)
    if not resp.get('version', None):
        print(f"plugin {plugin_name} not found in open-vsx")
        return None
    version = resp['version']
    pack_name = f'{plugin_name.lower()}-{version}.vsix'
    resource_url = resp['files']['download']
    return {
        'version': version,
        'pack_name': pack_name,
        'resource_url': resource_url,
        'src': 'open-vsx',
    }


def get_meta_from_microsoft(plugin_name: str) -> Union[dict, None]:
    print(f"get plugin {plugin_name} metadata from microsoft")
    url = "https://marketplace.visualstudio.com/_apis/public/gallery/extensionquery"
    headers = {
        'authority': 'marketplace.visualstudio.com',
        'accept': 'application/json;api-version=7.1-preview.1;excludeUrls=true',
        'accept-language': 'zh-CN,zh;q=0.9',
        'content-type': 'application/json',
        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36',
    }
    payload = json.dumps({
        "assetTypes": None,
        "filters": [
            {
                "criteria": [
                    {
                        "filterType": 7,
                        "value": plugin_name,
                    }
                ],
                "direction": 2,
                "pageSize": 100,
                "pageNumber": 1,
                "sortBy": 0,
                "sortOrder": 0,
                "pagingToken": None
            }
        ],
        "flags": 2151
    })
    response = requests.request("POST", url, headers=headers, data=payload)
    resp = json.loads(response.text)
    latest_version_meta = resp['results'][0]['extensions'][0]['versions'][0]
    version = latest_version_meta['version']
    pack_name = f'{plugin_name.lower()}-{version}.vsix'
    files = latest_version_meta['files']
    resource_url = ''
    for f in files:
        if f['assetType'] == "Microsoft.VisualStudio.Services.VSIXPackage":
            resource_url = f['source']
    return {
        'version': version,
        'pack_name': pack_name,
        'resource_url': resource_url,
        'src': 'microsoft',
    }


def download_plugin(plugin_name: str):
    # try to download from open-vsx.org
    meta = get_meta_from_open_vsx(plugin_name=plugin_name)
    if not meta:
        meta = get_meta_from_microsoft(plugin_name=plugin_name)
    if isinstance(meta, dict):
        lock_meta.append(meta)
        download_file(meta['resource_url'], meta['pack_name'])
    else:
        print('not found in open-vsx or microsoft')


def handle_plugins(file_name: str = "./plugins.list") -> None:
    file = open(file_name)
    while True:
        line = file.readline().rstrip()
        if not line:
            break
        print(f'download plugin {line}')
        download_plugin(line)
    write_lock_meta()


def main():
    handle_plugins()


if __name__ == "__main__":
    main()