diff --git a/editgroups/settings/common.py b/editgroups/settings/common.py index 0ac52fa..8970cb6 100644 --- a/editgroups/settings/common.py +++ b/editgroups/settings/common.py @@ -244,4 +244,4 @@ } } - +USER_AGENT = "EditGroups (https://www.wikidata.org/wiki/Wikidata:Edit_groups)" diff --git a/store/stream.py b/store/stream.py index 583a080..d84f513 100644 --- a/store/stream.py +++ b/store/stream.py @@ -6,12 +6,15 @@ class WikiEditStream(object): def __init__(self): self.url = 'https://stream.wikimedia.org/v2/stream/recentchange' self.wiki = settings.WIKI_CODENAME + self.headers = { + "User-Agent": settings.USER_AGENT + } def stream(self, from_time=None): url = self.url if from_time is not None: url += '?since='+from_time.isoformat().replace('+00:00', 'Z') - for event in EventSource(url, timeout=30): + for event in EventSource(url, timeout=30, headers=self.headers): if event.event == 'message': try: change = json.loads(event.data) diff --git a/store/tests.py b/store/tests.py index fc8c449..14550fb 100644 --- a/store/tests.py +++ b/store/tests.py @@ -364,6 +364,7 @@ def test_stream(self): if idx > 10: break self.assertEquals('wikidatawiki', edit['wiki']) + self.assertEqual(s.headers["User-Agent"], "EditGroups (https://www.wikidata.org/wiki/Wikidata:Edit_groups)") class PagesTest(TestCase):