在while循环中尝试和除外

Question

仍然是编程新手，我正在尝试创建一个脚本来检测远程主机上的端口不可用并更改 iptables 并发送单个通知。

我正在运行 Ubuntu 16.04，我正在使用 Python3，我正在使用 os 模块发送电子邮件和更改 IPtables

我几乎可以让它工作，但是，尝试将它发送到电子邮件一次是一个挑战。

import socket
import sys
import time
import datetime
import os
import logging

#Variables
remote_host = "syslog1"
drop = os.system("iptables-restore < iptables.conf")
undrop = os.system("iptables-restore < DR-iptables.conf")
print ('Starting Syslog DR Plan.')

# run continuously

while True:
    time.sleep(1)
    for remote_port in [514]:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(60)
        now = datetime.datetime.now()
        now_text = now.strftime("%Y-%m-%d %H:%M:%S")
        plco = 0

        try:
            sock.connect((remote_host, remote_port))
            print((now_text, 'Port is Open on', remote_port))


        except:
            socket.timeout
            plco = 1

    if plco == 1:
        undrop
        logging.basicConfig(filename='/var/log/syslog',level=logging.DEBUG , format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
        logging.error('#### Syslog DR Invoked - Immediate action required ####')
        os.system("mail -s '# INVESTIGATE IMMEDIATELY - SYSLOG1 port 514 unreachable #' alert@test.com < ./panic.txt")
        print( 'Port is closed')

    else:
        pass
sock.close()

我怀疑我需要做两件事中的一件，但到目前为止都失败了。一种是将其定义为一个函数，然后调用该函数，然后根据输出将另一个函数设置为电子邮件。

任何人都可以建议去哪里，或者更好的方法来做到这一点（我可能会以困难/错误的方式做事）。

非常感谢

Answer 1

您需要在尝试连接到端口的套接字超时时发送电子邮件的代码。这可以通过使用您的代码来实现：

import socket
import sys
import time
import datetime
import os
import logging

logging.basicConfig(filename='/var/log/syslog', level=logging.DEBUG,
                    format='%(asctime)s %(message)s',
                    datefmt='%m/%d/%Y %I:%M:%S %p')

log = logging.getLogger(__name__)

escalated = False
while True:
    time.sleep(1)
    for remote_port in [514]:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(60)
        now = datetime.datetime.now()
        now_text = now.strftime("%Y-%m-%d %H:%M:%S")


        try:
            sock.connect((remote_host, remote_port))
            print((now_text, 'Port is Open on', remote_port))
            escalated = False


        except (socket.timeout, ConnectionRefusedError):
            if not escalated: 
                log.error('#### Syslog DR Invoked - Immediate action required ####')
                os.system("mail -s '# INVESTIGATE IMMEDIATELY - SYSLOG1 port 514 "
                      "unreachable #' alert@test.com < ./panic.txt")
                escalated = True
            print('Port is closed')

    sock.close()