124 lines
4.5 KiB
Python
124 lines
4.5 KiB
Python
import logging
|
|
|
|
logging.basicConfig()
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
URL = 'https://docs.nvidia.com/vgpu/index.html'
|
|
|
|
BRANCH_STATUS_KEY = 'vGPU Branch Status'
|
|
VGPU_KEY, GRID_KEY, DRIVER_BRANCH_KEY = 'vGPU Software', 'vGPU Software', 'Driver Branch'
|
|
LINUX_VGPU_MANAGER_KEY, LINUX_DRIVER_KEY = 'Linux vGPU Manager', 'Linux Driver'
|
|
WINDOWS_VGPU_MANAGER_KEY, WINDOWS_DRIVER_KEY = 'Windows vGPU Manager', 'Windows Driver'
|
|
ALT_VGPU_MANAGER_KEY = 'vGPU Manager'
|
|
RELEASE_DATE_KEY, LATEST_KEY, EOL_KEY = 'Release Date', 'Latest Release in Branch', 'EOL Date'
|
|
JSON_RELEASES_KEY = '$releases'
|
|
|
|
|
|
def __driver_versions(html: 'BeautifulSoup'):
|
|
def __strip(_: str) -> str:
|
|
# removes content after linebreak (e.g. "Hello\n World" to "Hello")
|
|
_ = _.strip()
|
|
tmp = _.split('\n')
|
|
if len(tmp) > 0:
|
|
return tmp[0]
|
|
return _
|
|
|
|
# find wrapper for "DriverVersions" and find tables
|
|
data = html.find('div', {'id': 'driver-versions'})
|
|
items = data.find_all('bsp-accordion', {'class': 'Accordion-items-item'})
|
|
for item in items:
|
|
software_branch = item.find('div', {'class': 'Accordion-items-item-title'}).text.strip()
|
|
software_branch = software_branch.replace(' Releases', '')
|
|
matrix_key = software_branch.lower()
|
|
|
|
branch_status = item.find('a', href=True, string='Branch status')
|
|
branch_status = branch_status.next_sibling.replace(':', '').strip()
|
|
|
|
# driver version info from table-heads (ths) and table-rows (trs)
|
|
table = item.find('table')
|
|
ths, trs = table.find_all('th'), table.find_all('tr')
|
|
headers, releases = [header.text.strip() for header in ths], []
|
|
for trs in trs:
|
|
tds = trs.find_all('td')
|
|
if len(tds) == 0: # skip empty
|
|
continue
|
|
# create dict with table-heads as key and cell content as value
|
|
x = {headers[i]: __strip(cell.text) for i, cell in enumerate(tds)}
|
|
x.setdefault(BRANCH_STATUS_KEY, branch_status)
|
|
releases.append(x)
|
|
|
|
# add to matrix
|
|
MATRIX.update({matrix_key: {JSON_RELEASES_KEY: releases}})
|
|
|
|
|
|
def __debug():
|
|
# print table head
|
|
s = f'{VGPU_KEY:^13} | {LINUX_VGPU_MANAGER_KEY:^21} | {LINUX_DRIVER_KEY:^21} | {WINDOWS_VGPU_MANAGER_KEY:^21} | {WINDOWS_DRIVER_KEY:^21} | {RELEASE_DATE_KEY:>21} | {BRANCH_STATUS_KEY:^21}'
|
|
print(s)
|
|
|
|
# iterate over dict & format some variables to not overload table
|
|
for idx, (key, branch) in enumerate(MATRIX.items()):
|
|
for release in branch.get(JSON_RELEASES_KEY):
|
|
version = release.get(VGPU_KEY, release.get(GRID_KEY, ''))
|
|
linux_manager = release.get(LINUX_VGPU_MANAGER_KEY, release.get(ALT_VGPU_MANAGER_KEY, ''))
|
|
linux_driver = release.get(LINUX_DRIVER_KEY)
|
|
windows_manager = release.get(WINDOWS_VGPU_MANAGER_KEY, release.get(ALT_VGPU_MANAGER_KEY, ''))
|
|
windows_driver = release.get(WINDOWS_DRIVER_KEY)
|
|
release_date = release.get(RELEASE_DATE_KEY)
|
|
is_latest = release.get(VGPU_KEY) == branch.get(LATEST_KEY)
|
|
branch_status = __parse_branch_status(release.get(BRANCH_STATUS_KEY, ''))
|
|
|
|
version = f'{version} *' if is_latest else version
|
|
s = f'{version:<13} | {linux_manager:<21} | {linux_driver:<21} | {windows_manager:<21} | {windows_driver:<21} | {release_date:>21} | {branch_status:^21}'
|
|
print(s)
|
|
|
|
|
|
def __parse_branch_status(string: str) -> str:
|
|
string = string.replace('Production Branch', 'Prod. -')
|
|
string = string.replace('Long-Term Support Branch', 'LTS -')
|
|
|
|
string = string.replace('supported until', '')
|
|
|
|
string = string.replace('EOL since', 'EOL - ')
|
|
string = string.replace('EOL from', 'EOL -')
|
|
|
|
return string
|
|
|
|
|
|
def __dump(filename: str):
|
|
import json
|
|
|
|
file = open(filename, 'w')
|
|
json.dump(MATRIX, file)
|
|
file.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
MATRIX = {}
|
|
|
|
try:
|
|
import httpx
|
|
from bs4 import BeautifulSoup
|
|
except Exception as e:
|
|
logger.error(f'Failed to import module: {e}')
|
|
logger.info('Run "pip install beautifulsoup4 httpx"')
|
|
exit(1)
|
|
|
|
r = httpx.get(URL)
|
|
if r.status_code != 200:
|
|
logger.error(f'Error loading "{URL}" with status code {r.status_code}.')
|
|
exit(2)
|
|
|
|
# parse html
|
|
soup = BeautifulSoup(r.text, features='html.parser')
|
|
|
|
# build matrix
|
|
__driver_versions(soup)
|
|
|
|
# debug output
|
|
__debug()
|
|
|
|
# dump data to file
|
|
__dump('../app/static/driver_matrix.json')
|