diff --git a/test/create_driver_matrix_json.py b/test/create_driver_matrix_json.py index 2047d2e..a2afd9f 100644 --- a/test/create_driver_matrix_json.py +++ b/test/create_driver_matrix_json.py @@ -6,7 +6,7 @@ logger.setLevel(logging.INFO) URL = 'https://docs.nvidia.com/vgpu/index.html' -BRANCH_STATUS_KEY, SOFTWARE_BRANCH_KEY, = 'vGPU Branch Status', 'vGPU Software Branch' +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' @@ -26,12 +26,15 @@ def __driver_versions(html: 'BeautifulSoup'): # find wrapper for "DriverVersions" and find tables data = html.find('div', {'id': 'driver-versions'}) - items = data.findAll('bsp-accordion', {'class': 'Accordion-items-item'}) + 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') @@ -42,48 +45,20 @@ def __driver_versions(html: 'BeautifulSoup'): 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 __release_branches(html: 'BeautifulSoup'): - # find wrapper for "AllReleaseBranches" and find table - data = html.find('div', {'id': 'all-release-branches'}) - table = data.find('table') - - # branch releases info from table-heads (ths) and table-rows (trs) - ths, trs = table.find_all('th'), table.find_all('tr') - headers = [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]: cell.text.strip() for i, cell in enumerate(tds)} - - # get matrix_key - software_branch = x.get(SOFTWARE_BRANCH_KEY) - matrix_key = software_branch.lower() - - # add to matrix - MATRIX.update({matrix_key: MATRIX.get(matrix_key) | x}) - - def __debug(): # print table head - s = f'{SOFTWARE_BRANCH_KEY:^21} | {BRANCH_STATUS_KEY:^21} | {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} | {EOL_KEY:>21}' + 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()): - branch_status = branch.get(BRANCH_STATUS_KEY) - branch_status = branch_status.replace('Branch ', '') - branch_status = branch_status.replace('Long-Term Support', 'LTS') - branch_status = branch_status.replace('Production', 'Prod.') - - software_branch = branch.get(SOFTWARE_BRANCH_KEY).replace('NVIDIA ', '') 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, '')) @@ -92,13 +67,25 @@ def __debug(): 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 - eol = branch.get(EOL_KEY) if is_latest else '' - s = f'{software_branch:^21} | {branch_status:^21} | {version:<13} | {linux_manager:<21} | {linux_driver:<21} | {windows_manager:<21} | {windows_driver:<21} | {release_date:>21} | {eol:>21}' + 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 @@ -128,7 +115,6 @@ if __name__ == '__main__': # build matrix __driver_versions(soup) - __release_branches(soup) # debug output __debug()