Custom User-Agent header

mineo · Apr 16, 2016 · 7e3c6de · 7e3c6de
1 parent b0d84a2
commit 7e3c6de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/abzer/abzer.py b/abzer/abzer.py
@@ -67,7 +67,7 @@ def __init__(self, num_processes, essentia_path, profile_path, filenames):
         self.filenames = filenames
         self.num_processes = num_processes
         self.profile_path = profile_path
-        self.session = aiohttp.ClientSession()
+        self.session = aiohttp.ClientSession(headers=const.HEADERS)
         # Set the queue size to double the amount of processes. That way, more
         # filenames than subprocesses are in the queue, but not all of them
         # (which might be hundreds of thousands).

diff --git a/abzer/const.py b/abzer/const.py
@@ -2,6 +2,7 @@
 # coding: utf-8
 # Copyright © 2016 Wieland Hoffmann
 # License: MIT, see LICENSE for details
+from .version import version
 from os.path import expanduser, join
 
 HTTP_OK = 200
@@ -12,4 +13,5 @@
 LOGFILE = join(DEFAULT_PATH, "filelog.sqlite")
 DEFAULTS = {"path": "/usr/bin/streaming_extractor_music",
             "profile": join(DEFAULT_PATH, "profile.yaml")}
+HEADERS = {"User-Agent": "abzer %s (https://github.com/mineo/abzer)" % version}
 URL_BASE = "http://acousticbrainz.org/%s/low-level"