From c643ac5f6255f106e531586bf1025c3d3aa77d01 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Thu, 1 Jul 2021 19:56:40 +0200
Subject: [PATCH] controller can have defined AvalonMongoDB

---
 openpype/tools/new_publisher/control.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/openpype/tools/new_publisher/control.py b/openpype/tools/new_publisher/control.py
index 788126a206..50b2e755b5 100644
--- a/openpype/tools/new_publisher/control.py
+++ b/openpype/tools/new_publisher/control.py
@@ -13,11 +13,17 @@ from openpype.pipeline import (
 
 
 class PublisherController:
-    def __init__(self, headless=False):
+    def __init__(self, dbcon=None, headless=False):
         self.log = logging.getLogger("PublisherController")
         self.host = avalon.api.registered_host()
         self.headless = headless
 
+        if dbcon is None:
+            session = avalon.api.session_data_from_environment(True)
+            dbcon = avalon.api.AvalonMongoDB(session)
+        dbcon.install()
+        self.dbcon = dbcon
+
         self.creators = {}
         self.publish_plugins = []
         self.instances = []