#!/usr/bin/python3
import argparse
import re
import sys

import requests
from bs4 import BeautifulSoup


def parse_args(argv):
    """Parse the command-line arguments"""
    parser = argparse.ArgumentParser(description="Get the PCI-SIG Vendor IDs")
    parser.add_argument(
        "--http-proxy", action="store", help="HTTP Proxy Address"
    )
    parser.add_argument(
        "--https-proxy", action="store", help="HTTPS Proxy Address"
    )
    args = parser.parse_args(argv)
    return args


def main(argv):
    """Go to the PCI-SIG members page and construct a
    dictionary of member companies to their Vendor IDs"""
    args = parse_args(argv)

    proxyDict = {"http": args.http_proxy, "https": args.https_proxy}
    page = "https://pcisig.com/membership/member-companies"
    pciVendorIDs = {}
    while True:
        r = requests.get(page, proxies=proxyDict)
        soup = BeautifulSoup(r.text)

        for row in soup.table.tbody.find_all("tr"):
            fields = row.find_all("td")
            vendorID = fields[1].text.strip()
            if "hex" in vendorID.lower():
                match = re.match(r"\w+ \((\w+) hex\)", vendorID, re.I)
                if match is not None:
                    vendorID = match.group(1)
                else:
                    vendorID = ""
            if vendorID != "":
                vendorName = fields[0].text.replace('"', "").strip()
                pciVendorIDs[vendorName] = vendorID

        page = soup.find("a", title="Go to next page")
        if page is None:
            break
        page = "https://pcisig.com" + page["href"]

    for name, vid in sorted(pciVendorIDs.items(), key=lambda x: x[0].lower()):
        print('{{0x{}, "{}"}},'.format(vid, name))


if __name__ == "__main__":
    main(sys.argv[1:])