1# 

2# Copyright (C) 2019-2020 Leo P. Singer <leo.singer@ligo.org> 

3# 

4# This program is free software: you can redistribute it and/or modify 

5# it under the terms of the GNU General Public License as published by 

6# the Free Software Foundation, either version 3 of the License, or 

7# (at your option) any later version. 

8# 

9# This program is distributed in the hope that it will be useful, 

10# but WITHOUT ANY WARRANTY; without even the implied warranty of 

11# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

12# GNU General Public License for more details. 

13# 

14# You should have received a copy of the GNU General Public License 

15# along with this program. If not, see <https://www.gnu.org/licenses/>. 

16# 

17from ._version import get_versions 

18 

19 

20class SessionUserAgentMixin(object): 

21 """A mixin for :class:`requests.Session` to add a User-Agent header.""" 

22 

23 def __init__(self, **kwargs): 

24 super(SessionUserAgentMixin, self).__init__(**kwargs) 

25 version = get_versions()['version'] 

26 self.headers['User-Agent'] = '{}/{}'.format(__package__, version)