def handleVersionUp(self, event):
upVersionPath = self.getFilePath(True)
if not upVersionPath:
print "Can\'t find anything file"
return
node = commands.sourcesAtFrame(commands.frame())
commands.setSourceMedia(node[0], [upVersionPath], "")
def handleVersionDown(self, event):
downVersionPath = self.getFilePath(False)
if not allPath:
print "Can\'t find anything file"
return
node = commands.sourcesAtFrame(commands.frame())
commands.setSourceMedia(node[0], [downVersionPath], "")
def getCurrentColorNode(self):
check = commands.nodesOfType('RVSource')
if check:
sourceNode = extra_commands.sourceMetaInfoAtFrame(commands.frame(), commands.viewNode())
nodes = commands.nodesOfType('RVColor')
for node in nodes:
if node.startswith(sourceNode.get('node','').split('_')[0]):
_log.info('found %s' % node)
return node
else:
return
def _get_node_for_source(self, node_type):
"""
Finds the node of the given type for the currently viewed
source.
:param str node_type: RV node type, eg. RVColor, RVCDL
"""
# first find the source for the frame we're currently viewing
frame = commands.frame()
# file_source = None
for source in extra_commands.nodesInEvalPath(frame, "RVFileSource", None):
file_source = source
continue
source_group = commands.nodeGroup(file_source)
# now find the node of the given type for that source
pipe_node = group_member_of_type(source_group, "RVLookPipelineGroup")
return group_member_of_type(pipe_node, node_type)
def getFilePath(self, up=True):
node = commands.sourcesAtFrame(commands.frame())
if not node:
return
SourceNodeMedia = commands.sourceMediaInfoList(node[0])
currentVersion = None
for media in SourceNodeMedia:
currentVersion = media["file"]
result = re.findall(r".+_(v\d\d).*", currentVersion)
if not result:
print u"file type error %s"%currentVersion
return
rePath = currentVersion.replace(result[0], "v??")
upVersionPath = None
if "#." not in rePath:
allPath = glob.glob(rePath)
allPath.sort()
for index in range(len(allPath)):
if allPath[index].replace("\\", "/") == currentVersion:
if up:
if index == len(allPath) - 1:
print u"已经是最大版本了"
upVersionPath = currentVersion
else:
upVersionPath = allPath[index+1]
else:
if index == 0:
print u"已经是最小版本了"
upVersionPath = currentVersion
else:
upVersionPath = allPath[index-1]
else:
rep = re.findall(r".+\.([\d\#-]+)\..+", rePath)
if not rep:
print u"file type error %s" % currentVersion
return
rePath = rePath.replace(rep[0], "*")
allPath = glob.glob(rePath)
allPath.sort()
allPathList = list()
start = os.path.dirname(allPath[0]).replace("\\", "/")
temp = list()
for unit in allPath:
if os.path.dirname(unit).replace("\\", "/") == start:
temp.append(unit)
else:
start = os.path.dirname(unit).replace("\\", "/")
allPathList.append(temp)
temp = [unit]
if temp:
allPathList.append(temp)
seqPath = list()
for unit in allPathList:
firstFrame = re.findall(r".+\.([\d\#-]+)\..+", unit[0])[0]
lastFrame = re.findall(r".+\.([\d\#-]+)\..+", unit[-1])[0]
if firstFrame == lastFrame:
seqPath.append(unit[0])
else:
seqPath.append(unit[0].replace(firstFrame, "%s-%s#"%(firstFrame, lastFrame)).replace("\\", "/"))
for index in range(len(seqPath)):
if seqPath[index] == currentVersion:
if up:
if index == len(seqPath) - 1:
print u"this is min version"
upVersionPath = currentVersion
else:
upVersionPath = seqPath[index+1]
else:
if index == 0:
print u"this is mix version"
upVersionPath = currentVersion
else:
upVersionPath = seqPath[index-1]
return upVersionPath
def get_sources():
return [Source(source) for source in commands.sourcesAtFrame(commands.frame())]
