Beispiel #1
0
def addToCustomer(mc,island,custId,corpus='/tmp/corpus'):
    print 'addToCustomer()'
    amptool = ActiveMailboxPartitionTool()
    emscmdtool = EmsCommandLineTool()

    print 'executing amp cluster-location'
    amptool.runCommand('cluster-location',array(['-r'],String))


    args = []
    args.append("-f" )
    args.append( str(island))
    args.append("-s" )
    args.append( str(island))

    args.append( "-i" )
    args.append( corpus)
    args.append( "-o" )
    args.append( "/tmp")
    args.append( "-r" )
    args.append( "equal-to-sent")
    args.append( "-c")
    args.append(str(custId))
    args.append( "-q")
    argsArr = array(args,String)

    print 'executing make-test-customer'
    result = emscmdtool.runCommand('make-test-customer',argsArr)

    if result is False:
        raise Exception('make-test-customer failed, customer already exists?')
    ManagementContainer.getInstance().getMailRoutingConfigGenerator().waitForRegenComplete()

    # ensure policy_exploded_users updates with user account info
    rpm = mc.getRetentionPolicyManager()
    rpm.notifyOfExternalPolicyChanges(int(custId))

    # move the contents of staging to sftp.drop
    src = os.path.join('/tmp',str(custId))
    dst = os.path.join('/ems/bigdisk/sftp.drop',str(custId))
    if not os.path.exists(dst):
        os.mkdir(dst)
    for f in os.listdir(src):
        if f.endswith('.gzip.enc.done.ftp'):
            shutil.copy(os.path.join(src,f),dst)
            os.remove(os.path.join(src,f))

            idxF = f.replace('.gzip.enc.done.ftp','.index.done.ftp')
            shutil.copy(os.path.join(src,idxF),dst)
            os.remove(os.path.join(src,idxF))

    shutil.rmtree(src)

    return
Beispiel #2
0
def setupCustomer(mc,island,domain,isCloud,isOnPremises,userAccounts,loadCorpus=False):
    deleteCustomer(mc)
    amptool = ActiveMailboxPartitionTool()
    emscmdtool = EmsCommandLineTool()
    print 'creating customer...'
    cm = mc.getCustomerManager()
    print 'executing amp cluster-location'
    amptool.runCommand('cluster-location',array(['-r'],String))



    args = []
    args.append("-f" )
    args.append( str(island))
    args.append("-s" )
    args.append( str(island))
    args.append("-d" )
    args.append( domain)
    args.append( "-n" )
    args.append("systest_journaling_customer" )
    args.append("-e" )
    args.append( "nobody@" + domain)
    args.append("-q")
    if loadCorpus:
        args.append( "-i" )
        args.append( "/tmp/corpus")
        args.append( "-o" )
        args.append( "/ems/bigdisk/sftp.drop")

    if isCloud:
        args.append("--guid")
    if isOnPremises:
        args.append( "--bcc" )

    if userAccounts is not None:
        for address in userAccounts:
            args.append("--user-account" )
            args.append( address + "@" + domain )
    print "creating journal-enabled customer "
    Logger.getLogger( "cloud.py" ).info( "Creating systest_journaling_customer..." )
    result = emscmdtool.runCommand('make-test-customer',args)

    if result is False:
        raise Exception('make-test-customer failed, customer already exists?')
    ManagementContainer.getInstance().getMailRoutingConfigGenerator().waitForRegenComplete()
    cust = cm.findCustomers([SearchConstraint(ICustomerManager.PROP_NAME, SearchConstraintOperator.CONSTRAINT_EQUALS, 'systest_journaling_customer')])
    return cust[0]
Beispiel #3
0
def expireWinAuthPassword(email):
    emscmdtool = EmsCommandLineTool()
    mc = ManagementContainer.getInstance()
    connection = mc.getDBConnection(ManagementContainer.POOL_NAME)
    s = connection.createStatement()
    command = "update dat_user_account set password_expire_time=creation_date where primary_email = \'" + email + "\'"
    print "EXEC: " + command
    s.executeUpdate(command)
    connection.commit()
    emscmdtool.runCommand('clean-expired-passwords',array([''],String))

    command = "select hashed_password,hashed_imap_password from dat_user_account where primary_email = \'" + email + "\'"
    print "EXEC: " + command
    resultSet = s.executeQuery(command)
    resultSet.next()
    hashedPassword = resultSet.getString(1)
    hashedImapPassword = resultSet.getString(2)
    mc.safeReturnDBConnection(connection, ManagementContainer.POOL_NAME)
    DbUtils.safeClose(s)
    if not hashedPassword is None or not hashedImapPassword is None:
        print "Hashed password=\"", hashedPassword, "\" Hashed Imap Password=\"", hashedImapPassword, "\""
        sys.exit(1)
