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

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 = []
    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
    else:
        myMatterList = accessDB.RetrieveMyCaseList(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)
    return responsibleCases, 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 .17'


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 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 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",
                                            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 = []
        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'
                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 productionList
   
 
    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 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:
                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 UpdateProductionDetail(self, CLM, prodID, prodDate, begBates,endBates,prodDocCount, prodPageCount, prodNotes):
        """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 = prodDate
        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.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):
        """Format should be last, first"""
        daoRSObj = self.MakeNewRSConnection('tbl_MatterTrackingData', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('TPM').Value = tPMName
        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 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"""
        daoRSObj = self.MakeNewRSConnection('tbl_VendorFolderPath', specialWhere = "ClientMatterNum='%s'"%CLM)
        daoRSObj.Edit()
        daoRSObj.Fields('VendorFolderPath').Value = vendorFolderPath
        daoRSObj.Update()
        daoRSObj.Close()
    
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.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."""
        path = os.path.join(self.initialPath,path)
        args = [self.exeLocation,"%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()
        return fileList, dirList
       
    def GatherFullDirListMatrix(self,vendorFolder):
        """Returns a matrix of a list path, with all the files and dirs in a vendor folder"""
        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","%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,"%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:]:
                if line:
                    size = line.split("\t")[1]
                    itemType = line.split("\t")[3]
                    if itemType == "F":
                        rawFileSize = rawFileSize + int(size,16)
        else:
            args = [self.exeLocation,"%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()
            rawDirList = rawDirList.split("\n")[2]
            print rawDirList
            rawFileSize = 0
            if 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"""
        path = os.path.join(self.initialPath,path)
        args = [self.exeLocation,"%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)
        
    def CreateNewPath(self,path):
        """Creates a new path on the LN system"""
        path = os.path.join(self.initialPath,path)
        args = [self.exeLocation,"-n","%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)

    def MoveOnLN(self,absSourcePathAndFile,targetDirAndFile):
        """Performs a move from LN to LN"""
        targetDirAndFile = os.path.join(self.initialPath,targetDirAndFile)
        absSourcePathAndFile = os.path.join(self.initialPath,absSourcePathAndFile)
        #targetDir = targetDir + "\\"
        args = [self.exeLocation,"-m","%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,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:	266
Committed:	Mon Feb 4 19:25:36 2013 UTC (13 years, 1 month ago) by nino.borges
Content type:	text/x-python
File size:	36780 byte(s)
Log Message:	moved the list of archive file types to the lib, so that there would be one place only to change it, and pointed copy up here. Also added a warning dialog to copy up to warn when it's about to gather the compressed size of an archive.