From b1d23c85032d393a88fd5ba6a0650db0f0b7d4cb Mon Sep 17 00:00:00 2001
From: warrior-hub <py6400370@gmail.com>
Date: Sat, 20 Dec 2025 20:20:40 +0530
Subject: [PATCH] Add configurable User-Agent for web requests

---
 vulnerablecode/http.py     | 8 ++++++++
 vulnerablecode/settings.py | 7 +++++++
 2 files changed, 15 insertions(+)
 create mode 100644 vulnerablecode/http.py

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"
+)