mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-04-28 06:28:56 +00:00
303bac630a
batch 2 - update Python idiom to 3.7 using pyupgrade (#11342)
* batch 2 - update Python idiom to 3.7 using pyupgrade
* Apply suggestions from code review
(cherry picked from commit 266d9d3fb0)
Co-authored-by: Alexei Znamensky <103110+russoz@users.noreply.github.com>
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Copyright (c) 2018, Dag Wieers (@dagwieers) <dag@wieers.com>
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncore
|
|
import os.path
|
|
import ssl
|
|
import sys
|
|
|
|
# Handle TLS and non-TLS support
|
|
try:
|
|
import smtpd_tls
|
|
|
|
HAS_TLS = True
|
|
except ImportError:
|
|
import smtpd
|
|
|
|
HAS_TLS = False
|
|
print("Library smtpd-tls is missing or not supported, hence starttls is NOT supported.")
|
|
|
|
# Handle custom ports
|
|
port = "25:465"
|
|
if len(sys.argv) > 1:
|
|
port = sys.argv[1]
|
|
ports = port.split(":")
|
|
if len(ports) > 1:
|
|
port1, port2 = int(ports[0]), int(ports[1])
|
|
else:
|
|
port1, port2 = int(port), None
|
|
|
|
# Handle custom certificate
|
|
basename = os.path.splitext(sys.argv[0])[0]
|
|
certfile = basename + ".crt"
|
|
if len(sys.argv) > 2:
|
|
certfile = sys.argv[2]
|
|
|
|
# Handle custom key
|
|
keyfile = basename + ".key"
|
|
if len(sys.argv) > 3:
|
|
keyfile = sys.argv[3]
|
|
|
|
ssl_ctx = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
|
|
|
if HAS_TLS and ssl_ctx is not None:
|
|
print(f"Using {certfile} and {keyfile}")
|
|
ssl_ctx.load_cert_chain(certfile=certfile, keyfile=keyfile)
|
|
|
|
print("Start SMTP server on port", port1)
|
|
smtp_server1 = smtpd_tls.DebuggingServer(("127.0.0.1", port1), None, ssl_ctx=ssl_ctx, starttls=True)
|
|
if port2:
|
|
print("Start TLS SMTP server on port", port2)
|
|
smtp_server2 = smtpd_tls.DebuggingServer(("127.0.0.1", port2), None, ssl_ctx=ssl_ctx, starttls=False)
|
|
else:
|
|
print("Start SMTP server on port", port1)
|
|
smtp_server1 = smtpd.DebuggingServer(("127.0.0.1", port1), None) # pylint: disable=used-before-assignment
|
|
if port2:
|
|
print("WARNING: TLS is NOT supported on this system, not listening on port %s." % port2)
|
|
|
|
asyncore.loop()
|