fix: add User-Agent as per WMF policy (#157)

* fix: add User-Agent as per WMF policy

* fix: add headers parameter to EventSource

Author: arcstur

editgroups/settings/common.py

@@ -244,4 +244,4 @@
     }
 }
 
-
+USER_AGENT = "EditGroups (https://www.wikidata.org/wiki/Wikidata:Edit_groups)"

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)

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):