예제 #1
0
 def currentJob(job=None):
     import os
     if not job:
         if os.environ.has_key('PIPE_JOB'):
             job = os.environ['PIPE_JOB']
     if not job:
         return ''
     return "%s/%s" % (roots.jobs(), job)
예제 #2
0
    def __init__(self, projectID=None, projectName=None, adminUser="******"):
        if not projectID:
            if os.environ.has_key('PIPE_JOB'):
                tmp = os.environ['PIPE_JOB'].split('.')
                projectID = int(tmp[0])
                projectName = '.'.join(tmp[1:])
            else:
                raise Exception("ERROR: No Job Set!!")

        sudo.__init__(self)
        self.root = roots.jobs()
        self.proj = "%04d.%s" % (int(projectID), projectName.lower())
        self.__user = adminUser
        self.projPath = "%s/%s" % (self.root, self.proj)
예제 #3
0
파일: __init__.py 프로젝트: hradec/pipeVFX
def __go(args):
    import sys, os
    from glob import glob
    from  job import current as currentJob
    from  job import currentShot

    args = map( lambda x: x.lower(), args )

    if 'reset' in args or '-r' in args:
        return "\n".join([
            'unset PIPE_JOB',
            'unset PIPE_SHOT',
            '''PROMPT_COMMAND='echo -ne "\033]0;No Job/Shot set!!\007"' ''',
        ])

    reserved = ['asset', 'shot', 'shot']

    job = None
#    if os.environ.has_key('PIPE_JOB'):
#        job = os.environ['PIPE_JOB']

    shot = None
#    if os.environ.has_key('PIPE_SHOT'):
#        shot = os.environ['PIPE_SHOT']

    env = []
    # passed arguments
    if len(args)>1:

        if args[1].lower() not in reserved:
            jobs = filter( lambda x: args[1] in x, glob( "%s/*" % roots.jobs() ) )
            if not jobs:
                ret = ["echo 'ERROR: job %s doesnt exist! \n\nOptions are:'\n" % args[1].lower()]
                for each in glob( '%s/*' % (roots.jobs()) ):
                    beach = os.path.basename(each)
                    if beach[:4] != '0000':
                        ret.append( "echo '\t %s'" % beach )
                ret.append( "echo ' '")
                return "\n".join(ret)
            elif len(jobs)>1:
                ret = ["echo 'WARNING: There is more than one job with `%s` in it! \n\nOptions are:'\n" % args[1].lower()]
                for each in jobs:
                    beach = os.path.basename(each)
                    if beach[:4] != '0000':
                        ret.append( "echo '\t %s'" % beach )
                ret.append( "echo ' '")
                return "\n".join(ret)


            job = os.path.basename(jobs[0])
            env.append( 'export PIPE_JOB="%s"' % job )


        env.append( 'unset PIPE_SHOT' )
        for each in reserved:
            if each in args:
                value = ""
                try:
                    value = args[args.index(each)+1]
                    if not os.path.exists('%s/%s/%ss/%s' % (roots.jobs(), job, each, value) ):
                        raise
                    shot = "%s@%s" %  (each, value)
                    env.append( 'export PIPE_SHOT="%s"' %  shot )

                except:
                    ret = ["echo 'ERROR: %s %s doesnt exist!\n\nOptions are:'" % (each, value)]

                    for l in glob( '%s/%s/%ss/*' % (roots.jobs(), job, each) ):
                        ret.append( "echo '\t%s %s'" % (each, os.path.basename(l)) )
                    ret.append( "echo ' '")
                    return "\n".join(ret)

    if not job:
        return "\n".join([
            "echo 'ERROR: No current job set.'",
            '''PROMPT_COMMAND='echo -ne "\033]0;No Job/Shot set!!\007"' ''',
        ])


    # cd to job
    env.append( 'cd "%s"' % currentJob(job) )
    # add job tools/python to pythonpath
    path = '%s/tools/scripts' % currentJob(job)
    env.append( 'export PATH=%s:$__PATH' % path )
    pythonpath = '%s/tools/python' % currentJob(job)
    env.append( 'export PYTHONPATH=%s:$__PYTHONPATH' % pythonpath )

    # deal with shot
    values = ['Shot','']
    if shot:
        if os.path.exists( "%s/users/%s" % (currentShot(job, shot), admin.username()) ):
            env.append( 'cd "%s/users/%s"' % (currentShot(job, shot), admin.username()) )
        else:
            env.append( 'cd "%s/users"' % currentShot(job, shot) )
        values = shot.split('@')
        values[0] = "%s%s" % (values[0][0].upper(), values[0][1:])

        # add shot/user/tools/python to pythonpath
        path += ':%s/tools/scripts' % currentShot(job, shot)
        path += ':%s/users/%s/tools/scripts' % (currentShot(job, shot), admin.username())
        pythonpath += ':%s/tools/python' % currentShot(job, shot)
        pythonpath += ':%s/users/%s/tools/python' % (currentShot(job, shot), admin.username())


    env.append( 'export PATH=%s:$__PATH' % path )
    env.append( 'export PYTHONPATH=%s:$__PYTHONPATH' % pythonpath )

    env.append( 'echo "Job: %s\n%s: %s\nPath: $(pwd)"' % (job, values[0], values[1] ) )

    env.append( '''PROMPT_COMMAND='echo -ne "\033]0;Job: %s | %s: %s\007"' ''' % (job, values[0], values[1]) )


    # if running in Qube, set PIPE_FARM_USER
    if os.environ.has_key('QBJOBID'):
        user = ''
        for line in os.popen('/atomo/apps/linux/x86_64/qube/current/qube-core/local/pfx/qube/bin/qbjobs --expression "id==%s" --user "" ' % os.environ['QBJOBID']).readlines():
            if os.environ['QBJOBID'] in line:
                user = line[38:48].strip()
        env.append( 'export PIPE_FARM_USER=%s' % user )
        env.append( 'export PIPE_FARM_JOBID=%s' % os.environ['QBJOBID'] )

        
       
    return '%s \n %s' % (init(), '\n'.join(env))