code_gallery

Web News Blog Press Pricing

Learn

Training

Profiles

Executive

Templates

VFX

Code

Code Gallery Designer

v5.20

Maya

Check Ref Download

Jorge Hernandez Ibañez 07/10/2025

Tags
references download asset

Check Ref Download

Jorge Herna…

Action v1

07/10/2025

Maya

Pseudo-Code

Parameters

Talk with TD Support

Import to my Pipeline

Code

import maya.cmds as cmds
import os
import sys

from krakenPlugin.utils.api.files import get_file_data
from krakenPlugin.utils.downloads import file_requires_download, download_file_list

def get_maya_references():
    """
    Get all external Maya references in the current scene.
    
    Returns:
        list: List of reference file paths
    """
    try:
        # Get all reference nodes
        ref_nodes = cmds.ls(type='reference')
        ref_files = []
        
        for ref_node in ref_nodes:
            # Skip the sharedReferenceNode
            if ref_node == 'sharedReferenceNode':
                continue
                
            try:
                # Get the reference file path
                ref_file = cmds.referenceQuery(ref_node, filename=True, withoutCopyNumber=True)
                if ref_file and os.path.isabs(ref_file):
                    ref_files.append(ref_file)
            except Exception as e:
                print(f"Error querying reference {ref_node}: {e}")
                continue
                
        return ref_files
    except Exception as e:
        print(f"Error getting Maya references: {e}")
        return []

def get_references_file_data(ref_files):
    """
    Get file data for all reference files from the database.
    
    Args:
        ref_files (list): List of reference file paths
        
    Returns:
        list: List of file data dictionaries with metadata
    """
    extra_data = [
        "fileSize",
        "link", 
        "lastUpdate",
        "creationDate",
        "completed",
        "deleted"
    ]
    
    files_with_data = []
    
    for ref_file in ref_files:
        try:
            # Extract filename from path
            filename = os.path.basename(ref_file)
            
            # Get file data from database
            file_data = get_file_data(filename, extra_data)
            
            if file_data:
                # Add the local file path for download checking
                file_data['fileLocal'] = ref_file
                files_with_data.append(file_data)
            else:
                print(f"No file data found for reference: {filename}")
                
        except Exception as e:
            print(f"Error getting file data for {ref_file}: {e}")
            continue
            
    return files_with_data

def check_process():
    """
    Check if all Maya references are up to date.
    
    Returns:
        tuple: (check_result: bool, check_error: str or False)
    """
    try:
        # Get all Maya references
        ref_files = get_maya_references()
        
        if not ref_files:
            # No references found, check passes
            return True, False
            
        # Get file data for all references
        files_with_data = get_references_file_data(ref_files)
        
        if not files_with_data:
            # No file data found, can't check - this might be an error condition
            return False, "No file data found for any references"
            
        outdated_files = []
        
        # Check each file to see if it requires download
        for file_data in files_with_data:
            try:
                file_path = file_data.get('fileLocal')
                last_update = file_data.get('lastUpdate')
                
                if not file_path:
                    continue
                    
                # Check if file requires download (is outdated)
                requires_download = file_requires_download(file_path, last_update)
                
                if requires_download:
                    outdated_files.append(file_path)
                    
            except Exception as e:
                print(f"Error checking file {file_data.get('fileName', 'unknown')}: {e}")
                continue
        
        if outdated_files:
            error_msg = f"Found {len(outdated_files)} outdated reference(s): {', '.join([os.path.basename(f) for f in outdated_files])}"
            return False, error_msg
        else:
            return True, False
            
    except Exception as e:
        return False, f"Error checking references: {str(e)}"

