Allow users to set their own user-agent.

sigmavirus24 · sigmavirus24 · commit 32196606d472 · 2012-10-15T14:20:37.000-04:00
github/developer.github.com@0f7a142
diff --git a/github3/github.py b/github3/github.py
@@ -993,6 +993,13 @@ def search_email(self, email):
         u = json.get('user', {})
         return LegacyUser(u, self) if u else None
 
+    def set_user_agent(self, user_agent):
+        """Allows the user to set their own user agent string to identify with
+        the API."""
+        if not user_agent:
+            return
+        self._session.config['base_headers']['User-Agent'] = user_agent
+
     def star(self, login, repo):
         """Star to login/repo
 
diff --git a/github3/models.py b/github3/models.py
@@ -67,7 +67,7 @@ def __init__(self, json, ses=None):
         self._session.headers.update({'Accept-Charset': 'utf-8'})
         # Identify who we are
         self._session.config['base_headers'].update(
-                {'User-Agent': 'github3.py/pre-alpha'})
+                {'User-Agent': 'github3.py/0.1b0'})
 
         # set a sane default
         self._github_url = 'https://api.github.com'
diff --git a/tests/test_github.py b/tests/test_github.py
@@ -444,6 +444,11 @@ def test_subscribe_unsub(self):
         expect(self._g.subscribe(*args)).isinstance(bool)
         expect(self._g.unsubscribe(*args)).isinstance(bool)
 
+    def test_set_user_agent(self):
+        ua = 'Foo Bar Bogus'
+        self.g.set_user_agent(ua)
+        expect(self.g._session.config['base_headers']['User-Agent']) == ua
+
     def test_star_unstar(self):
         args = (self.gh3py, self.test_repo)
         self.raisesGHE(self.g.star, *args)
