Append ssh public key to remote servers with fabric

fabfile.py

from fabric.api import local, run, env, cd
from fabric.contrib import files
import os

#env.hosts = []

PUB_KEY = "~/.ssh/id_rsa.pub"

def install_public_key(pubkey_file=PUB_KEY):
    with open(os.path.expanduser(pubkey_file)) as fd:
        ssh_key = fd.readline().strip()
    with cd('~/'):
        run('mkdir -p .ssh')
        run('chmod 700 ~/.ssh')
        files.append('.ssh/authorized_keys', ssh_key)
        run('chmod 644 .ssh/authorized_keys')