75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
import requests
|
|
import re
|
|
from bs4 import BeautifulSoup
|
|
import sys
|
|
import lzma
|
|
from urllib.parse import urlparse
|
|
from contextlib import suppress
|
|
import os
|
|
from os import path
|
|
from tqdm import tqdm
|
|
import shutil
|
|
import time
|
|
from pathlib import Path
|
|
|
|
def fucked_up_get(*args, **kwargs):
|
|
for retry in range(5):
|
|
for timeout in [0.1, 0.5, 1.0, 2.0, 5.0]:
|
|
kwargs.update({'timeout': timeout})
|
|
try:
|
|
return requests.get(*args, **kwargs)
|
|
except requests.exceptions.Timeout:
|
|
pass
|
|
time.sleep(0.5)
|
|
raise requests.exceptions.Timeout()
|
|
|
|
def crop(s, length=80):
|
|
if len(s) <= length:
|
|
return s + (' ' * (length-len(s)))
|
|
|
|
if length > 40:
|
|
return s[:length - 20 - 3] + '...' + s[-20:]
|
|
|
|
return s[:length]
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-u', '--update-site', default='http://sw-center.st.com/stm32cubeide/updatesite1')
|
|
parser.add_argument('artifact_id')
|
|
parser.add_argument('-v', '--version', action='store_true')
|
|
parser.add_argument('--write-version', type=argparse.FileType('w'))
|
|
args = parser.parse_args()
|
|
|
|
if args.write_version:
|
|
args.version = True
|
|
version_file = args.write_version
|
|
else:
|
|
version_file = sys.stdout
|
|
|
|
if args.version:
|
|
print(f'[Artifact {args.artifact_id}]', file=version_file)
|
|
|
|
res = fucked_up_get(f'{args.update_site}/compositeContent.xml')
|
|
res.raise_for_status()
|
|
soup = BeautifulSoup(res.content, features='html.parser')
|
|
latest_version = soup.find('repository')['version']
|
|
if args.version:
|
|
print('IDE version:', latest_version, file=version_file)
|
|
|
|
res = fucked_up_get(f'{args.update_site}/{latest_version}/artifacts.xml.xz')
|
|
res.raise_for_status()
|
|
soup = BeautifulSoup(lzma.LZMADecompressor().decompress(res.content), features='html.parser')
|
|
|
|
artifact = soup.find('artifact', recursive=True, id=args.artifact_id)
|
|
art_id, art_ver, art_cls = artifact['id'], artifact['version'], artifact['classifier']
|
|
if args.version:
|
|
print('Artifact version:', artifact['version'], file=version_file)
|
|
|
|
filename = f'{art_id}_{art_ver}.jar'
|
|
url = f'{args.update_site}/{latest_version}/plugins/{filename}'
|
|
|
|
if not args.version or args.write_version:
|
|
print(f'{url}')
|
|
|