Active_prgs/MCP/MCP_Lib.py

"""

MCP_Lib

Created by
Emanuel Borges
10.25.2010

This wil be the main library for the MCP aplications. Shared methods will be
kept here.  Version informaiton for the entire MCP program will be kept here to.

"""

import os,sys, win32com.client,subprocess,ftplib,NinoGenTools, win32com.decimal_23

def GetCaseList(sys_Overwrite ="", accessConnectionObj = None):
    """ Main method to parse and return usable lists of cases. returns list of my cases, office cases and all cases.
    also returns the cases directory."""
    #print os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])),"settings.sys")
    if sys_Overwrite:
        casesDir = open(os.path.join(sys_Overwrite,"settings.sys")).readline().replace("\n","")
    else:
        casesDir = open(os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])),"settings.sys")).readline().replace("\n","")
    #casesDir = r"C:\Documents and Settings\eborges\My Documents\My Dropbox\Documents\Cases"
    responsibleCases = []
    activeCases = []
    officeCases = []
    allCases = []
    tPMName,tPMID,nul = GetTPMInfo()
    if accessConnectionObj:
        accessDB = accessConnectionObj
    else:
        print "connecting to matter tracking access db..."
        ##  Production version
        accessDB = AccessDBConnection(r"\\chiads01\app\DS_CaseTrack\TPM_CaseTracking.mdb")
        
        ##  Testing version
        #accessDB = AccessDBConnection(r"W:\Manny\testing\TPM_CaseTracking.mdb")
        print "connected to DB."
    officeList = GetFullOfficeList()
    
    cliMatList = []
    for office in officeList:
            cliMatList.extend(accessDB.RetrieveOfficeCaseList(office))
    cliMatList.sort()
    for cliMat in cliMatList:
        caseName = accessDB.GetCaseName(cliMat)
        allCases.append('%s_(%s)'%(caseName,cliMat))
    
    if tPMName == 'ANALYST':
        casesDir = None
        officeCases = allCases
        responsibleCases = allCases
        activeCases = allCases
    else:
        myMatterList = accessDB.RetrieveMyCaseList(tPMID)
        myActiveMatterList = accessDB.RetrieveMyActiveCaseList(tPMID)
        for file in os.listdir(casesDir):
            if os.path.isdir(os.path.join(casesDir,file)):
                if file == "_Template_":
                    pass
                elif file == "zzzz_dormant_":
                    pass
                else:
                    officeCases.append(file)
                    ## if the case's matter is in my matter list, also append to responsibleCases
                    clientMatter = file.split("_(")[1]
                    clientMatter = clientMatter[:-1]
                    clientMatter = clientMatter.replace('-','.')
                    if clientMatter in myMatterList:
                        responsibleCases.append(file)
                    if clientMatter in myActiveMatterList:
                        activeCases.append(file)
    return responsibleCases, activeCases, officeCases, allCases, casesDir

def GetTPMInfo(sys_Overwrite = ""):
    """ Main method to parse and return the TMP and the Employee ID"""
    if sys_Overwrite:
        syscaseDir,tPMName, tPMID,office = open(os.path.join(sys_Overwrite,"settings.sys")).readlines()
    else:
        syscaseDir,tPMName, tPMID,office = open(os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])),"settings.sys")).readlines()
    
    #syscaseDir,tPMName, tPMID = open(os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])),"settings.sys")).readlines()
    tPMName = tPMName.replace("\n","")
    tPMID  = tPMID.replace("\n","")
    office = office.replace("\n","")
    return tPMName, tPMID, office
       
def GetFullOfficeList():
    """Simply returns a list of all of the offices."""
    officeList = ['Boston','Brussels','Chicago','Huston','London','Los Angeles','Miami','Moscow','Munich','New York',
                  'Orange County','Rome','San Diego','Silicon Valley','Washington, D.C.']
    return officeList

def GetOtherCaseFolder(office):
    """This method will return the case folder for cases that you dont own"""
    ##  So this is in two places that I now have to maintian... not sure I like that.
    caseFolderMatrix={'Boston':r"\\bstads01\app\Manny\Cases",
                      'Chicago':r"\\chiads01\data\CLI\Litigation_Support\MCP\Cases",
                      'Los Angeles':r"\\lasads01\data\Cli\_Lit_Support\MCP",
                      'New York':r"\\nykads01\data\cli\LitSupport\MCP\Cases",
                      'Miami':r"\\nykads01\data\cli\LitSupport\MCP\Cases",
                      'Orange County':r"\\lasads01\data\Cli\_Lit_Support\MCP",
                      'San Diego':r"\\lasads01\data\Cli\_Lit_Support\MCP",
                      'Silicon Valley':r"\\lasads01\data\Cli\_Lit_Support\MCP",
                      'Washington, D.C.':r"\\wdcads01\data\Cli\_MCP\Cases"}
    try:
        location = caseFolderMatrix[office]
    except:
        location = None
    return location

def GetArchiveFileTypeList():
    """Returns a list of extensions of archive type files"""
    return [".ZIP", ".RAR",".TAR",".7Z"]

def GetMCPVersion():
    """Returns the current version of the entire MCP program set"""
    return 'v 1.8.0'


