| 1 |
nino.borges |
570 |
"""
|
| 2 |
|
|
|
| 3 |
|
|
Relativity Library
|
| 4 |
|
|
|
| 5 |
|
|
Created by
|
| 6 |
|
|
Emanuel Borges
|
| 7 |
|
|
05.22.2014
|
| 8 |
|
|
|
| 9 |
|
|
This class will be my main Relativity library
|
| 10 |
|
|
Should DocumentTable and COTable be their own classes that subclass RelativityConnection?
|
| 11 |
|
|
|
| 12 |
|
|
|
| 13 |
|
|
TODO:
|
| 14 |
|
|
Make it so taht you van overide the site, username and pw. Right now it's created before you can overide.
|
| 15 |
|
|
|
| 16 |
|
|
"""
|
| 17 |
|
|
|
| 18 |
|
|
import requests,json
|
| 19 |
|
|
|
| 20 |
|
|
class Relativity_Connection(object):
|
| 21 |
|
|
"""An Advanced Relativity Connection Library and Toolkit"""
|
| 22 |
|
|
version = '0.0.1'
|
| 23 |
nino.borges |
605 |
|
| 24 |
|
|
## Prod
|
| 25 |
|
|
rootSite = 'https://relativity5.advanceddiscovery.com'
|
| 26 |
|
|
rootUserName = 'emanuel.borges@advanceddiscovery.com'
|
| 27 |
|
|
rootPassword = 'M@y_1255'
|
| 28 |
|
|
|
| 29 |
|
|
|
| 30 |
|
|
## ## Dev
|
| 31 |
|
|
## #rootSite = 'http://dev.mcdermottdiscovery.com'
|
| 32 |
|
|
## #rootUserName = 'eborges@mwe.com'
|
| 33 |
|
|
## #rootPassword = 'MWEreldev2@'
|
| 34 |
|
|
##
|
| 35 |
|
|
## ## Prod
|
| 36 |
|
|
## rootSite = 'https://www.mcdermottdiscovery.com'
|
| 37 |
|
|
## rootUserName = 'eborges@mwe.com'
|
| 38 |
|
|
## rootPassword = 'QEst3kU2!'
|
| 39 |
nino.borges |
570 |
|
| 40 |
|
|
def __init__(self, workSpace = None):
|
| 41 |
|
|
self.UpdateWorkspaceDict()
|
| 42 |
|
|
self.workSpace = workSpace
|
| 43 |
|
|
self.availableTables = None
|
| 44 |
|
|
|
| 45 |
|
|
def UpdateWorkspaceDict(self):
|
| 46 |
|
|
"""Updates the instance varible workSpaceDict with a {workspaceName:ID} dictionary"""
|
| 47 |
|
|
workSpaceDict = {}
|
| 48 |
|
|
r = requests.get(self.rootSite+'/Relativity.REST/Relativity/Workspace',auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword), headers={'X-CSRF-Header':""})
|
| 49 |
|
|
rawJson = r.json()
|
| 50 |
|
|
rawResults = rawJson['Results']
|
| 51 |
|
|
for i in rawResults:
|
| 52 |
|
|
workSpaceDict[i['Relativity Text Identifier']] = i['Artifact ID']
|
| 53 |
|
|
|
| 54 |
|
|
self.workSpaceDict = workSpaceDict
|
| 55 |
|
|
|
| 56 |
|
|
def GetAvilableTables(self):
|
| 57 |
|
|
"""updates the list of available tables"""
|
| 58 |
|
|
pass
|
| 59 |
|
|
|
| 60 |
|
|
def LoadWorkspace(self, workspaceName):
|
| 61 |
|
|
r = requests.get(self.rootSite+ '/Relativity.REST/Relativity/Workspace/%s'% self.workSpaceDict[workSpaceName],auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword), headers={'X-CSRF-Header':""})
|
| 62 |
|
|
self.workSpace = workSpaceName
|
| 63 |
|
|
|
| 64 |
|
|
|
| 65 |
|
|
|
| 66 |
|
|
class DocumentTable(Relativity_Connection):
|
| 67 |
|
|
|
| 68 |
|
|
def __init__ (self,workSpace):
|
| 69 |
|
|
self.workSpace = workSpace
|
| 70 |
|
|
super(DocumentTable,self).__init__(workSpace)
|
| 71 |
|
|
r = requests.get(self.rootSite+ '/Relativity.REST/Relativity/Workspace/%s/Document'% self.workSpaceDict[self.workSpace],auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword), headers={'X-CSRF-Header':""})
|
| 72 |
|
|
self.availableTables = r
|
| 73 |
|
|
#return r.__dict__
|
| 74 |
|
|
|
| 75 |
|
|
def UpdateField(self,field):
|
| 76 |
|
|
d = {
|
| 77 |
|
|
"Artifact ID":1078252,"Artifact Type Name":"Tasks","Billing Time Used":5
|
| 78 |
|
|
}
|
| 79 |
|
|
|
| 80 |
|
|
r = requests.put('https://dev.mcdermottdiscovery.com/Relativity.REST/Workspace/1016359/Tasks/1078252',
|
| 81 |
|
|
auth=requests.auth.HTTPBasicAuth('eborges@mwe.com','MWEreldev2@'),
|
| 82 |
|
|
headers={'X-CSRF-Header':"",'Content-Type':"application/json; charset=utf-8"},
|
| 83 |
|
|
data=json.dumps(d))
|
| 84 |
|
|
|
| 85 |
|
|
|
| 86 |
|
|
|
| 87 |
|
|
|
| 88 |
|
|
class CustomObjectTable(Relativity_Connection):
|
| 89 |
|
|
'''You can just insts this object if you already know the workspace # and objectName. Otherwise insts a sep rl object
|
| 90 |
|
|
to gather the workspace number and objectName'''
|
| 91 |
|
|
def __init__ (self, workSpace, objectName):
|
| 92 |
|
|
self.workSpace = workSpace
|
| 93 |
|
|
super(CustomObjectTable,self).__init__(workSpace)
|
| 94 |
|
|
r = requests.get(self.rootSite+ '/Relativity.REST/Workspace/%s/%s'% (self.workSpaceDict[self.workSpace],objectName),
|
| 95 |
|
|
auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword),
|
| 96 |
|
|
headers={'X-CSRF-Header':""})
|
| 97 |
|
|
|
| 98 |
|
|
self.rlResponse = r.json()
|
| 99 |
|
|
|
| 100 |
|
|
rlResponseCode = r.status_code
|
| 101 |
|
|
print rlResponseCode
|
| 102 |
|
|
|
| 103 |
|
|
## This is the ID of the workspace, which is needed when you update or create new items
|
| 104 |
|
|
## This will only work if there are already items in this table, got to fix this...
|
| 105 |
|
|
self.parentID = self.rlResponse['Results'][0]['Parent Artifact']['Artifact ID']
|
| 106 |
|
|
|
| 107 |
|
|
## You wont get the field list until after you return 1 item from the RDO
|
| 108 |
|
|
self.fieldMatrix = None
|
| 109 |
|
|
|
| 110 |
|
|
## Count of the results both in total and on this page.
|
| 111 |
|
|
self.totalResultCount = self.UpdateTotalResultCount()
|
| 112 |
|
|
self.pageResultCount = self.UpdateCurrentResultCount()
|
| 113 |
|
|
|
| 114 |
|
|
## The matrix if returned items with artifactID:name. Cant do other way around, since name can dupe.
|
| 115 |
|
|
self.itemMatrix = self.UpdateItemMatrix()
|
| 116 |
|
|
|
| 117 |
|
|
|
| 118 |
|
|
## The objectName stays the same, since you need a new instance for each table.
|
| 119 |
|
|
self.objectName = objectName
|
| 120 |
|
|
|
| 121 |
|
|
## Loading the item will update the field list and make that the active item
|
| 122 |
|
|
self.currentItems = None
|
| 123 |
|
|
|
| 124 |
|
|
## You may load 1 item using single item or query and get multiple items.
|
| 125 |
|
|
self.currentItemCount = None
|
| 126 |
|
|
|
| 127 |
|
|
|
| 128 |
|
|
|
| 129 |
|
|
def LoadSingleItem(self, itemID):
|
| 130 |
|
|
'''loads a requested item into the instance and updates the field matrix'''
|
| 131 |
|
|
r = requests.get(self.rootSite+ '/Relativity.REST/Workspace/%s/%s/%s'% (self.workSpaceDict[self.workSpace],self.objectName,itemID),
|
| 132 |
|
|
auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword),
|
| 133 |
|
|
headers={'X-CSRF-Header':""})
|
| 134 |
|
|
rlResponse = r.json()
|
| 135 |
|
|
rlResponseCode = r.status_code
|
| 136 |
|
|
print rlResponseCode
|
| 137 |
|
|
self.currentItems = [rlResponse,]
|
| 138 |
|
|
self.UpdateFieldMatrix(self.currentItems[0]['__Fields'])
|
| 139 |
|
|
self.currentItemCount = len(self.currentItems)
|
| 140 |
|
|
|
| 141 |
|
|
def QueryForItems(self,searchString):
|
| 142 |
|
|
'''takes a search string and brings back a matrix containing only those items that matched the search'''
|
| 143 |
|
|
payload = {
|
| 144 |
|
|
"condition":searchString,
|
| 145 |
|
|
"fields":["*"]
|
| 146 |
|
|
}
|
| 147 |
|
|
## Note that it will give you a max of 100 here because I'm setting the page size.
|
| 148 |
|
|
r = requests.post(self.rootSite+ '/Relativity.REST/Workspace/%s/%s/QueryResult?pagesize=100'% (self.workSpaceDict[self.workSpace],self.objectName),
|
| 149 |
|
|
auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword),
|
| 150 |
|
|
headers={'X-CSRF-Header':"",'Content-Type':"application/json; charset=utf-8"},
|
| 151 |
|
|
data=json.dumps(payload))
|
| 152 |
|
|
rlResponse = r.json()
|
| 153 |
|
|
rlResponseCode = r.status_code
|
| 154 |
|
|
print rlResponseCode
|
| 155 |
|
|
self.currentItems = rlResponse['Results']
|
| 156 |
|
|
#print self.currentItems
|
| 157 |
nino.borges |
579 |
if self.currentItems:
|
| 158 |
|
|
self.UpdateFieldMatrix(self.currentItems[0]['__Fields'])
|
| 159 |
|
|
self.currentItemCount = len(self.currentItems)
|
| 160 |
|
|
|
| 161 |
nino.borges |
570 |
return self.currentItems
|
| 162 |
|
|
|
| 163 |
|
|
def UpdateItemMatrix(self):
|
| 164 |
|
|
''' Updates the item matrix with ArtifactID:name and also puts all names in the itemlist'''
|
| 165 |
|
|
itemMatrix = {}
|
| 166 |
|
|
for i in self.rlResponse['Results']:
|
| 167 |
|
|
itemMatrix[i['Artifact ID']] = i['Relativity Text Identifier']
|
| 168 |
|
|
return itemMatrix
|
| 169 |
|
|
|
| 170 |
|
|
|
| 171 |
|
|
def UpdateTotalResultCount(self):
|
| 172 |
|
|
'''Returns the current totalResultCount'''
|
| 173 |
|
|
return self.rlResponse['TotalResultCount']
|
| 174 |
|
|
|
| 175 |
|
|
def UpdateCurrentResultCount(self):
|
| 176 |
|
|
'''Returns the current result count for the page that you are on'''
|
| 177 |
|
|
return self.rlResponse['ResultCount']
|
| 178 |
|
|
|
| 179 |
|
|
def UpdateFieldMatrix(self,rawFieldDict):
|
| 180 |
|
|
'''Returns the list of fields in the table object'''
|
| 181 |
|
|
fieldMatrix = {}
|
| 182 |
|
|
for i in rawFieldDict:
|
| 183 |
|
|
fieldMatrix[i['Name']] = i['Field Type']
|
| 184 |
|
|
self.fieldMatrix = fieldMatrix
|
| 185 |
|
|
|
| 186 |
|
|
|
| 187 |
|
|
def CreateNewItem(self, itemFieldMatrix):
|
| 188 |
|
|
'''Allows you to create a new item in the DO, by passing a matrix of fields and their values'''
|
| 189 |
|
|
itemFieldMatrix["Artifact Type Name"] = self.objectName
|
| 190 |
|
|
itemFieldMatrix["Parent Artifact"] = {"Artifact ID":self.parentID}
|
| 191 |
|
|
|
| 192 |
|
|
r = requests.post(self.rootSite+ '/Relativity.REST/Workspace/%s/%s'% (self.workSpaceDict[self.workSpace],self.objectName),
|
| 193 |
|
|
auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword),
|
| 194 |
|
|
headers={'X-CSRF-Header':"",'Content-Type':"application/json; charset=utf-8"},
|
| 195 |
|
|
data=json.dumps(itemFieldMatrix))
|
| 196 |
|
|
rlResponse = r.json()
|
| 197 |
|
|
rlResponseCode = r.status_code
|
| 198 |
|
|
print rlResponseCode
|
| 199 |
|
|
artID = rlResponse['Results'][0]['ArtifactID']
|
| 200 |
|
|
## Returning the artifact id?, not sure about this. maybe should return status and allow you to get at artifact...
|
| 201 |
|
|
return artID
|
| 202 |
|
|
#d = {
|
| 203 |
|
|
# "Name":"TEST","Artifact Type Name":"Tasks","Billing Time Used":5,"Parent Artifact":{"Artifact ID":1003663}
|
| 204 |
|
|
#}
|
| 205 |
|
|
#
|
| 206 |
|
|
#r = requests.post('https://dev.mcdermottdiscovery.com/Relativity.REST/Workspace/1016359/Tasks',
|
| 207 |
|
|
# auth=requests.auth.HTTPBasicAuth('eborges@mwe.com','MWEreldev2@'),
|
| 208 |
|
|
# headers={'X-CSRF-Header':"",'Content-Type':"application/json; charset=utf-8"},
|
| 209 |
|
|
# data=json.dumps(d))
|
| 210 |
|
|
|
| 211 |
|
|
|
| 212 |
|
|
def UpdateField(self,field):
|
| 213 |
|
|
|
| 214 |
|
|
d = {
|
| 215 |
|
|
"Artifact ID":1078252,"Artifact Type Name":"Tasks","Billing Time Used":5,"Parent Artifact":{"Artifact ID":1003663}
|
| 216 |
|
|
}
|
| 217 |
|
|
|
| 218 |
|
|
r = requests.put(self.rootSite+ '/Relativity.REST/Workspace/1016359/Tasks/1078252',
|
| 219 |
|
|
auth=requests.auth.HTTPBasicAuth(self.rootUserName,self.rootPassword),
|
| 220 |
|
|
headers={'X-CSRF-Header':"",'Content-Type':"application/json; charset=utf-8"},
|
| 221 |
|
|
data=json.dumps(d))
|
| 222 |
|
|
|
| 223 |
|
|
|