Beispiel #4
0
def setupCustomer(mc,island,corpus='/tmp/corpus',domain='enron.com',cleanupFirst=True,name='systest_customer',recvDate='equal-to-sent',enableFolderSync=False,runAsync=False,isCloud=False,isOnPremises=False,userAccounts=[],folderid=None, corpusFolders=False,
                  sourceDomain='enron.com',copyMail = False):

    print 'setupCustomer()'
    amptool = ActiveMailboxPartitionTool()
    file = File('/tmp/%s.log')
    emscmdout = PrintStream(FileOutputStream(file))
    emscmdtool = EmsCommandLineTool()
    emscmdtool.setOutputStream(emscmdout)


    if cleanupFirst:
        customerList=[name]
        domainList=[domain]
        deleteCustomers( customerList, domainList )

    if (sourceDomain is not None)  and (domain != sourceDomain):
        print 'renaming domain in corpus', corpus, 'from', sourceDomain, 'to', domain
        os.system("find " + corpus +" -type f -print | xargs -n 25 sed -i 's/"+sourceDomain+"/"+domain+"/g'")


    cm = mc.getCustomerManager()
    amptool.runCommand('cluster-location',array(['-r'],String))

    if runAsync is False:

        args = []
        args.append("-f" )
        args.append( str(island))
        args.append("-s" )
        args.append( str(island))

        args.append( "-i" )
        args.append( corpus)
        args.append( "-o" )
        args.append( "/ems/bigdisk/sftp.drop")
        args.append( "-r" )
        args.append( recvDate)
        args.append("-d" )
        args.append( domain)
        args.append( "-n" )
        args.append( name)
        args.append("-e" )
        args.append( "nobody@" + domain)
        args.append("-q")
        #args.append("-t")
        #args.append("10")
        if copyMail:
            arg.append(" -y")

        if enableFolderSync:
            args.append( "-x" )

        if isCloud:
            args.append("--guid")

        if folderid is not None:
            args.append("-j")

            args.append(folderid)
        elif corpusFolders is True:
            args.append( "-k" )

        if isOnPremises:
            args.append( "--bcc" )

        if userAccounts is not None:
            for address in userAccounts:
                args.append("--user-account" )
                args.append( address + "@" + domain )



        argsArr = array(args,String)
        print 'executing make-test-customer command'
        result = emscmdtool.runCommand('make-test-customer',argsArr)
        if result is False:
            raise Exception('make-test-customer failed, customer already exists?')
        ManagementContainer.getInstance().getMailRoutingConfigGenerator().waitForRegenComplete()
        custIds = cm.findCustomers([SearchConstraint(ICustomerManager.PROP_NAME, SearchConstraintOperator.CONSTRAINT_EQUALS, name)])
        custid = custIds[0].getCustID()
    if runAsync:

        cmdline = 'ulimit -n 4096 ; /opt/ems/bin/emscmd ' + ' make-test-customer ' + ' -f ' + str(island) + ' -s ' + str(island)
        cmdline = cmdline + ' -d ' + domain + ' -n ' + ' ' + name + ' '
        cmdline = cmdline + ' -e ' + ' nobody@' + domain
        cmdline = cmdline + " -q"
        if copyMail:
            cmdline = cmdline + " -y"

        if enableFolderSync:
            cmdline = cmdline + " -x"

        if isCloud:
            cmdline = cmdline + ' --guid'

        if folderid is not None:
            cmdline = cmdline + " -j '"+folderid+"'"
        elif corpusFolders is True:
            cmdline = cmdline + " -k"

        if isOnPremises:
            cmdline = cmdline + ' --bcc'

        if userAccounts is not None:
            for address in userAccounts:
                cmdline = cmdline + ' --user-account "' + address + '@' + domain  + '"'

        print 'executing',cmdline

        rc = os.system('/bin/bash -c ' + cmdline)
        if rc != 0:
            raise Exception('make-test-customer failed, customer already exists?')
        custIds = cm.findCustomers([SearchConstraint(ICustomerManager.PROP_NAME, SearchConstraintOperator.CONSTRAINT_EQUALS, name)])
        custid = custIds[0].getCustID()

        cmdline = 'ulimit -n 4096 ; /opt/ems/bin/emscmd ' + ' make-test-customer ' + ' -f ' + str(island) + ' -s ' + str(island)
        cmdline = cmdline + ' -i ' + corpus
        cmdline = cmdline + ' -o ' + ' /ems/bigdisk/sftp.drop '
        cmdline = cmdline + ' -c ' + str(custid)
        cmdline = cmdline + ' -r ' + ' ' + recvDate + ' '
        cmdline = cmdline + " -q"
        if copyMail:
            cmdline = cmdline + " -y"

        if enableFolderSync:
            cmdline = cmdline + " -x"

        cmdline = cmdline + ' >/tmp/%s.log 2>&1' % name
        print 'executing',cmdline
        os.popen(cmdline,'r')


    return custid