SetKeys Python script for Maya

## http://flynnsyard.blogspot.co.uk ##
## Set keys script in python ##

## Use this script to ensure all selected items and or channels have a key based on what you see in the timeline (or even highlighted timeline!) of maya ##
## Also can be used to remove frames and or keys from all selected ##
## enjoy ##


import maya.cmds as cmds
Yel_Btn_Col = ( .84 , .76 , .31 )
Yel2_Btn_Col = ( .98 , .56 , .11 )

def SetKeys_Win_Cmd():
    SetKeys_Win = "SetKeys_Win"

    if cmds.window('SetKeys_Win', exists=True):
        cmds.deleteUI('SetKeys_Win', window=True)
        cmds.windowPref( 'SetKeys_Win', remove=True )
    cmds.window( 'SetKeys_Win' ,  rtf=True , toolbox=True , widthHeight = [125 , 70] )

    cmds.rowColumnLayout( 'SkinForce_MainWin' , numberOfColumns=1  )
    cmds.button( bgc= Yel_Btn_Col , label = 'SET KEYS', width = 125 , height = 30 , command = 'SetKeys_cmd( 0 )' )
    cmds.button( bgc= Yel2_Btn_Col , label = 'Del every odd key', width = 125 , height = 40 , command = 'SetKeys_cmd( 1 )' )

    cmds.showWindow( 'SetKeys_Win' )


def Items_Selected():   #returns selected + error msgs
    selcted= []
    selcted = cmds.ls( selection=True , flatten=True)
    if selcted is None:
        pass
    else:
        return selcted


def SetKeysmainchunk_cmd( type ):

    List_Sel = Items_Selected()
    Source = List_Sel[0]
    TimeCurrent = cmds.currentTime(query = True )
    #grab the timeline
    Time_Selected = cmds.timeControl( 'timeControl1' , query=True , rv=True )

    if Time_Selected is True:
        Time = cmds.timeControl( 'timeControl1' , query=True , rng=True )
        tmp1 = Time.replace('"' , '')
        tmp2 = tmp1.split(':')
        Time_Start = tmp2[0]
        Time_End = tmp2[1]
        Time_End = (int(Time_End) -1)
    else:
        Time_Start = cmds.playbackOptions(query=True  ,  minTime=True )
        Time_End = cmds.playbackOptions(query=True  ,  maxTime=True )

    #check if item 1 has keys, it should!
    Has_Keys_Test = cmds.keyframe( Source , query=True ,kc=True)

    #has to have keys to continue!
    if Has_Keys_Test is None:
        print 'No keys found, add keys to continue'
    else:
        Source_Keys_All = []
        for items in List_Sel:
            tmp = cmds.keyframe(  items , query=True , time=( Time_Start , Time_End )  )
            if tmp is None:
                pass
            else:
                for times in tmp:
                    Source_Keys_All.append( times )


        Source_Keys = set(Source_Keys_All)
        Source_Keys2 = []
        for x in Source_Keys:
            Source_Keys2.append(x)

        Source_Keys2.sort()

        cmds.select(clear=True)
        cmds.select(List_Sel)

        if type is 0:
            TimeCount = 0
        if type is 1:
            TimeCount = 1

        while TimeCount < len(Source_Keys2):
            CurrentKeyFrame = Source_Keys2[TimeCount]
            cmds.currentTime( CurrentKeyFrame )

            if type is 0:
                cmds.setKeyframe( List_Sel )
                TimeCount +=1   #increments the time by every key found
            if type is 1:
                print CurrentKeyFrame
                cmds.cutKey( List_Sel ,  time=(CurrentKeyFrame,) )
                TimeCount +=2   #increments the time by every other key

            # count +=1 #increments the objects selected
    cmds.currentTime( TimeCurrent )



def SetKeys_cmd( type ):
    test = Items_Selected()

    if test is None:
        print 'select more than one item to set keys on.'
    else:
        if len(test) < 1:
            print 'select more than one item to set keys on.'
        else:


            if type is 1:
                gChannelBoxName = mel.eval('$temp=$gChannelBoxName')
                chList = cmds.channelBox (gChannelBoxName, q=True, sma = True)
                if chList:
                    print 'cutting keys from channels on selected'
                    CutKeysChannel_cmd()
                else:
                    print 'cutting keys on selected'
                    SetKeysmainchunk_cmd( type )
            else: #type is 0
                print 'setting keys on all selected'
                SetKeysmainchunk_cmd( type )




def CutKeysChannel_cmd():

    List_Sel = Items_Selected()


    #grab the timeline
    Time_Selected = cmds.timeControl( 'timeControl1' , query=True , rv=True )
    TimeCurrent = cmds.currentTime(query = True )

    if Time_Selected is True:
        Time = cmds.timeControl( 'timeControl1' , query=True , rng=True )
        tmp1 = Time.replace('"' , '')
        tmp2 = tmp1.split(':')
        Time_Start = tmp2[0]
        Time_End = tmp2[1]
        Time_End = (int(Time_End) -1)
    else:
        Time_Start = cmds.playbackOptions(query=True  ,  minTime=True )
        Time_End = cmds.playbackOptions(query=True  ,  maxTime=True )
    #grab the timeline


    for x in List_Sel:

        gChannelBoxName = mel.eval('$temp=$gChannelBoxName')
        chList = cmds.channelBox (gChannelBoxName, q=True, sma = True)

        for channels in chList:
            tmp = cmds.keyframe( x , attribute = channels , query=True ,  time=( Time_Start , Time_End )  )

            print tmp
            Source_Keys = set(tmp)
            Source_Keys2 = []
            for time in Source_Keys:
                Source_Keys2.append( time )

            Source_Keys2.sort()

            cmds.select(clear=True)
            cmds.select( x )

            TimeCount = 1
            while TimeCount < len(Source_Keys2):
                CurrentKeyFrame = Source_Keys2[TimeCount]
                cmds.currentTime( CurrentKeyFrame )


                cmds.cutKey( List_Sel ,  attribute= channels , time=(CurrentKeyFrame,) )
                TimeCount +=2   #increments the time by every other key


    cmds.currentTime( TimeCurrent )




SetKeys_Win_Cmd()