diff --git a/reportIPBlock.py b/reportIPBlock.py index d5893a7..81b8a34 100644 --- a/reportIPBlock.py +++ b/reportIPBlock.py @@ -55,18 +55,22 @@ def main(): if dataBuffer: data = dataBuffer.pop(0) - - if config.loggingMode == 'rabbitmq': - - rabbitmq.publish(f"{data}") + rabbitmq.publish(f"{data}") else: - hostname = socket.gethostname() - ipAddress = socket.gethostbyname(hostname) - data = data.split('|') - manager = log.logsManager(config.sqlServer, config.sqlDatabase, config.sqlUsername, config.sqlPassword) - manager.insertIPBlock(hostname, ipAddress, data[1], data[5], data[6], data[2], config.ipinfoAPIToken) + while not (stopEvent.is_set() and not failureEvent.is_set()): + + time.sleep(1) + + if dataBuffer: + + data = dataBuffer.pop(0) + hostname = socket.gethostname() + ipAddress = socket.gethostbyname(hostname) + data = data.split('|') + manager = log.logsManager(config.sqlServer, config.sqlDatabase, config.sqlUsername, config.sqlPassword) + manager.insertIPBlock(hostname, ipAddress, data[1], data[5], data[6], data[2], config.ipinfoAPIToken) except Exception: failureEvent.set()