Python loop / array

[mrwolfy]

Trying to build a simple script to attach keyframe data to a maya channel. This code works great to import raw keyframes from an external file to a maya channel .

 import maya.cmds as mc
 def keys():
       file = open('/path/to/keyframe/Data.txt', 'r')        
       lines = file.readlines()
       file.close()
       for i in range(len(lines)):
           mc.setKeyframe('pCube1', at='ty', v=float(lines[i]), t=i, itt='linear', ott='linear')
 keys()

What I would like to do is use a Maya command like ls -sl to put a number of nodes into an array and then attach animation data from external text files with corresponding names. So below would be an example, Maya Node on the left, keyframe data file on the right.

node: pCube1, file: 001.txt
node: pCube2, file: 002.txt
node: pCube3, file: 003.txt
and so on...

Any help? Thanks for reading!!!

[joojaa]

import maya.cmds as mc
def keys(lst):
     for name, fileN in lst:
         file = open(fileN, 'r')        
         lines = file.readlines()
         file.close()
         for i, line in enumerate(lines):
           mc.setKeyframe(name, at='ty', v=float(line), t=i, itt='linear', ott='linear')
 
lst=[
('pCube1','001.txt'),
('pCube2','002.txt'),
('pCube3','003.txt')
]
keys(lst)

but just out an idea wouldnt it be easier to call the filename cube.ty.txt and then do

import maya.cmds as mc
def keys(lst):
     for fileN in list:
         file = open(fileN, 'r')
         name = '.'.join(fileN.split('.')[:-1])
 
         lines = file.readlines()
         file.close()
         for i, line in enumerate(lines):
           mc.setKeyframe(name, v=float(line), t=i, itt='linear', ott='linear')
 
lst=['pCube1.ty.txt','pCube2.ty.txt','pCube3.tz.txt']
keys(lst)

[mrwolfy]

Thanks! I'm a bit of a newbie in python, so it's not as easy to dissect your code, but it's very helpful.

I'm looking for something that does what this code is supposed to do. This script is supposed grab a new text file from the directory, for each item in "objs". Seems not to work exactly, rather attatches the same data to each item selected.

 import os
 import maya.cmds as mc
 rootdir='/path'
 objs= mc.ls(sl=True)
 for o in objs:
     for subdir, dirs, files in os.walk(rootdir):
         for thisFile in files:
             file = open((os.path.join(rootdir, thisFile)), 'r')
             lines = file.readlines()
             file.close()
             for i in range(len(lines)):
                 mc.setKeyframe(o, at='ty', v=float(lines[i]), t=i, itt='linear', ott='linear')

[joojaa]

Well ista  bit hard to say anything definitive your attack on the problem is a bit badly structured, but i guess you want something like this. You should really not rely on code like this. It has too many variables that can crap up. Instead derive the name from the file somehow because otherswise it will sooner or later go wrong.

import os
import maya.cmds as mc
rootdir='/path'

objs= mc.ls(sl=True)
for subdir, dirs, files in os.walk(rootdir):
    for thisFile, o in zip(files, objs):
        file = open((os.path.join(rootdir, thisFile)), 'r')
        lines = file.readlines()
        file.close()
        for i in range(len(lines)):
            mc.setKeyframe(o, at='ty', v=float(lines[i]), t=i, itt='linear', ott='linear')

[mrwolfy]

worked like a charm! Thanks!

Instead derive the name from the file somehow because otherswise it will sooner or later go wrong.

I wil take your advice, as I learn more Python I can implement that idea.

[joojaa]

Tip: dont use  i in range(len(lines)) instead use i, line in enumarate(lines) it looks cleaner and is more practical