diff --git a/linkstatus/linkstatus.py b/linkstatus/linkstatus.py index 3f2517d..fa80773 100644 --- a/linkstatus/linkstatus.py +++ b/linkstatus/linkstatus.py @@ -3,6 +3,7 @@ from shutil import get_terminal_size import click +import pkg_resources import requests from linkstatus.parser import link_validator @@ -20,11 +21,13 @@ def link_status(link, timeout=5): Returns: tuple of status (bool) and status code """ - + headers = { + "User-Agent": "linkstatus/{}".format(pkg_resources.get_distribution("linkstatus").version) + } try: - status_code = requests.get(link, timeout=timeout).status_code + status_code = requests.get(link, headers=headers, timeout=timeout).status_code except requests.exceptions.SSLError: - status_code = requests.get(link, verify=False, timeout=timeout).status_code + status_code = requests.get(link, verify=False, headers=headers, timeout=timeout).status_code except Exception: # noqa # TODO: include exception in logging status_code = None