Source code for requests_gracedb.user_agent

from ._version import get_versions


class SessionUserAgentMixin(object):
    """A mixin for :class:`requests.Session` to add a User-Agent header."""

    def __init__(self, **kwargs):
        super(SessionUserAgentMixin, self).__init__(**kwargs)
        version = get_versions()['version']
        self.headers['User-Agent'] = '{}/{}'.format(__package__, version)