Пример #1
0
def upload_file(file_path_list,
                remote_user,
                remote_host,
                remote_path,
                password,
                remote_port=22):
    '''
    Copy a file to remote machine. return 0 if success
    '''
    logger = Logging.getLogger(log_file='run_task.log')
    try:
        file_path = ' '  # multi files
        for path_item in file_path_list:
            file_path += path_item
            file_path += ' '
        if not file_path or not remote_user or not remote_host or not remote_path:
            logger.error('Empty parameter given')
            raise Excpetion, 'Empty parameter given'
        cmd_text = 'scp -P %d %s %s@%s:%s' % \
            (remote_port , file_path, remote_user, remote_host, remote_path)

        print cmd_text
        logger.info(cmd_text)
        global timeout_
        cmd = pexpect.spawn(cmd_text, timeout=timeout_)
        expect_res = cmd.expect(
            ['(yes/no)', 'password', pexpect.EOF, pexpect.TIMEOUT])
        if expect_res == 0:
            logger.info('pexpect expects (yes/no)')
            cmd.sendline('yes')
            cmd.expect('password')
            cmd.sendline(password)
        elif expect_res == 1:
            logger.info('pexpect expects (password)')
            cmd.sendline(password)
        elif expect_res == 2:
            logger.info('pexpect expects (EOF)')
        else:
            logger.info('pexpect expects (TIMEOUT)')
            raise Exception, 'TIMEOUT'
        expect_res = cmd.expect(['Permission denied', pexpect.EOF])
        if expect_res == 0:
            logger.info('Permission denied')
            cmd.close()
            cmd.kill(0)
        else:
            cmd.close()
        return cmd.exitstatus
    except Exception, e:
        cmd.close()
        logger.error(e)
        return 1
def upload_file(file_path_list , remote_user , remote_host ,remote_path , password , remote_port = 22) :
    '''
    Copy a file to remote machine. return 0 if success
    '''
    logger = Logging.getLogger(log_file='upload_remotefile.log')
    try :
        file_path = ' '# multi files
        for path_item in file_path_list :
            file_path += path_item
            file_path += ' '
        if not file_path or not remote_user or not remote_host or not remote_path :
            logger.error('Empty parameter given')
            raise Excpetion , 'Empty parameter given'
        cmd_text = 'scp -P %d %s %s@%s:%s' % \
            (remote_port , file_path, remote_user, remote_host, remote_path)
        
        print cmd_text
        logger.info(cmd_text)
        global timeout_
        cmd = pexpect.spawn(cmd_text,timeout=timeout_)
        expect_res = cmd.expect(['(yes/no)','password',pexpect.EOF,pexpect.TIMEOUT])
        if expect_res == 0 :
            logger.info('pexpect expects (yes/no)')
            cmd.sendline('yes')
            cmd.expect('password')
            cmd.sendline(password)
        elif expect_res == 1 :
            logger.info('pexpect expects (password)')
            cmd.sendline(password)
        elif expect_res == 2 :
            logger.info('pexpect expects (EOF)')
        else:
            logger.info('pexpect expects (TIMEOUT)')
            raise Exception , 'TIMEOUT'
        expect_res = cmd.expect(['Permission denied',pexpect.EOF])
        if expect_res == 0 :
            logger.info('Permission denied')
            cmd.close()
            cmd.kill(0)
        else :
            cmd.close()
        return cmd.exitstatus
    except Exception ,e :
        cmd.close()
        logger.error(e)
        return 1
def stop_remotecommand(key_code_list, server_array) :
    logger = Logging.getLogger(log_file='upload_remotefile.log')
 
    log_code = key_code_list[0][:len(key_code_list[0])/2]
    log_name = log_code + '.log'
    result = Queue.Queue()
    def do_stop_query(key_code_list, machine) :
        try:
            s = pxssh.pxssh()
            is_login = s.login(server = machine['remote_host'], username = machine['remote_user'], password = machine['password'] , login_timeout = 100)
            if is_login == False :
                raise Exception , 'Permission denied'
            else: 
                print 'login success'
                
            cnt = 0
            while(True) :
                s.sendline('ps -aux | grep --color=never ' + log_code)
                s.prompt()
                words = s.before.split()
                index = 0
                PID = -1
                FIND = False
                if machine['remote_path'].endswith('/') == False :
                    machine['remote_path'] += '/'
                for item in words :
                    if item.find(machine['remote_path'] + log_code) >= 0 :
                        PID = int(words[index-10])
                        for code in key_code_list :
                            if item.find(code) >= 0 :
                                FIND = True
                                break
                        break
                    index+=1
                
                logger.info("Find %s, key_code_list=%s" % (str(FIND), key_code_list))
                print "Find %s, key_code_list=%s" % (str(FIND), key_code_list)
                if FIND == False or cnt > 10:
                    cmd_text = ''
                    log_name_ = machine['remote_path'] + log_name
                    if cnt > 10 :
                        cmd_text = 'echo "failed to stop: ' + log_code + '" >> ' + log_name_ + ' &'
                    else :
                        cmd_text = 'echo "manumally stop: ' + log_code + '" >> ' + log_name_ + ' &'
		    print cmd_text
                    s.sendline(cmd_text)
                    s.prompt()
                    break #double check
                else:
                    print "kill: PID %s, key_code_list=%s" % (PID, key_code_list)
                    logger.info("kill: PID %s, key_code_list=%s" % (PID, key_code_list))
                    cnt += 1
                    if cnt > 0 :
                        kill_res = s.sendline('kill -9 '+str(PID))
                    else:
                        kill_res = s.sendline('kill '+str(PID))
                    s.prompt()
            s.logout()
            result.put([s.exitstatus,key_code_list,log_code,machine])#success
        except Exception, e: 
            print e
            result.put([-1,[],'',machine])#fail