find render version

find_version.py

"""
Find render version

import sys
module_path = '/home/shrimo/Desktop/course/vfx_dev/nuke'
if module_path not in sys.path:
    sys.path.append(module_path)
    
import find_version
reload(find_version)

find_version.main()

"""


import os


def find_version(render, root):
    file = os.path.splitext(root.split('/')[-1])
    shot = file[0].split('-')[0]
    elements = render.split('/')
    index = 0
    for element in elements:
        if shot == element:
            return elements[index+1]
        index += 1
    return False


def main():

    # root = nuke.root().knob('name').value()
    root = '/bstorage/strg01/mnt/projects/mavka/sequences/ep0050/ep0050_sh0010/cmp/work/nuke/ep0050_sh0010-comp.v003.nk'
    render = '/bstorage/strg01/mnt/projects/mavka/render/sequences/ep0050/ep0050_sh0010/v02/fg/ep0050-fg-albedo.0001.exr'
    # key_words = ['']
    print find_version(render, root)

main()