server/net: make the user-agent configurable

Fixes #127

1 files changed, 3 insertions, 0 deletions

diff --git a/server/szurubooru/func/net.py b/server/szurubooru/func/net.py
index a6e1821..e6326c0 100644
--- a/server/szurubooru/func/net.py
+++ b/server/szurubooru/func/net.py
@@ -1,10 +1,13 @@
 import urllib.request
+from szurubooru import config
 from szurubooru import errors
 
 
 def download(url: str) -> bytes:
     assert url
     request = urllib.request.Request(url)
+    if config.config['user_agent']:
+        request.add_header('User-Agent', config.config['user_agent'])
     request.add_header('Referer', url)
     try:
         with urllib.request.urlopen(request) as handle: