Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2026-01-01 16:34:53 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

io_nonsingleton is same as avalon's io

Browse source
This commit is contained in:
iLLiCiTiT 2020-06-21 15:52:04 +02:00
parent b4b3ed4bf4
commit d185b5d54c
2 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

24
pype/modules/adobe_communicator/lib/io_nonsingleton.py
View file

@ -16,7 +16,7 @@ import contextlib
from avalon import schema
from avalon.vendor import requests
from pype.api import get_default_components, compose_url
from avalon.io import extract_port_from_url
# Third-party dependencies
import pymongo
@ -73,15 +73,17 @@ class DbConnector(object):
self.Session.update(self._from_environment())
timeout = int(self.Session["AVALON_TIMEOUT"])
mongo_url = self.Session["AVALON_MONGO"]
kwargs = {
"host": mongo_url,
"serverSelectionTimeoutMS": timeout
}
components = get_default_components()
port = components.pop("port")
host = compose_url(**components)
self._mongo_client = pymongo.MongoClient(
host=host,
port=port,
serverSelectionTimeoutMS=timeout
)
port = extract_port_from_url(mongo_url)
if port is not None:
kwargs["port"] = int(port)
self._mongo_client = pymongo.MongoClient(**kwargs)
for retry in range(3):
try:
@ -389,6 +391,10 @@ class DbConnector(object):
if document is None:
break
if document.get("type") == "master_version":
_document = self.find_one({"_id": document["version_id"]})
document["data"] = _document["data"]
parents.append(document)
return parents

21
pype/modules/ftrack/lib/io_nonsingleton.py
View file

@ -16,6 +16,7 @@ import contextlib
from avalon import schema
from avalon.vendor import requests
from avalon.io import extract_port_from_url
# Third-party dependencies
import pymongo
@ -72,11 +73,17 @@ class DbConnector(object):
self.Session.update(self._from_environment())
timeout = int(self.Session["AVALON_TIMEOUT"])
self._mongo_client = pymongo.MongoClient(
host=os.environ["AVALON_MONGO_HOST"],
port=int(os.environ["AVALON_MONGO_PORT"]),
serverSelectionTimeoutMS=timeout
)
mongo_url = self.Session["AVALON_MONGO"]
kwargs = {
"host": mongo_url,
"serverSelectionTimeoutMS": timeout
}
port = extract_port_from_url(mongo_url)
if port is not None:
kwargs["port"] = int(port)
self._mongo_client = pymongo.MongoClient(**kwargs)
for retry in range(3):
try:
@ -384,6 +391,10 @@ class DbConnector(object):
if document is None:
break
if document.get("type") == "master_version":
_document = self.find_one({"_id": document["version_id"]})
document["data"] = _document["data"]
parents.append(document)
return parents