Skip to content

Commit

Permalink
Add client tool to get single CVE
Browse files Browse the repository at this point in the history
  • Loading branch information
credbbl committed Dec 15, 2023
1 parent e6e9cd5 commit c35cd78
Show file tree
Hide file tree
Showing 2 changed files with 77 additions and 0 deletions.
1 change: 1 addition & 0 deletions src/glvd/cli/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

# Import to register all the commands
from . import ( # noqa: F401
client_cve,
combine_all,
combine_deb,
ingest_debsec,
Expand Down
76 changes: 76 additions & 0 deletions src/glvd/cli/client_cve.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
# SPDX-License-Identifier: MIT

from __future__ import annotations

import json
import logging
import sys
import urllib.parse
from datetime import datetime, timedelta, timezone
from typing import Any

import asyncio
from sqlalchemy import func, select
from sqlalchemy.dialects.postgresql import insert
from sqlalchemy.ext.asyncio import (
AsyncConnection,
AsyncEngine,
create_async_engine,
)

from ..database import Base, NvdCve
from ..util import requests
from . import cli


logger = logging.getLogger(__name__)


class ClientCve:
server: str

@staticmethod
@cli.register(
'cve',
arguments=[
cli.add_argument(
'cve',
help='the CVE to look up',
metavar='CVE',
),
cli.add_argument(
'--server',
default='http://localhost:5000',
help='the server to use',
),
cli.add_argument(
'--debug',
action='store_true',
help='enable debug output',
),
]
)
def run(cve: str, server: str, debug: bool) -> None:
logging.basicConfig(level=debug and logging.DEBUG or logging.INFO)
ClientCve(cve, server)()

def __init__(self, cve: str, server: str) -> None:
self.cve = cve
self.server = server

def __call__(self) -> None:
with requests.RetrySession() as rsession:
resp = rsession.get(
urllib.parse.urljoin(self.server, f'v1/cves/{self.cve}'),
)
if resp.status_code == 200:
data = resp.json()
json.dump(data, sys.stdout, indent=2)
elif resp.status_code == 404:
print(f'{self.cve} not found', file=sys.stderr)
else:
resp.raise_for_status()


if __name__ == '__main__':
ClientCve.run()

0 comments on commit c35cd78

Please sign in to comment.