Ensure SSH tunnel is closed atexit

This commit is contained in:
2Shirt 2019-11-18 20:57:00 -07:00
parent 4b7e34eb17
commit 095c00c6ee
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -1,5 +1,6 @@
# Wizard Kit: Functions - osTicket
import atexit
import mysql.connector as mariadb
from functions.data import *
@ -50,7 +51,11 @@ class osTicket():
# Only open tunnel if one doesn't exist
if self.tunnel_proc is None or self.tunnel_proc.poll() is not None:
if self.tunnel_proc:
# Unregister previous terminate
atexit.unregister(self.tunnel_proc.terminate)
self.tunnel_proc = popen_program(cmd)
atexit.register(self.tunnel_proc.terminate)
# Connect to database
for x in range(5):