class AccessDBConnection:
    def __init__(self, accessDB):
        daoEngine = win32com.client.Dispatch('DAO.DBEngine.36')
        self.daoDB = daoEngine.OpenDatabase(accessDB)
        
    def MakeNewRSConnection(self, table, specialSelect = "*", specialWhere="" ):
        """ This method will make a new RS connection for either reading or writing."""
        if specialWhere:
            statement = "SELECT %s FROM %s WHERE %s"%(specialSelect,table, specialWhere)
        else:
            statement = "SELECT %s FROM %s"%(specialSelect,table)
        daoRSObj = self.daoDB.OpenRecordset(statement)
        return daoRSObj
        
    def CloseAccessConnection(self):
        """This closes the entire connection and not just the RS"""
        self.daoDB.Close()
        
    def RetrieveAllTPMMatrix(self):
        """Retrieves a dictionary of all TPMs and their offices. {TPM:(timekeeperID,OfficeLocation)}"""
        daoRSObj = self.MakeNewRSConnection('tbl_Ref_Employee', specialWhere = "staffCategory = 'TPM'")
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        tpmDict = {}
        for i in range(fullCount):
            tpmDict[daoRSObj.Fields('TPMName').Value] = (daoRSObj.Fields('TimekeeperID').Value,daoRSObj.Fields('OfficeLocation').Value)
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return tpmDict
        
    
    def RetrieveAllCaseList(self):
        """Retrieves all of the cases in the database regardless of assignment.(client matter list)"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData')
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        accessDBCLMList = []
        for i in range(fullCount):
            accessDBCLMList.append(daoRSObj.Fields('ClientMatterNum').Value)
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return accessDBCLMList
        
    def RetrieveMyCaseList(self, empID):
        """Retrieves just 1 employees case list (client matter list)"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "empID = '%s'"%empID)
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        accessDBCLMList = []
        for i in range(fullCount):
            accessDBCLMList.append(daoRSObj.Fields('ClientMatterNum').Value)
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return accessDBCLMList
    
    def RetrieveMyActiveCaseList(self,empID):
        """Retrieves only the Active cases for just 1 employees case list (client matter list)"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "empID = '%s' and ProjectStatus = 'Active'"%empID)
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        accessDBCLMList = []
        for i in range(fullCount):
            accessDBCLMList.append(daoRSObj.Fields('ClientMatterNum').Value)
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return accessDBCLMList
    
    
    def RetrieveOfficeCaseList(self, primaryOffice):
        """Retrieves the case list for an entire office. Note, a case can be owned by a TPM in another office but
        the primary office might still be an office that that TPM does not work for.  Also people are not always
        good about updating the primary office...(client matter list)"""
        accessDBCLMList = []
        try:
            daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "PrimaryOffice = '%s'"%primaryOffice)
            daoRSObj.MoveLast()
            fullCount = daoRSObj.RecordCount
            daoRSObj.MoveFirst()
            
            
            for i in range(fullCount):
                accessDBCLMList.append(daoRSObj.Fields('ClientMatterNum').Value)
                daoRSObj.MoveNext()
            daoRSObj.Close()
        except:
            pass
        return accessDBCLMList
    
    def RetrieveProductionsByCLM(self,CLM):
        """This will retrieve a FULL production matrix for a specific clm"""
        daoRSObj = self.MakeNewRSConnection('Tbl_MatterProductionDetail',
                                            "ProductionID,ProductionComplDate, BegBates,EndBates,ProdDocCount,ProdPageCount,Notes,ProducedTo,ProductionMedia,ProductionSource,RequestedBy,ProductionDate,ProductionMediaPassword",
                                            specialWhere = "ClientMatterNum='%s' ORDER BY ProductionComplDate"%CLM)
        try:
            ## Getting an exception that I cant capture when table is empty.
            ## this will test and return empty if empty, until I can figure out a better test.
            daoRSObj.MoveLast()
            tableHasData = True
        except:
            tableHasData = False
        productionList = []
        productionMatrix = {}
        if tableHasData:
            fullCount = daoRSObj.RecordCount
            daoRSObj.MoveFirst()
            for i in range(fullCount):
                try:
                    pgCount = str(int(daoRSObj.Fields('ProdPageCount').Value))
                except:
                    pgCount = '0'
                try:
                    docCount = str(int(daoRSObj.Fields('ProdDocCount').Value))
                except:
                    docCount = '0'
                grouping = str(daoRSObj.Fields('ProducedTo').Value)
                if grouping:
                    pass
                else:
                    grouping = 'default'
                if grouping in productionMatrix.keys():
                    productionMatrix[grouping].append((str(daoRSObj.Fields('ProductionID').Value),
                                       str(daoRSObj.Fields('ProductionComplDate').Value).split(' ')[0],
                                       str(daoRSObj.Fields('ProductionDate').Value).split(' ')[0],
                                       str(daoRSObj.Fields('BegBates').Value),str(daoRSObj.Fields('EndBates').Value),
                                       docCount,pgCount,
                                       str(daoRSObj.Fields('ProducedTo').Value),
                                       str(daoRSObj.Fields('RequestedBy').Value),
                                       str(daoRSObj.Fields('ProductionMedia').Value),
                                       str(daoRSObj.Fields('ProductionMediaPassword').Value),
                                       str(daoRSObj.Fields('ProductionSource').Value),
                                       str(daoRSObj.Fields('Notes').Value)))
                else:
                    productionMatrix[grouping] = [(str(daoRSObj.Fields('ProductionID').Value),
                                       str(daoRSObj.Fields('ProductionComplDate').Value).split(' ')[0],
                                       str(daoRSObj.Fields('ProductionDate').Value).split(' ')[0],
                                       str(daoRSObj.Fields('BegBates').Value),str(daoRSObj.Fields('EndBates').Value),
                                       docCount,pgCount,
                                       str(daoRSObj.Fields('ProducedTo').Value),
                                       str(daoRSObj.Fields('RequestedBy').Value),
                                       str(daoRSObj.Fields('ProductionMedia').Value),
                                       str(daoRSObj.Fields('ProductionMediaPassword').Value),
                                       str(daoRSObj.Fields('ProductionSource').Value),
                                       str(daoRSObj.Fields('Notes').Value))]
                #productionList.append((str(daoRSObj.Fields('ProductionID').Value),
                #                      str(daoRSObj.Fields('ProductionComplDate').Value).split(' ')[0],
                #                       str(daoRSObj.Fields('BegBates').Value),str(daoRSObj.Fields('EndBates').Value),
                #                       docCount,pgCount,
                #                       str(daoRSObj.Fields('Notes').Value)))
                daoRSObj.MoveNext()
        #print len(productionList)
        daoRSObj.Close()
        return productionMatrix
   
 
    def RetrieveUploadsByCLM(self, CLM):
        """This will retrieve a FULL upload matrix for a specific clm. This wil also pick the best Size (gb,mb,kb), since it's just for display."""
        daoRSObj = self.MakeNewRSConnection('tbl_ClientDataQuantity',
                                            "UEPOCH,ReviewPlatform,DataLoadDate,DataLoadedGB,DataLoadedMB,DataLoadedKB",
                                            specialWhere = "ClientMatterNum='%s'"%CLM)
        try:
            ## Getting an exception that I cant capture when table is empty.
            ## this will test and return empty if empty, until I can figure out a better test.
            daoRSObj.MoveLast()
            tableHasData = True
        except:
            tableHasData = False
        caseUploadList = []
        if tableHasData:
            fullCount = daoRSObj.RecordCount
            daoRSObj.MoveFirst()
            
            for i in range(fullCount):
                if daoRSObj.Fields('DataLoadedKB').Value:
                    size = str(daoRSObj.Fields('DataLoadedKB').Value) + ' KB'
                elif daoRSObj.Fields('DataLoadedMB').Value:
                    size = str(daoRSObj.Fields('DataLoadedMB').Value) + ' MB'
                else:
                    size = str(daoRSObj.Fields('DataLoadedGB').Value) + ' GB'
                    
                caseUploadList.append((str(daoRSObj.Fields('UEPOCH').Value),daoRSObj.Fields('ReviewPlatform').Value,
                                       str(daoRSObj.Fields('DataLoadDate').Value).split(' ')[0],size))
                daoRSObj.MoveNext()
        daoRSObj.Close()
        return caseUploadList
    
    def RetrieveOfficeUploads(self, primaryOffice):
        """This only returns a matrix of clm{[EPOCH]}, since there isnt a reason to sync uploads from access"""
        myClientMatterList = self.RetrieveOfficeCaseList(primaryOffice)
        daoRSObj = self.MakeNewRSConnection('tbl_ClientDataQuantity')
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        caseUploadMatrix = {}
        for i in range(fullCount):
            currentClientMatter = daoRSObj.Fields('ClientMatterNum').Value
            if currentClientMatter in myClientMatterList:
                epoch = str(daoRSObj.Fields('UEPOCH').Value)
                try:
                    caseUploadMatrix[currentClientMatter].append(epoch)
                except:
                    caseUploadMatrix[currentClientMatter] = [epoch]
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return caseUploadMatrix
    
    def RetrieveMyCaseUploads(self,empID):
        """This only returns a matrix of clm{[EPOCH]}, since there isnt a reason to sync uploads from access"""
        myClientMatterList = self.RetrieveMyCaseList(empID)
        daoRSObj = self.MakeNewRSConnection('tbl_ClientDataQuantity')
        daoRSObj.MoveLast()
        fullCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        
        caseUploadMatrix = {}
        for i in range(fullCount):
            currentClientMatter = daoRSObj.Fields('ClientMatterNum').Value
            if currentClientMatter in myClientMatterList:
                epoch = daoRSObj.Fields('UEPOCH').Value
                if epoch:
                    epoch = str(epoch)
                    if '.' in epoch:
                        epoch = epoch.split('.')[0]
                
                try:
                    caseUploadMatrix[currentClientMatter].append(epoch)
                except:
                    caseUploadMatrix[currentClientMatter] = [epoch]
                #loadedDate = daoRSObj.Fields('DataLoadDate').Value
                #loadedDate = str(loadedDate.Format('%Y%m%d'))
                #loadedSize =daoRSObj.Fields('DataLoadedGB').Value
                #loadedSizeType = " GB"
                #
                #if loadedSize:
                #    ##  If the GB loaded size exists, use that.
                #    pass
                #else:
                #    ##  if it does not, use the MB loaded field.
                #    loadedSize =daoRSObj.Fields('DataLoadedMB').Value
                #    loadedSizeType = " MB"
                #loadedSize = str(loadedSize)
                #if "." in loadedSize:
                #    loadedSize = loadedSize[:loadedSize.index('.')+3]
                #try:
                #    caseUploadMatrix[currentClientMatter].append((loadedDate,loadedSize+loadedSizeType))
                #except:
                #    caseUploadMatrix[currentClientMatter] = [(loadedDate,loadedSize+loadedSizeType)]
            daoRSObj.MoveNext()
        daoRSObj.Close()
        return caseUploadMatrix
    
    def AddNewCase(self, caseName, CLM, TPM_Name, empID, chargable = False, reviewPlatform = "", responsibleVendor = "", responsibleOffice = ""):
        """Adds a new case in the access DB."""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData')
        daoRSObj.AddNew()
        daoRSObj.Fields('ClientMatterNum').Value = CLM
        daoRSObj.Fields('CaseName').Value = caseName
        daoRSObj.Fields('EmpID').Value = empID
        daoRSObj.Fields('TPM').Value = TPM_Name
        daoRSObj.Fields('PrimaryOffice').Value = responsibleOffice
        daoRSObj.Update()
        daoRSObj.Close()
        
        ## Per aaron, add to two other tables
        clientNumber = CLM.split('.')[0]
        daoRSObj = self.MakeNewRSConnection('tbl_Ref_Client')
        try:
            daoRSObj.AddNew()
            daoRSObj.Fields('ClientNum').Value = clientNumber
            daoRSObj.Update()
        except:
            pass
        daoRSObj.Close()
        
        daoRSObj = self.MakeNewRSConnection('tbl_Ref_ClientMatter')
        try:
            daoRSObj.AddNew()
            daoRSObj.Fields('ClientNum').Value = clientNumber
            daoRSObj.Fields('ClientMatterNum').Value = CLM
            daoRSObj.Update()
        except:
            pass
        daoRSObj.Close()
        
        ## These below make their own record sets.
        if chargable:
            ## change this one aaron adds it.
            pass
        if reviewPlatform:
            self.UpdateReviewPlatform(CLM, reviewPlatform)
        if responsibleVendor:
            self.UpdateResponsibleVendor(CLM,responsibleVendor)
        
        
    def UpdateChargable(self, CLM, chargable):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('BillableMatter').Value = chargable
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetChargable(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        chargable = daoRSObj.Fields('BillableMatter').Value
        daoRSObj.Close()
        return chargable
    
    def UpdateDisclosureLetter(self,CLM, letterSaved, pathToLetter):
        """Allows you to check or uncheck the state of the disclosure letter and the link to it."""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('DisclosureLetterSaved').Value = letterSaved
        daoRSObj.Fields('LinktoDisclosureLetter').Value = pathToLetter
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetDisclosureLetter(self, CLM):
        """Returns the state of the saved letter and the path to the disclosure letter"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        letterSaved = daoRSObj.Fields('DisclosureLetterSaved').Value
        pathToLetter = daoRSObj.Fields('LinktoDisclosureLetter').Value
        #pathToLetter = "foo"
        #print letterSaved
        daoRSObj.Close()
        return letterSaved,pathToLetter
        
    def UpdateReviewPlatform(self, CLM, reviewPlatform):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('ReviewPlatform').Value = reviewPlatform
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetReviewPlatform(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        platform = daoRSObj.Fields('ReviewPlatform').Value
        daoRSObj.Close()
        return platform
        
    def UpdateResponsibleVendors(self, CLM,responsibleVendorTpl):
        """VendorTpl should be a tuple of (processing,scanning,hosting) with None if you dont have it"""
        processingVendor, scanningVendor, hostingVendor = responsibleVendorTpl
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        if processingVendor:
            daoRSObj.Edit()
            daoRSObj.Fields('ProcessingVendor').Value = processingVendor
            daoRSObj.Update()
        if scanningVendor:
            daoRSObj.Edit()
            daoRSObj.Fields('ScanningVendor').Value = scanningVendor
            daoRSObj.Update()
        if hostingVendor:
            daoRSObj.Edit()
            daoRSObj.Fields('HostingVendor').Value = hostingVendor
            daoRSObj.Update()
        daoRSObj.Close()
        
    def GetResponsibleVendorTpl(self, CLM):
        """Returns a tuple of (processing,scanning,hosting) with None if you dont have it"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        processingVendor = daoRSObj.Fields('ProcessingVendor').Value
        scanningVendor = daoRSObj.Fields('ScanningVendor').Value
        hostingVendor = daoRSObj.Fields('HostingVendor').Value
        daoRSObj.Close()
        tmpTpl = (processingVendor,scanningVendor,hostingVendor)
        return tmpTpl
        
    def UpdateCaseUpload(self, CLM, reviewPlatform, uploadMatrix):
        """Format for matrix is epoch is key and uploads are in a tpl. (12/23/2010,49MB) Date format is mm/dd/yyyy"""
        daoRSObj = self.MakeNewRSConnection('tbl_ClientDataQuantity')
        
        ##  Dont for get to convert the date first somewhere.
        for i in uploadMatrix.keys():
            (date, size) = uploadMatrix[i]
            daoRSObj.AddNew()
            daoRSObj.Fields('ClientMatterNum').Value = CLM
            daoRSObj.Fields('UEPOCH').Value = i
            daoRSObj.Fields('DataLoadDate').Value = date
            daoRSObj.Fields('ReviewPlatform').Value = reviewPlatform
            
            if "GB" in size:
                daoRSObj.Fields('dataLoadedGB').Value = float(size.replace("GB",""))
            elif "MB" in size:
                daoRSObj.Fields('dataLoadedMB').Value = float(size.replace("MB",""))
            else:
                if "bytes" in size:
                    size = "1 KB"
                daoRSObj.Fields('DataLoadedKB').Value = float(size.replace("KB",""))
            daoRSObj.Update()
        daoRSObj.Close()
        
    def DeleteCaseUpload(self,CLM,UEPOCH):
        """This method will let you delete a single case upload but only if it was added by the MCP.
        i.e has a UEPOCH"""
        ##  When calling UEPOCHs, dont put it in a ''
        daoRSObj = self.MakeNewRSConnection('tbl_ClientDataQuantity', specialWhere = "UEPOCH=%s"%UEPOCH)
        ##  WHERE AND is not working so I'm going to have to WHERE for the UEPOCH and then do a quick search for the CLM, to be safe.
        daoRSObj.MoveLast()
        recordCount = daoRSObj.RecordCount
        daoRSObj.MoveFirst()
        ##  This wont allow you to delete if there are multiple records with same UEPOCH
        if recordCount == 1:
            val2 = daoRSObj.Fields('ClientMatterNum').Value
            if val2 == CLM:
                daoRSObj.Delete()
                errorRpt = False
            else:
                errorRpt = True
        else:
            errorRpt = True
        daoRSObj.Close()
        #print errorRpt
        return errorRpt
    
    def GetProducedToEntities(self,CLM):
        """This will return, for a specific matter, a list of the current 'produced to' options, as kind of a history. """
        daoRSObj = self.MakeNewRSConnection('Tbl_MatterProductionDetail',
                                            specialWhere = "ClientMatterNum='%s'"%CLM)
        prodToList = []
        try:
            daoRSObj.MoveLast()
            tableHasData = True
        except:
            tableHasData = False
        productionMatrix = {}
        if tableHasData:
            fullCount = daoRSObj.RecordCount
            daoRSObj.MoveFirst()
            for i in range(fullCount):
                if daoRSObj.Fields('ProducedTo').Value:
                    productionMatrix[daoRSObj.Fields('ProducedTo').Value] = 1
                daoRSObj.MoveNext()
            prodToList = productionMatrix.keys()
        daoRSObj.Close()
        return prodToList
    
    def GetProductionRequestedByNames(self,CLM):
        """This will return, for a specific matter, a list of the current 'production requested by names, as kind of a history. """
        daoRSObj = self.MakeNewRSConnection('Tbl_MatterProductionDetail',
                                            specialWhere = "ClientMatterNum='%s'"%CLM)
        reqByList = []
        try:
            daoRSObj.MoveLast()
            tableHasData = True
        except:
            tableHasData = False
        reqByMatrix = {}
        if tableHasData:
            fullCount = daoRSObj.RecordCount
            daoRSObj.MoveFirst()
            for i in range(fullCount):
                if daoRSObj.Fields('RequestedBy').Value:
                    reqByMatrix[daoRSObj.Fields('RequestedBy').Value] = 1
                daoRSObj.MoveNext()
            reqByList = reqByMatrix.keys()
        daoRSObj.Close()
        return reqByList
        
    def UpdateProductionDetail(self, CLM, prodID, prodProcessedDate, begBates,endBates,prodDocCount, prodPageCount, prodNotes, prodTo, prodMedia,prodSource,prodReqBy,prodSentDate,prodMediaPassword):
        """This method will let you add an entire produciton record to the production table.  All fields are manadatory.
        the date is in mm/dd/yyyy format."""
        try:
            testdaoRSObj = self.MakeNewRSConnection('Tbl_MatterProductionDetail', specialSelect ="ProductionID",specialWhere = "ClientMatterNum='%s'"%CLM)
            testdaoRSObj.MoveLast()
            testRecordCount = testdaoRSObj.RecordCount
            testdaoRSObj.MoveFirst()
            prodList = []
            if testRecordCount:
                for i in range(testRecordCount):
                    prodList.append(testdaoRSObj.Fields('ProductionID').Value)
                    testdaoRSObj.MoveNext()
            testdaoRSObj.Close()
            #print testRecordCount
            #print prodList
            if prodID in prodList:
                prodIncrement = NinoGenTools.Counter(1)
                prodID = prodID+ "_"+"%0*d"%(4,prodIncrement.count)
                prodIncrement.inc()
            while prodID in prodList:
                prodID = prodID[:-4] +"%0*d"%(4,prodIncrement.count)
                prodIncrement.inc()
        except:
            pass
        
        daoRSObj = self.MakeNewRSConnection('Tbl_MatterProductionDetail')
        daoRSObj.AddNew()
        daoRSObj.Fields('ClientMatterNum').Value = CLM
        daoRSObj.Fields('ProductionID').Value = prodID
        daoRSObj.Fields('ProductionComplDate').Value = prodProcessedDate
        daoRSObj.Fields('BegBates').Value = begBates
        daoRSObj.Fields('EndBates').Value = endBates
        daoRSObj.Fields('ProdDocCount').Value = prodDocCount
        daoRSObj.Fields('ProdPageCount').Value = prodPageCount
        daoRSObj.Fields('Notes').Value = prodNotes
        daoRSObj.Fields('ProducedTo').Value =prodTo
        daoRSObj.Fields('ProductionMedia').Value =prodMedia
        daoRSObj.Fields('ProductionSource').Value =prodSource
        daoRSObj.Fields('RequestedBy').Value =prodReqBy
        daoRSObj.Fields('ProductionDate').Value =prodSentDate
        daoRSObj.Fields('ProductionMediaPassword').Value =prodMediaPassword
        
        daoRSObj.Update()
        daoRSObj.Close()
        
    def UpdateResponsibleAttorney(self, CLM, responsibleAttny):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('PrimaryAttorneyContact').Value = responsibleAttny
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetResponsibleAttorney(self, CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('PrimaryAttorneyContact').Value
        daoRSObj.Close()
        return val
        
    def UpdateOtherAttorneys(self,CLM, otherAttorneysList):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        ## add this when aaron adds it.
        daoRSObj.Update()
        daoRSObj.Close()
        
    def UpdateResponsibleParalegal(self,CLM, responsibleParalegal):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('ParalegalContact').Value = responsibleParalegal
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetResponsibleParalegal(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('ParalegalContact').Value
        daoRSObj.Close()
        return val
    
    def UpdatePrimaryOffice(self,CLM, primaryOffice):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('PrimaryOffice').Value = primaryOffice
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetPrimaryOffice(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('PrimaryOffice').Value
        daoRSObj.Close()
        return val
    
    def UpdateResponsibleTPM(self, CLM, tPMName,empID):
        """Format should be last, first"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('TPM').Value = tPMName
        daoRSObj.Fields('EmpID').Value = empID
        daoRSObj.Update()
        daoRSObj.Close()
    
    def GetResponsibleTPM(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('TPM').Value
        daoRSObj.Close()
        return val
    
    def GetCaseStatus(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('ProjectStatus').Value
        daoRSObj.Close()
        return val
    
    def UpdateCaseStatus(self,CLM,status):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('ProjectStatus').Value = status
        daoRSObj.Update()
        daoRSObj.Close()
    
    def GetCaseName(self,CLM):
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('CaseName').Value
        daoRSObj.Close()
        return val
    
    def UpdateCaseName(self,CLM,caseName):
        """This is just the internal name for the case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('CaseName').Value = caseName
        daoRSObj.Update()
        daoRSObj.Close()
        
    def UpdateClientMatterNum(self,oldCLM,newCLM):
        """Changes the client matter number for a case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%oldCLM)
        daoRSObj.Edit()
        daoRSObj.Fields('ClientMatterNum').Value = newCLM
        daoRSObj.Update()
        daoRSObj.Close()
    
    def GetAlternateMediaPath(self,CLM):
        """Returns the alternate media path"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('AlternateMediaPath').Value
        daoRSObj.Close()
        return val
    
    def UpdateAlternateMediaPath(self,CLM,alternateMediaPath):
        """Updates the alternate Media Path in the db"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('AlternateMediaPath').Value = alternateMediaPath
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetVendorFolderPath(self,CLM):
        """Returns the vendor Folder path"""
        try:
            daoRSObj = self.MakeNewRSConnection('tbl_VendorFolderPath', specialWhere = "ClientMatterNum='%s'"%CLM)
            val = daoRSObj.Fields('VendorFolderPath').Value
            daoRSObj.Close()
        except:
            val = None
        return val
    
    def UpdateVendorFolderPath(self,CLM,vendorFolderPath):
        """Updates the Vendor Folder Path in the db"""
        #print vendorFolderPath
        try:
            daoRSObj = self.MakeNewRSConnection('tbl_VendorFolderPath', specialWhere = "ClientMatterNum='%s'"%CLM)
            daoRSObj.Edit()
        except:
            daoRSObj = self.MakeNewRSConnection('tbl_VendorFolderPath')
            daoRSObj.Edit()
            daoRSObj.Fields('ClientMatterNum').Value = CLM
        daoRSObj.Fields('VendorFolderPath').Value = vendorFolderPath
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetUploadCostRate(self,CLM):
        """Returns the upload cost for a case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('UploadCost_GB').Value
        daoRSObj.Close()
        return val
    
    def UpdateUploadCostRate(self,CLM, uploadCost):
        """Updates the upload cost for a case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('UploadCost_GB').Value = uploadCost
        daoRSObj.Update()
        daoRSObj.Close()
    
    def GetStorageCostRate(self,CLM):
        """Returns the Storage cost for a case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        val = daoRSObj.Fields('StorageCost_GB').Value
        daoRSObj.Close()
        return val
        
    def UpdateStorageCostRate(self,CLM, storageCost):
        """Updates the upload cost for a case"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('StorageCost_GB').Value = storageCost
        daoRSObj.Update()
        daoRSObj.Close()
        
    def GetDefaultUploadCostRate(self, platform = "Relativity"):
        """Returns the current default upload cost"""
        if platform == "Relativity":
            cost = "50"
        else:
            cost = "50"
        return cost
    
    def GetDefaultStorageCostRate(self, platform = "Relativity"):
        """Returns the current default storage cost"""
        if platform == "Relativity":
            cost = "20"
        else:
            cost = "20"
        return cost
    
class ExpeDatConnection:
    def __init__(self, platform):
        if platform == "Relativity":
            self.initialPath = "Relativity"
            #self.userName = "eborges"
        else:
            self.initialPath = "Concordance Data"
            #self.userName = "eborgesc"
            
        self.userName = "eborges"
        #self.hostName = "@mweftp.litigation.lexisnexis.com"
        self.hostName = "@transfer.mcdermottdiscovery.com"
        self.userPassword = "9atrEFeh"
        #self.exeLocation = r"C:\Documents and Settings\eborges\My Documents\MyDownloads\ExpDat\movedat.exe"
        self.exeLocation = r"C:\Program Files\ExpDat\movedat.exe"
        
    def ListDir(self,path):
        """Returns the contents of a directory. returns files, dirs."""
        print "debug: listing dir"
        path = os.path.join(self.initialPath,path)
        args = [self.exeLocation,"-K","%s:%s%s:%s"%(self.userName,self.userPassword,self.hostName,path)]
        fnull = open(os.devnull, 'w')
        process = subprocess.Popen(args,stdout = subprocess.PIPE, stderr = fnull)
        rawFileList,errCode =process.communicate()
        rawFileList = rawFileList.split("\n")
        fileList = []
        dirList = []
        if rawFileList:
            for i in rawFileList:
                if i:
                    i = i.split("  ")[-1]
                    if i[-1] == "/":
                        dirList.append(i)
                    else:
                        fileList.append(i)
        fnull.close()
        print "debug: dir listed."
        return fileList, dirList
    
    def ExpeDatWalk(self,top, topdown=True, onerror=None):
        """
        Generator that yields tuples of (root, dirs, nondirs).
        """
        ##  Here I'm creating a generator like os.walk
        
        # Make the FTP object's current directory to the top dir.
        #ftp.cwd(top)
    
        # We may not have read permission for top, in which case we can't
        # get a list of the files the directory contains.  os.path.walk
        # always suppressed the exception then, rather than blow up for a
        # minor reason when (say) a thousand readable directories are still
        # left to visit.  That logic is copied here.
        #try:
        nondirs,dirs = self.ListDir(top)
        #except os.error, err:
        #    if onerror is not None:
        #        onerror(err)
        #    return
    
        if topdown:
            yield top, dirs, nondirs
        for entry in dirs:
            #dname = entry[0]
            dname = entry
            path = os.path.join(top, dname)
            print path
            #if entry[-1] is None: # not a link
            for x in self.ExpeDatWalk(path, topdown, onerror):
                yield x
        if not topdown:
            yield top, dirs, nondirs    

       
    def GatherFullDirListMatrix(self,vendorFolder):
        """Returns a matrix of a list path, with all the files and dirs in a vendor folder"""
        ## UPDATE TO USE GENERATOR ABOVE INSTEAD
        fileList,dirList = self.ListDir(vendorFolder)
        folderMatrix = []
        currentPath = vendorFolder
        for indFile in fileList:
            folderMatrix.append(indFile)
        #while dirList:
        for dir in dirList:
            #print dir
            newPath = os.path.join(currentPath, dir)
            print currentPath
            subFileList,subDirList = self.ListDir(newPath)
            newList = []
            for indFile in subFileList:
                newList.append(indFile)
                #print indFile
            folderMatrix.append([dir,newList])
        #print folderMatrix
        #while dirList:
        return folderMatrix
    
    def GatherSize(self,path):
        """Returns the size of a given path or archive"""
        extension = os.path.splitext(path)[1]
        path = os.path.join(self.initialPath,path)
        #path = self.initialPath + "/" + path
        if extension.upper() == '.ZIP':
            args = [self.exeLocation,"-D","-K","%s:%s%s:%s=zipsize"%(self.userName,self.userPassword,self.hostName,path)]
            fnull = open(os.devnull, 'w')
            process = subprocess.Popen(args,stdout = subprocess.PIPE, stderr = fnull)
            rawFileSize,errCode =process.communicate()
        elif extension == "":
            args = [self.exeLocation,"-K","%s:%s%s:%s=*lr"%(self.userName,self.userPassword,self.hostName,path)]
            #print args
            fnull = open(os.devnull, 'w')
            process = subprocess.Popen(args,stdout = subprocess.PIPE, stderr = fnull)
            rawDirList,errCode =process.communicate()
            rawDirList = rawDirList.split("\n")
            rawFileSize = 0
            for line in rawDirList[1:]:
                #print "Debug:" + line
                if line:
                    size = line.split("\t")[1]
                    itemType = line.split("\t")[3]
                    if itemType == "F":
                        rawFileSize = rawFileSize + int(size,16)
        else:
            args = [self.exeLocation,"-K","%s:%s%s:%s=*ls"%(self.userName,self.userPassword,self.hostName,path)]
            #print args
            fnull = open(os.devnull, 'w')
            process = subprocess.Popen(args,stdout = subprocess.PIPE, stderr = fnull)
            rawDirList,errCode =process.communicate()
            #print rawDirList
            #outputFile = open(r"c:\test.txt",'w')
            #outputFile.writelines(rawDirList)
            #outputFile.close()
            
            rawDirList = rawDirList.split("\n")[1]
            #print rawDirList
            rawFileSize = 0
            if rawDirList:
                #print "Debug:" + rawDirList
                size = rawDirList.split("\t")[1]
                rawFileSize = int(size,16)
        fnull.close()
        if rawFileSize:
            print "size is %s"%rawFileSize
        else:
            rawFileSize = False
        return rawFileSize
    
    def TestPath(self,path):
        """Tests to see if path already exists"""
        #print "debug: Testing path"
        path = os.path.join(self.initialPath,path)
        args = [self.exeLocation,"-K","%s:%s%s:%s"%(self.userName,self.userPassword,self.hostName,path)]
        #print args
        fnull = open(os.devnull, 'w')
        errCode = subprocess.call(args,stdout = fnull, stderr = fnull)
        fnull.close()
        if errCode == 0:
            return True
        if errCode == 25:
            return False
        else:
            print "OTHER ERROR CODE %s. CALL MANNY!"%str(errCode)
        #print "debug: Testing path done."
        
    def CreateNewPath(self,path):
        """Creates a new path on the LN system"""
        path = os.path.join(self.initialPath,path)
        #print "debug: creating new path %s"% path
        args = [self.exeLocation,"-n","-K","%s:%s%s:%s"%(self.userName,self.userPassword,self.hostName,path)]
        fnull = open(os.devnull, 'w')
        errCode = subprocess.call(args,stdout = fnull, stderr = fnull)
        fnull.close()
        #return errCode
        if errCode == 0:
            return True
        if errCode == 34:
            ##  Path already exists
            return False
        else:
            print "OTHER ERROR CODE %s. CALL MANNY!"%str(errCode)
        #print "debug: new path created."

    def MoveOnLN(self,absSourcePathAndFile,targetDirAndFile):
        """Performs a move from LN to LN"""
        targetDirAndFile = os.path.join(self.initialPath,targetDirAndFile)
        print targetDirAndFile
        absSourcePathAndFile = os.path.join(self.initialPath,absSourcePathAndFile)
        print absSourcePathAndFile
        #targetDir = targetDir + "\\"
        args = [self.exeLocation,"-m","-K","%s:%s%s:%s"%(self.userName,self.userPassword,self.hostName,absSourcePathAndFile),"%s"%targetDirAndFile]
        fnull = open(os.devnull, 'w')
        errCode = subprocess.call(args,stdout = fnull, stderr = fnull)
        fnull.close()
        #return errCode
        if errCode == 0:
            return True
        if errCode == 34:
            ##  Path already exists
            return False
        else:
            print "OTHER ERROR CODE %s. CALL MANNY!"%str(errCode)
            
    def CopyToLN(self,absSourcePathAndFile,targetDir):
        """copies absPath to LN. targetDir should be a dir not a file."""
        ##  Expedat requires that if it's a target dir, you end it in a trailing slash
        targetDir = os.path.join(self.initialPath,targetDir)
        targetDir = targetDir + "\\"
        args = [self.exeLocation,"-K",absSourcePathAndFile,"%s:%s%s:%s"%(self.userName,self.userPassword,self.hostName,targetDir)]
        fnull = open(os.devnull, 'w')
        errCode = subprocess.call(args,stdout = fnull, stderr = fnull)
        fnull.close()
        #return errCode
        if errCode == 0:
            return True
        if errCode == 25:
            """The target path does not exist"""
            return False
        elif errCode == 29:
            """The source file or path does not exist"""
            return False
        else:
            print "OTHER ERROR CODE %s. CALL MANNY!"%str(errCode)
            
class MweFtpConnection:
    def __init__(self):
        self.userName = "eborges"
        self.hostName = "disftp.mwe.com"
        self.userPassword = "rever78"
        self.connection = ftplib.FTP_TLS(self.hostName)
        self.connection.login(self.userName,self.userPassword)
        self.connection.prot_p()
    
    def TestPath(self,path):
        startDir = self.connection.pwd()
        try:
            self.connection.cwd(path)
            sucess = True
        except:
            sucess = False
        self.connection.cwd(startDir)
        return sucess
    
    def CreateNewPath(self,path):
        self.connection.mkd(path)
        
    def CopyToDis(self,absSourcePathAndFile,targetDir):
        startDir = self.connection.pwd()
        fileName = os.path.split(absSourcePathAndFile)[1]
        self.connection.cwd(targetDir)
        activePackage = open(absSourcePathAndFile,'rb')
        self.connection.storbinary("STOR %s"%fileName,activePackage)
        activePackage.close()
        self.connection.cwd(startDir)
        
    def CloseConnection(self):
        self.connection.close()
Revision:	459
Committed:	Fri Sep 20 20:32:02 2013 UTC (12 years, 6 months ago) by nino.borges
Content type:	text/x-python
File size:	48743 byte(s)
Log Message:	Added support for transffering cases between TPMs in other offices and intra office. Updated the main View Edit to support this. Now just need a busy dialog and a dialog for transfering through the menu option.