from fabric.api import *

env.hosts = [
    'root@10.211.55.12:32772',
    'root@10.211.55.12:32773',
    'root@10.211.55.12:32774',
]
# Set the username
#env.user = "root"

# Set the password [NOT RECOMMENDED]
env.password = "test123"
# env.passwords = {'root@10.211.55.12:32772': 'test123', 'root@10.211.55.12:32773': 'test123', 'root@10.211.55.12:32774': 'test123',}

# env.hosts = ['user1@host1:port1', 'user2@host2.port2']
# env.passwords = {'user1@host1:port1': 'password1', 'user2@host2.port2': 'password2'}

@parallel
def install():
    cmd = "celery -A test_celery worker --app=test_celery.celeryapp:app --concurrency=10 --loglevel=debug"
    run("apt-get install dtach && apt-get install -y python-pip && pip install celery && pip install requests && pip install pymongo")
    with settings(warn_only=True):
        runResult = run("mkdir -p /root/code/test_celery")
        if runResult.return_code == 1:
            print 'file exist ok'
        with cd('/root/code'):
            result = run("ls -l")
            a = put("./test_celery/celeryapp.py", "/root/code/test_celery")
            b = put("./test_celery/run_tasks.py", "/root/code/test_celery")
            c = put("./test_celery/tasks.py", "/root/code/test_celery")
            d = put("./test_celery/ua.txt", "/root/code/test_celery")
            d = put("./test_celery/__init__.py", "/root/code/test_celery")
            e = run('dtach -n `mktemp -u /tmp/%s.XXXX` %s' % ('dtach', cmd))
    return result, a.succeeded, b.succeeded, c.succeeded, d.succeeded, e.succeeded