def select_process():
    """
    Select outdated Maya references in the scene.
    
    Returns:
        tuple: (select_result: bool, select_error: str or False)
    """
    try:
        # Get all Maya references
        ref_files = get_maya_references()
        
        if not ref_files:
            return True, False
            
        # Get file data for all references
        files_with_data = get_references_file_data(ref_files)
        
        outdated_ref_nodes = []
        
        # Check each file and find corresponding reference nodes
        for file_data in files_with_data:
            try:
                file_path = file_data.get('fileLocal')
                last_update = file_data.get('lastUpdate')
                
                if not file_path:
                    continue
                    
                # Check if file requires download (is outdated)
                requires_download = file_requires_download(file_path, last_update)
                
                if requires_download:
                    # Find the reference node for this file
                    ref_nodes = cmds.ls(type='reference')
                    for ref_node in ref_nodes:
                        if ref_node == 'sharedReferenceNode':
                            continue
                        try:
                            ref_file = cmds.referenceQuery(ref_node, filename=True, withoutCopyNumber=True)
                            if ref_file == file_path:
                                outdated_ref_nodes.append(ref_node)
                                break
                        except:
                            continue
                            
            except Exception as e:
                print(f"Error checking file {file_data.get('fileName', 'unknown')}: {e}")
                continue
        
        if outdated_ref_nodes:
            # Select the outdated reference nodes
            cmds.select(outdated_ref_nodes, replace=True)
            return True, False
        else:
            # Clear selection if no outdated references
            cmds.select(clear=True)
            return True, False
            
    except Exception as e:
        return False, f"Error selecting outdated references: {str(e)}"

def fix_process():
    """
    Download and update all outdated Maya references.
    
    Returns:
        bool: True if fix was successful, False otherwise
    """
    try:
        # Get all Maya references
        ref_files = get_maya_references()
        
        if not ref_files:
            return True
            
        # Get file data for all references
        files_with_data = get_references_file_data(ref_files)
        
        outdated_files_data = []
        outdated_ref_nodes = []
        
        # Check each file and collect outdated ones
        for file_data in files_with_data:
            try:
                file_path = file_data.get('fileLocal')
                last_update = file_data.get('lastUpdate')
                
                if not file_path:
                    continue
                    
                # Check if file requires download (is outdated)
                requires_download = file_requires_download(file_path, last_update)
                
                if requires_download:
                    # Ensure file_data has required fields for download
                    if file_data.get('link') and file_data.get('completed', True) and not file_data.get('deleted', False):
                        outdated_files_data.append(file_data)
                        
                        # Find the corresponding reference node for reloading
                        ref_nodes = cmds.ls(type='reference')
                        for ref_node in ref_nodes:
                            if ref_node == 'sharedReferenceNode':
                                continue
                            try:
                                ref_file = cmds.referenceQuery(ref_node, filename=True, withoutCopyNumber=True)
                                if ref_file == file_path:
                                    outdated_ref_nodes.append(ref_node)
                                    break
                            except:
                                continue
                        
            except Exception as e:
                print(f"Error checking file {file_data.get('fileName', 'unknown')}: {e}")
                continue
        
        if outdated_files_data:
            print(f"Downloading {len(outdated_files_data)} outdated reference file(s)...")
            
            # Download all outdated files using the download utility
            download_results = download_file_list(
                outdated_files_data,
                max_threads=4,
                force_download=False,
                check_file_dates=True,
                override_publish=False
            )
            
            # Check if all downloads were successful
            error_count = sum(1 for result in download_results if result.startswith("Error"))
            
            if error_count > 0:
                print(f"Failed to download {error_count} file(s)")
                return False
            else:
                print("All outdated references have been updated successfully")
                
                # Reload/refresh the references that were updated
                if outdated_ref_nodes:
                    print(f"Reloading {len(outdated_ref_nodes)} reference(s) in Maya...")
                    
                    for ref_node in outdated_ref_nodes:
                        try:
                            # Reload the reference to get the updated file
                            cmds.file(loadReference=ref_node)
                            print(f"Reloaded reference: {ref_node}")
                        except Exception as e:
                            print(f"Warning: Failed to reload reference {ref_node}: {e}")
                            # Continue with other references even if one fails
                            continue
                    
                    print("Reference reloading completed")
                
                return True
        else:
            print("No outdated references found to fix")
            return True
            
    except Exception as e:
        print(f"Error fixing outdated references: {str(e)}")
        return False

#---------------------------------------------------------------------------------------------
def check(): # UI Functions, dont change
    check_result, check_error = check_process()
    
    if check_result:
        return True
    else:
        return check_error

def select():
    select_result, select_error = select_process()
    
    if select_result:
        return True
    else:
        return select_error

def fix(): # UI Functions, dont change
    fix_result = fix_process()
    check_result = check()
    return fix_result, check_result

Import Action

1. Copy Key

Copy

2. Go to your Action pannel

Go