Esempio n. 1
0
def test_replace():
    with warnings.catch_warnings():
        fname = os.tmpnam()
    
    t1 = 'some data'
    t2 = 'other text'
    header = '-' * 50
    footer = '=' * 50
    
    mkfl = lambda x : header + '\n' + x + '\n' + footer
    
    core.put_rf(fname, mkfl(t1))
    try:
        ok(core.replace_in_file(fname, t2, t1)) == False
        ok(core.replace_in_file(fname, t1, t2)) == True
        ok(core.get_rf(fname)) == mkfl(t2)
        
        sudo('chown root.root ' + fname)
        sudo('chmod o-w ' + fname)
        
        ok(lambda : core.replace_in_file(fname, t2, t1)).raises(SystemExit)
        
        ok(core.replace_in_file(fname, t1, t2, use_sudo=True)) == False
        ok(core.replace_in_file(fname, t2, t1, use_sudo=True)) == True
        
        ok(core.get_rf(fname)) == mkfl(t1)
    finally:
        sudo('rm ' + fname)
Esempio n. 2
0
def test_get_put():
    with warnings.catch_warnings():
        fname = os.tmpnam()
    
    t1 = 'some data'
    t2 = 'other text'
    
    core.put_rf(fname, t1)
    
    try:
        ok(core.get_rf(fname)) == t1
        
        sudo('chown root.root ' + fname)
        sudo('chmod o-w ' + fname)
        
        ok(lambda : core.put_rf(fname, t2)).raises(SystemExit)
        
        core.put_rf(fname, t2, use_sudo=True)
        ok(core.get_rf(fname)) == t2
        
        with core.remote_fl(fname, use_sudo=True) as fc:
            ok(fc.getvalue()) == t2
            fc.setvalue(t1)

        ok(core.get_rf(fname)) == t1
            
    finally:
        sudo('rm ' + fname)
Esempio n. 3
0
def run_sensor(opts):
    put_sensor()
    sensor_output_file = start_sensor(opts, run)
    sensor_result = {}
    
    try:
        yield sensor_result
        sensor_result.update(parse_sensor_file(get_rf(sensor_output_file)))
    finally:
        stop_sensor(run, sensor_output_file)
Esempio n. 4
0
def add_supervisor_prog(name, **params):
    scfg = get_rf(supervisord_config)
    prog_line = '[program:{0}]'.format(name)
    if prog_line not in scfg:

        lines = [prog_line]
        lines.extend("{0}={1}".format(*itm) for itm in params.items())

        scfg =  scfg.rstrip() + "\n" + "\n".join(lines) + "\n"
        
        put_rf(supervisord_config, scfg, use_sudo=True)
Esempio n. 5
0
def install_oracle_java(ver='1.6.0_27'):
    
    if ver != '1.6.0_27':
        raise RuntimeError("Only '1.6.0_27' version of java support by now")
    
    JAVAC = check_cmd('javac')
    JAVA_RT = check_cmd('java')
    
    if JAVAC and JAVA_RT:
        return
    
    JAVAH = '/usr/local/java'
    
    JDK_DIR = os.path.join(JAVAH, 'jdk1.6.0_27')
    JRE_DIR = os.path.join(JAVAH, 'jre1.6.0_27')
    
    JDK_URL = 'http://download.oracle.com/otn-pub/java/jdk/6u27-b07/jdk-6u27-linux-x64.bin'
    JRE_URL = 'http://download.oracle.com/otn-pub/java/jdk/6u27-b07/jre-6u27-linux-x64.bin'
    
    if not exists(JAVAH):
        sudo('mkdir ' + JAVAH)

    with cd(JAVAH):
        expect_cmd = 'spawn sh {0};' + \
                     ' expect "Press Enter to continue.....\\n"; send "\\n"'
        
        if not exists(os.path.join(JDK_DIR,'bin/javac')):
            
            if exists(JDK_DIR):
                sudo('rm -rf {0}/*'.format(JDK_DIR))
                
            wget(JDK_URL, use_sudo=True)    
            sudo("expect -c '{0}'".format(
                expect_cmd.format('jdk-6u27-linux-x64.bin')))
            sudo('rm ' + 'jdk-6u27-linux-x64.bin')
            
        if not exists(os.path.join(JRE_DIR,'bin/java')):
            
            if exists(JRE_DIR):
                sudo('rm -rf {0}/*'.format(JRE_DIR))
            
            wget(JRE_URL, use_sudo=True)    
            sudo("expect -c '{0}'".format(
                expect_cmd.format('jre-6u27-linux-x64.bin')))
            sudo('rm ' + 'jre-6u27-linux-x64.bin')
    ln1 = 'export JAVA_HOME=' + JRE_DIR
    ln2 = 'export PATH="$PATH:{0}/bin:{1}/bin"'.format(JRE_DIR, JDK_DIR)
    
    profile = get_rf('/etc/profile')
    put_rf('/etc/profile',
           "\n".join((profile, ln1, ln2, '')),
           use_sudo=True)
Esempio n. 6
0
def add_to_hosts(host, ip):
    cont = get_rf('/etc/hosts')

    for i in cont.split('\n'):
        i = i.strip()
        if i.startswith('#') or i == '' or ' ' not in i:
            continue
        
        fip, fhost = i.split(' ', 1)
        if fhost.strip() == host and fip.strip() == ip:
            return
    
    put_rf('/etc/hosts', cont + '\n{} {}'.format(ip,host))
Esempio n. 7
0
def set_host_name(name):
    fc = get_rf('/etc/hostname')
    if name != fc.strip():
        put_rf('/etc/hostname', name, use_sudo=True)