Add files via upload

Author: Loong1992

Deadline/Data/deadlineData.xml

@@ -0,0 +1,13 @@
+<data>
+	<settings>
+		<dept></dept>
+		<grp>cml_blade</grp>
+		<pool>show_cml</pool>
+		<s_pool>show_cml</s_pool>
+		<prio>58</prio>
+		<fpt>2</fpt>
+		<threads>24</threads>
+		<initialStatus>Suspended</initialStatus>
+		<frame>FF</frame>
+	</settings>
+</data>

Deadline/Deadline_API/Balancer.py

@@ -0,0 +1,113 @@
+from ConnectionProperty import ConnectionProperty
+import json
+
+class Balancer:
+    """
+        Class used by DeadlineCon to send Balancer requests. 
+        Stores the address of the web service for use in sending requests.
+    """    
+    def __init__(self, connectionProperties):
+        self.connectionProperties = connectionProperties
+        
+    def GetBalancerNames(self):
+        """ Gets all the balancer names.
+        Returns: The list of balancer names
+        """
+        return self.connectionProperties.__get__("/api/balancer?NamesOnly=true")
+
+    def GetBalancerInfo(self, name):
+        """ Gets a balancer info.
+            Input: name: The balancer name
+            Returns: The balancer info.
+        """
+        return self.connectionProperties.__get__("/api/balancer?Name="+name.replace(' ','+')+"&Info=true")
+
+    def GetBalancerInfos(self, names = None):
+        """ Gets a list of balancer infos.
+            Input: name: The balancer names to be retrieved. If None then gets all balancer infos
+            Returns: The balancer infos.
+        """
+        script = "/api/balancer?Info=true"
+        if names != None:
+            script = script+"&Names="+ArrayToCommaSeperatedString(names).replace(' ','+')
+        return self.connectionProperties.__get__(script)
+
+    def SaveBalancerInfo(self, info):
+        """ Saves a balancer info to the database.
+            Input:    info: JSon object of the Balancer info
+            Returns: Success message
+        """
+        
+        info = json.dumps(info)
+        
+        body = '{"Command":"saveinfo", "BalancerInfo":'+info+'}'
+        
+        return self.connectionProperties.__put__("/api/balancer", body)
+        
+    def GetBalancerSettings(self, name):
+        """ Gets a balancer settings.
+            Input: name: The balancer name
+            Returns: The balancer settings.
+        """
+        return self.connectionProperties.__get__("/api/balancer?Name="+name.replace(' ','+')+"&Settings=true")
+
+    def GetBalancerSettingsList(self, names = None):
+        """ Gets a list of balancer settings.
+            Input: name: The balancer names to be retrieved. If None then gets all balancer settings
+            Returns: The balancer settings.
+        """
+        script = "/api/balancer?Settings=true"
+        if names != None:
+            script = script+"&Names="+ArrayToCommaSeperatedString(names).replace(' ','+')
+        
+        return self.connectionProperties.__get__(script)
+
+    def SaveBalancerSettings(self, settings):
+        """ Saves a balancer settings to the database.
+            Input:    settings: JSon object of the Balancer settings
+            Returns: Success message
+        """
+        
+        settings = json.dumps(settings)
+        
+        body = '{"Command":"savesettings", "BalancerSettings":'+settings+'}'
+        
+        return self.connectionProperties.__put__("/api/balancer", body)
+        
+    def GetBalancerInfoSettings(self, name):
+        """ Gets a balancer info settings.
+            Input: name: The balancer name
+            Returns: The balancer info settings.
+        """
+        return self.connectionProperties.__get__("/api/balancer?Name="+name.replace(' ','+')+"&Settings=true&Info=true")
+
+    def GetBalancerInfoSettingsList(self, names = None):
+        """ Gets a list of balancer info settings.
+            Input: name: The balancer names to be retrieved. If None then gets all balancer info settings
+            Returns: The balancer info settings.
+        """
+        script = "/api/balancer"
+        if names != None:
+            script = script+"?Names="+ArrayToCommaSeperatedString(names).replace(' ','+')
+        
+        return self.connectionProperties.__get__(script)
+        
+    def DeleteBalancer(self, name):
+        """ Deletes the Balancer instance associated with the name provided.
+            Input: name: The balancer name to delete.
+            Returns: Success message
+        """
+        return self.connectionProperties.__delete__("/api/balancer?Name="+name)
+
+#Helper function to seperate arrays into strings
+def ArrayToCommaSeperatedString(array):
+    if isinstance(array, basestring):
+        return array
+    else:
+        i=0
+        script=""
+        for i in range(0,len(array)):
+            if(i!=0):
+                script+=","
+            script += str(array[i]);
+        return script

