Colorspace Management and Distribution

common/openpype_common/distribution/file_handler.py

@@ -1,4 +1,3 @@
-import enlighten
 import os
 import re
 import urllib
@@ -252,22 +251,21 @@ def _get_confirm_token(response):
  if key.startswith('download_warning'):
  return value
 
+ # handle antivirus warning for big zips
+ found = re.search("(confirm=)([^&.+])", response.text)
+ if found:
+ return found.groups()[1]
+
  return None
 
  @staticmethod
  def _save_response_content(
  response_gen, destination,
  ):
  with open(destination, "wb") as f:
- pbar = enlighten.Counter(
- total=None, desc="Save content", units="%", color="green")
- progress = 0
  for chunk in response_gen:
  if chunk: # filter out keep-alive new chunks
  f.write(chunk)
- progress += len(chunk)
-
- pbar.close()
 
  @staticmethod
  def _quota_exceeded(first_chunk):

openpype/hosts/nuke/plugins/publish/extract_render_local.py

@@ -7,8 +7,7 @@
 from openpype.pipeline import publish
 
 
-class NukeRenderLocal(publish.Extractor):
- # TODO: rewrite docstring to nuke
+class NukeRenderLocal(publish.ExtractorColormanaged):
  """Render the current Nuke composition locally.
 
  Extract the result of savers by starting a comp render
@@ -22,6 +21,10 @@ class NukeRenderLocal(publish.Extractor):
  families = ["render.local", "prerender.local", "still.local"]
 
  def process(self, instance):
+ # get colorspace settings data
+ config_data, file_rules = self.get_colorspace_settings(
+ instance.context)
+
  families = instance.data["families"]
 
  node = None
@@ -67,6 +70,7 @@ def process(self, instance):
  )
 
  ext = node["file_type"].value()
+ colorspace = node["colorspace"].value()
 
  if "representations" not in instance.data:
  instance.data["representations"] = []
@@ -90,6 +94,14 @@ def process(self, instance):
  'files': filenames,
  "stagingDir": out_dir
  }
+
+ # inject colorspace data
+ self.set_representation_colorspace(
+ repre, instance.context,
+ config_data, file_rules,
+ colorspace=colorspace
+ )
+
  instance.data["representations"].append(repre)
 
  self.log.info("Extracted instance '{0}' to: {1}".format(