diff --git a/vulnerablecode/http.py b/vulnerablecode/http.py new file mode 100644 index 000000000..a5ab61b26 --- /dev/null +++ b/vulnerablecode/http.py @@ -0,0 +1,8 @@ +import requests +from vulnerablecode.settings import VULNERABLECODE_USER_AGENT + + +def get(url, **kwargs): + headers = kwargs.pop("headers", {}) + headers.setdefault("User-Agent", VULNERABLECODE_USER_AGENT) + return requests.get(url, headers=headers, **kwargs) diff --git a/vulnerablecode/settings.py b/vulnerablecode/settings.py index 05a3d0fa8..637f6ede3 100644 --- a/vulnerablecode/settings.py +++ b/vulnerablecode/settings.py @@ -385,3 +385,10 @@ "DEFAULT_TIMEOUT": env.int("VULNERABLECODE_REDIS_DEFAULT_TIMEOUT", default=3600), } } + +import os + +VULNERABLECODE_USER_AGENT = os.getenv( + "VULNERABLECODE_USER_AGENT", + "VulnerableCode" +)