Deadline/Deadline_API/ConnectionProperty.py

@@ -0,0 +1,47 @@
+import sys, subprocess, os
+import json
+import traceback
+import DeadlineSend
+
+class ConnectionProperty:
+
+    def __init__(self, address, useAuth=False):
+        self.address = address
+        self.useAuth = useAuth
+        self.user = ""
+        self.password = ""
+        
+    def GetAddress(self):
+        return self.address
+        
+    def SetAddress(self, address):
+        self.address = address
+        
+    def GetAuthentication(self):
+        return self.user, self.password
+        
+    def SetAuthentication(self, user, password):
+        self.user = user
+        self.password = password
+        
+    def AuthenticationEnabled(self):
+        return self.useAuth
+        
+    def EnableAuthentication(self, enable):
+        self.useAuth = enable
+        
+    def __get__(self, commandString):
+        
+        return DeadlineSend.send(self.address,commandString, "GET", self.useAuth, self.user, self.password)
+        
+    def __put__(self, commandString, body):
+        
+        return DeadlineSend.pSend(self.address, commandString, "PUT", body, self.useAuth, self.user, self.password)
+        
+    def __delete__(self, commandString):
+        
+        return DeadlineSend.send(self.address,commandString, "DELETE", self.useAuth, self.user, self.password)
+        
+    def __post__(self, commandString, body):
+        
+        return DeadlineSend.pSend(self.address, commandString, "POST", body, self.useAuth, self.user, self.password)

Deadline/Deadline_API/DeadlineConnect.py

@@ -0,0 +1,81 @@
+import sys, subprocess, os
+import json
+import traceback
+import Jobs
+import SlavesRenderingJob
+import JobReports
+import TaskReports
+import Limits
+import Tasks
+import Pulse
+import Repository
+import MappedPaths
+import MaximumPriority
+import Pools
+import Groups
+import Plugins
+import Slaves
+import Users
+import Balancer
+from ConnectionProperty import ConnectionProperty
+
+#http://docs.python.org/2/library/httplib.html
+
+class DeadlineCon:
+    """
+        Object used by user to communicate with the web service.
+        Host name of the web service, as well as the port number the
+        web service is listening on are required for construction.
+        Call other API functions through this object.
+    """
+    def __init__(self, host, port):
+        """ Constructs an instance of DeadlineCon.
+            Params: host name of the web service (string)
+                    port number the web service is listening on (integer)
+        """
+        
+        #Builds the ConnectionProperty object used for sending requests
+        address = host+":"+str(port)
+        self.connectionProperties = ConnectionProperty(address)
+        
+        #The different request groups use the ConnectionProperty object to send their requests
+        self.Jobs = Jobs.Jobs(self.connectionProperties)
+        self.SlavesRenderingJob = SlavesRenderingJob.SlavesRenderingJob(self.connectionProperties)
+        self.Tasks = Tasks.Tasks(self.connectionProperties)
+        self.TaskReports = TaskReports.TaskReports(self.connectionProperties)
+        self.JobReports = JobReports.JobReports(self.connectionProperties)
+        self.LimitGroups = Limits.LimitGroups(self.connectionProperties)
+        self.Pulse = Pulse.Pulse(self.connectionProperties)
+        self.Repository = Repository.Repository(self.connectionProperties)
+        self.MappedPaths = MappedPaths.MappedPaths(self.connectionProperties)
+        self.MaximumPriority = MaximumPriority.MaximumPriority(self.connectionProperties)
+        self.Pools = Pools.Pools(self.connectionProperties)
+        self.Groups = Groups.Groups(self.connectionProperties)
+        self.Plugins = Plugins.Plugins(self.connectionProperties)
+        self.Slaves = Slaves.Slaves(self.connectionProperties)
+        self.Users = Users.Users(self.connectionProperties)
+        self.Balancer = Balancer.Balancer(self.connectionProperties)
+        
+    def EnableAuthentication(self, enable=True):
+        """
+            Toggles authentication mode. If enabled, requests sent through this DeadlineCon object will attempt authentication with the current user name and password credentials.
+            If the authentication credentials have not been set, authentication will fail. Required to be enabled if the Web Service requires authentication.
+            Params: whether to disable or enable authentication mode (enabled by default, bool)
+        """
+        self.connectionProperties.EnableAuthentication(enable)
+        
+    def SetAuthenticationCredentials(self, username, password, enable=True):
+        """
+            Sets the authentication credentials to be used when attempting authentication.
+            Params: the username credential (string)
+                    the password credential (string)
+                    whether to enable authentication mode or not (enabled by default, bool)
+        """
+        self.connectionProperties.SetAuthentication(username, password)
+        self.connectionProperties.EnableAuthentication(enable)
+        
+    def AuthenticationModeEnabled(self):
+        """
+            Returns whether authentication mode is enabled for this DeadlineCon or not. If not, then authentication will fail if the Web Service requires authentication."
+        """
+        return self.connectionProperties.AuthenticationEnabled()

