python-build/pythonbrew/curl.py

import sys
import re
import subprocess
from subprocess import Popen, PIPE
from pythonbrew.log import logger
from pythonbrew.util import to_str
from pythonbrew.exceptions import CurlFetchException

class Curl(object):
    def __init__(self):
        returncode = subprocess.call("command -v curl > /dev/null", shell=True)
        if returncode:
            logger.log("pythonbrew required curl. curl was not found in your path.")
            sys.exit(1)
    
    def read(self, url):
        p = Popen("curl -skL %s" % url, stdout=PIPE, shell=True)
        p.wait()
        if p.returncode:
            raise Exception('Failed to read.')
        return p.stdout.read()
    
    def readheader(self, url):
        p = Popen("curl --head -skL %s" % url, stdout=PIPE, shell=True)
        p.wait()
        if p.returncode:
            raise Exception('Failed to readheader.')
        respinfo = {}
        for line in p.stdout:
            line = to_str(line.strip())
            if re.match('^HTTP.*? 200 OK$', line):
                break
        for line in p.stdout:
            line = to_str(line.strip()).split(":", 1)
            if len(line) == 2:
                respinfo[line[0].strip().lower()] = line[1].strip()
        return respinfo
    
    def fetch(self, url, filename):
        p = Popen("curl -# -kL %s -o %s" % (url, filename), shell=True)
        p.wait()
        if p.returncode:
            raise CurlFetchException('Failed to fetch.')