VeloPythonAPI/VeloAPI.py at master · ACME-OUI/VeloPythonAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
'''
Created on Mar 16, 2015

@author: raju332
'''
import jpype
import time
from jpype import *
jvmPath=jpype.getDefaultJVMPath()

class Velo:
    def __init__(self):
        pass
    def start_jvm(self):
        #include the velo python API jar file here
        jpype.startJVM(jvmPath, "-Djava.class.path=C:/Users/raju332/Desktop/ACME/VeloAPI/JPype/VeloAPI.jar")
        global velo, cms, jobConfig, fileObj, tifConstants , fileServerMap
        velo=JPackage("velo").mgr.VeloManager
        cms=JPackage("gov").pnnl.velo.model.CmsPath
        jobConfig = JPackage("gov").pnnl.velo.tif.model.JobConfig
        tifConstants = JPackage("gov").pnnl.velo.util.VeloTifConstants

        fileServerMap = jpype.java.util.HashMap()

    def init_velo(self,username,password):
        global resMgr
        resMgr = velo.init(username, password)
        return resMgr
    def get_cms_service(self):
        cms = velo.getCmsService()
        return cms
    def get_job_status(self):
        cms_service = Velo.get_cms_service(self)

        status = cms_service.getProperty(con.getContextPath(), tifConstants.JOB_STATUS)
        starttime = cms_service.getProperty(con.getContextPath(), tifConstants.JOB_START_TIME)
        stoptime = cms_service.getProperty(con.getContextPath(), tifConstants.JOB_STOP_TIME)
        submittime = cms_service.getProperty(con.getContextPath(), tifConstants.JOB_SUBMIT_TIME)
        #print "Job Status:", status

    def create_folder(self, foldername): # create a folder
        homeFolder = Velo.get_homefolder(self)
        cmspath = cms(homeFolder).append(foldername)
        resMgr.createFolder(cmspath)
    def upload_file(self,location, filename): # upload file in velo
        fileObj = jpype.java.io.File(location+"/"+filename)
        homeFolder = Velo.get_homefolder(self)
        cmspath = homeFolder.append(filename)
        fileServerMap.put(fileObj,cmspath)
        #velo.uploadFile(location, filename , fileServerMap, None)
        resMgr.bulkUpload(fileServerMap, None)
    def launch_job(self, acmeusername): #launch the fake job
        secMgr = Velo.get_security_manager(self)
        config = jobConfig("fake_acme_job");
        config.setCmsUser(secMgr.getUsername())
        config.setCodeId("acmeworkflow")
        codereg = JPackage("gov").pnnl.velo.tif.service.CodeRegistry
        config.setCodeVersion(codereg.VERSION_DEFAULT)
        remoteDir = "/data/acme/velotestruns/fake_acme_case" #specify the fake_case directory
        config.setRemoteDir(remoteDir)

        remotedirfile = jpype.java.io.File(remoteDir).getName()

        now = time.strftime("%Y-%m-%d_%H-%M-%S")
        contextPathName = remotedirfile+"_"+now
        contextPath = Velo.get_homefolder(self).append(contextPathName)
        resMgr.createFolder(contextPath)
        time.sleep(5)
        print "folder created: ", contextPathName
        config.setContextPath(contextPath.toAssociationNamePath())
        config.setDoNotQueue(True)
        config.setJobId(contextPathName)
        config.setMachineId("localhost")
        # provide username for the acmetest server as the fake job is in acmetest
        config.setUserName(acmeusername)
        config.setPollingInterval(5)
        config.setLocalMonitoring(jpype.java.lang.Boolean(False))
        jobLaunchService = Velo.get_job_launch_service(self)
        global job_config
        job_config = jobLaunchService.launchJob(config, None)
        global con
        con = config
        #job_config = velo.launchJob()
        print "Fake Job submitted"
        return job_config
    def get_homefolder(self): # get user's home folder
        folder = resMgr.getHomeFolder()
        home_folder = folder.getPath();
        return home_folder
    def create_instance(self): # create tool instance in user's home folder
        velo.createToolInstance()
    def create_remotelink(self):
        velo.remoteLink()
    def get_job_launch_service(self):
        jobService = velo.getJobLaunchService()
        return jobService
    def get_security_manager(self):
        secmgr = velo.getSecurityManager()
        return secmgr
    def shutdown_jvm(self):
        jpype.shutdownJVM()