diff options
| -rw-r--r-- | src/scadere/check.py | 3 | ||||
| -rw-r--r-- | src/scadere/listen.py | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/src/scadere/check.py b/src/scadere/check.py index a14543d..8c3e60b 100644 --- a/src/scadere/check.py +++ b/src/scadere/check.py @@ -8,6 +8,7 @@ from base64 import urlsafe_b64encode as base64 from datetime import datetime, timedelta, timezone from email.utils import parsedate_to_datetime as parsedate from itertools import chain +from os.path import basename from socket import AF_INET, socket from ssl import create_default_context as tls_context from sys import argv, stderr, stdout @@ -81,7 +82,7 @@ def check(netlocs, after, output, fake_ca=None): base64_from_str(ca), file=output) -def main(prog=argv[0], arguments=argv[1:]): +def main(prog=basename(argv[0]), arguments=argv[1:]): """Run TLS checker.""" desc = ('Check TLS certificate expiration of HOST,' ' where PORT defaults to 443.\n\n' diff --git a/src/scadere/listen.py b/src/scadere/listen.py index 1ae7281..18f5323 100644 --- a/src/scadere/listen.py +++ b/src/scadere/listen.py @@ -10,6 +10,7 @@ from datetime import datetime, timezone from functools import lru_cache, partial from http import HTTPStatus from operator import call +from os.path import basename from pathlib import Path from string import digits from typing import assert_never @@ -298,7 +299,7 @@ def with_trailing_slash(base_url): return base_url if base_url.endswith('/') else f'{base_url}/' -def main(prog=argv[0], arguments=argv[1:]): +def main(prog=basename(argv[0]), arguments=argv[1:]): """Launch server.""" desc = ('Serve at URL Atom feeds for TLS certificate renewal reminder.' ' It is possible for clients to filter domains' |