Deadline/Deadline_API/DeadlineSend.py

@@ -0,0 +1,102 @@
+import socket
+import httplib
+import json
+import urllib2
+import traceback
+
+def send(address, message, requestType, useAuth=False, username="", password=""):
+    """
+        Used for sending requests that do not require message body, like GET and DELETE.
+        Params: address of the webservice (string)
+                message to the webservice (string)
+                request type for the message (string, GET or DELETE)
+    """
+    try:
+        if not address.startswith("http://"):
+            address = "http://"+address
+        url = address + message
+        
+        if useAuth:
+            password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
+
+            password_mgr.add_password(None, url, username, password)
+
+            handler = urllib2.HTTPBasicAuthHandler(password_mgr)
+            opener = urllib2.build_opener(handler)
+            request = urllib2.Request(url)
+            request.get_method = lambda: requestType
+            
+            response = opener.open(request)
+        else:
+            opener = urllib2.build_opener(urllib2.HTTPHandler)
+            request = urllib2.Request(url)
+            request.get_method = lambda: requestType
+            
+            response = opener.open(request)
+
+        data = response.read()
+            
+        data = data.replace('\n',' ')
+        
+    except urllib2.HTTPError, err:
+        data = traceback.format_exc()
+        if err.code == 401:
+            data = "Error: HTTP Status Code 401. Authentication with the Web Service failed. Please ensure that the authentication credentials are set, are correct, and that authentication mode is enabled."
+        else:
+            data = err.read()
+    try:
+        data = json.loads(data)
+    except:
+        pass
+
+    return data
+    
+def pSend(address, message, requestType, body, useAuth=False, username="", password=""):
+    """
+        Used for sending requests that require a message body, like PUT and POST.
+        Params: address of the webservice (string)
+                message to the webservice (string)
+                request type for the message (string, PUT or POST)
+                message body for the request (string, JSON object)
+    """
+    response = ""
+    try:
+        if not address.startswith("http://"):
+            address = "http://"+address
+        url = address + message
+        
+        if useAuth:
+            password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
+            
+            password_mgr.add_password(None, url, username, password)
+
+            handler = urllib2.HTTPBasicAuthHandler(password_mgr)
+            opener = urllib2.build_opener(handler)
+            
+            request = urllib2.Request(url, data=body)
+            request.get_method = lambda: requestType
+            
+            response = opener.open(request)
+        
+        else:
+            opener = urllib2.build_opener(urllib2.HTTPHandler)
+            request = urllib2.Request(url, data=body)
+            request.get_method = lambda: requestType
+            response = opener.open(request)
+            
+        data = response.read()
+
+    except urllib2.HTTPError, err:
+        data = traceback.format_exc()
+        if err.code == 401:
+            data = "Error: HTTP Status Code 401. Authentication with the Web Service failed. Please ensure that the authentication credentials are set, are correct, and that authentication mode is enabled."
+        else:
+            data = err.read()
+
+        
+    try:
+        data = json.loads(data)
+    except:
+        pass
+        
+    return data