vulmon.py

import httpx
from bs4 import BeautifulSoup
from urllib.parse import urljoin
from colorama import init

init(autoreset=True)

def fetch_vulmon_data(cve_code):
    url = f"https://vulmon.com/vulnerabilitydetails?qid={cve_code}&sortby=bydate"
    data_list = []

    try:
        response = httpx.get(url, follow_redirects=True, timeout=10)

        if response.status_code == 200:
            soup = BeautifulSoup(response.text, 'html.parser')

            def get_element_text(element, default="LOST"):
                return element.text.strip() if element else default

            title = get_element_text(soup.select_one('h1.ui.header.column.jstitle1'))
            cvss_values = [value.text.strip() for value in soup.find_all('span', string=lambda text: text and ('CVSSv' in text))]
            description = get_element_text(soup.select_one('p.jsdescription1.content_overview'))
            meta = soup.select_one('div.meta')

            if meta:
                dates = meta.text.strip().split("Updated:")
                if len(dates) == 2:
                    published_date = dates[0].replace("Published:", "").strip()
                    updated_date = dates[1].strip()
                else:
                    published_date = updated_date = "LOST"
            else:
                published_date = updated_date = "LOST"

            links = [link['href'] for link in soup.select('div.ui.list.ex5 a[href]')]
            if not links:
                links = ["LOST"]

            vulmon_data = [
                title,
                cvss_values if cvss_values else ["LOST"],
                description,
                published_date,
                updated_date,
                links
            ]

            vulmon_cleaned_result = [value if value != "LOST" else "LOST" for value in vulmon_data]
            data_list.append(vulmon_cleaned_result)

        else:
            print(f"Error: Received status code {response.status_code}")
            data_list = [["LOST"] * 6]

    except httpx.RequestError as e:
        print(f"Request error: {e}")
        data_list = [["LOST"] * 6]

    except httpx.TimeoutException:
        print("Error: Request timeout.")
        data_list = [["LOST"] * 6]

    except Exception as e:
        print(f"Unexpected error: {e}")
        data_list = [["LOST"] * 6]

    return data_list