From 1ca674f33f18e7821aa47f039f0b808b50293b99 Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Wed, 23 Oct 2019 15:05:07 +0200 Subject: [PATCH] added atexit to custom db connector to run uninstall on exit --- pype/ftrack/lib/custom_db_connector.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pype/ftrack/lib/custom_db_connector.py b/pype/ftrack/lib/custom_db_connector.py index f4cc00444b..222a565ded 100644 --- a/pype/ftrack/lib/custom_db_connector.py +++ b/pype/ftrack/lib/custom_db_connector.py @@ -13,6 +13,7 @@ import logging import tempfile import functools import contextlib +import atexit import requests @@ -71,7 +72,7 @@ class DbConnector: """Establish a persistent connection to the database""" if self._is_installed: return - + atexit.register(self.uninstall) logging.basicConfig() self._mongo_client = pymongo.MongoClient( @@ -113,6 +114,7 @@ class DbConnector: self._mongo_client = None self._database = None self._is_installed = False + atexit.unregister(self.uninstall) def tables(self): """List available tables