File: rest.py

package info (click to toggle)
python-bumps 1.0.3-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,200 kB
  • sloc: python: 24,517; xml: 493; ansic: 373; makefile: 211; javascript: 99; sh: 94
file content (154 lines) | stat: -rw-r--r-- 7,060 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
# This code is in the public domain
# Author: Paul Kienzle

# TODO: better cache control; some requests shouldn't be cached
# TODO: compression
# TODO: streaming file transfer
# TODO: single/multifile response objects?

# Inspirations:
#   Upload files in python:
#     http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/146306
#   urllib2_file:
#     Fabien Seisen: <fabien@seisen.org>
#   MultipartPostHandler:
#     Will Holcomb <wholcomb@gmail.com>
#   python-rest-client
#     Benjamin O'Steen
from io import StringIO
from urllib import parse

import email
import httplib2
import mimetypes
import uuid


DEFAULT_CONTENT = "application/octet-stream"


class Connection(object):
    def __init__(self, url, username=None, password=None):
        self.url = url
        http = httplib2.Http()  # ".cache")
        http.follow_all_redirects = True
        if username and password:
            http.add_credentials(username, password)
        self.url = url
        self.http = http

    def get(self, resource, fields={}):
        return _request(self.http, "GET", self.url + resource, fields=fields)

    def head(self, resource, fields={}):
        return _request(self.http, "HEAD", self.url + resource, fields=fields)

    def post(self, resource, fields={}, body=None, mimetype=DEFAULT_CONTENT):
        return _request(self.http, "POST", self.url + resource, fields=fields, body=body, mimetype=mimetype)

    def put(self, resource, fields={}, body=None, mimetype=DEFAULT_CONTENT):
        return _request(self.http, "PUT", self.url + resource, fields=fields, body=body, mimetype=mimetype)

    def postfiles(self, resource, fields={}, files=None):
        return _request(self.http, "POST", self.url + resource, fields=fields, files=files)

    def putfiles(self, resource, fields={}, files=None):
        return _request(self.http, "PUT", self.url + resource, fields=fields, files=files)

    def delete(self, resource, fields={}):
        return _request(self.http, "DELETE", self.url + resource, fields=fields)


def _request(http, verb, location, fields=None, body=None, mimetype=None, files=None):
    headers = {"User-Agent": "Basic Agent"}
    if files:
        if body:
            raise TypeError("Use fields instead of body with file upload")
        # Note: this section is public domain; the old code wasn't working
        boundary = uuid.uuid4().hex
        buf = StringIO()
        for key, value in fields.items():
            buf.write("--%s\r\n" % boundary)
            buf.write('Content-Disposition: form-data; name="%s"' % key)
            buf.write("\r\n\r\n%s\r\n" % value)
        for key, filename in enumerate(files):
            content_type = mimetypes.guess_type(filename)[0] or "application/octet-stream"
            buf.write("--%s\r\n" % boundary)
            buf.write('Content-Disposition: form-data; name="file"; filename="%s"\r\n' % filename)
            buf.write("Content-Type: %s\r\n\r\n" % content_type)
            buf.write(open(filename, "rb").read())
            buf.write("\r\n")
        buf.write("--%s--\r\n" % boundary)
        body = buf.getvalue()
        headers["Content-Type"] = "multipart/form-data; boundary=" + boundary
        headers["Content-Length"] = str(len(body))
        # print "===== body =====\n",body
    elif body:
        if fields:
            raise TypeError("Body, if included, should encode fields directly.")
        headers["Content-Type"] = mimetype
        headers["Content-Length"] = str(len(body))
    elif fields:
        if verb == "GET":
            location += "?" + parse.urlencode(fields)
            body = ""
        else:
            headers["Content-Type"] = "application/x-www-form-urlencoded"
            body = parse.urlencode(fields)

    # print "uri",location
    # print "body",body
    # print "headers",headers
    # print "method",verb
    try:
        response, content = http.request(location, verb, body=body, headers=headers)
    except AttributeError:
        raise IOError("Could not open " + location)

    return response, content.decode("UTF-8")


# Table mapping response codes to messages; entries have the
# form {code: (shortmessage, longmessage)}.
RESPONSE = {
    100: ("Continue", "Request received, please continue"),
    101: ("Switching Protocols", "Switching to new protocol; obey Upgrade header"),
    200: ("OK", "Request fulfilled, document follows"),
    201: ("Created", "Document created, URL follows"),
    202: ("Accepted", "Request accepted, processing continues off-line"),
    203: ("Non-Authoritative Information", "Request fulfilled from cache"),
    204: ("No Content", "Request fulfilled, nothing follows"),
    205: ("Reset Content", "Clear input form for further input."),
    206: ("Partial Content", "Partial content follows."),
    300: ("Multiple Choices", "Object has several resources -- see URI list"),
    301: ("Moved Permanently", "Object moved permanently -- see URI list"),
    302: ("Found", "Object moved temporarily -- see URI list"),
    303: ("See Other", "Object moved -- see Method and URL list"),
    304: ("Not Modified", "Document has not changed since given time"),
    305: ("Use Proxy", "You must use proxy specified in Location to access this " "resource."),
    307: ("Temporary Redirect", "Object moved temporarily -- see URI list"),
    400: ("Bad Request", "Bad request syntax or unsupported method"),
    401: ("Unauthorized", "No permission -- see authorization schemes"),
    402: ("Payment Required", "No payment -- see charging schemes"),
    403: ("Forbidden", "Request forbidden -- authorization will not help"),
    404: ("Not Found", "Nothing matches the given URI"),
    405: ("Method Not Allowed", "Specified method is invalid for this server."),
    406: ("Not Acceptable", "URI not available in preferred format."),
    407: ("Proxy Authentication Required", "You must authenticate with " "this proxy before proceeding."),
    408: ("Request Timeout", "Request timed out; try again later."),
    409: ("Conflict", "Request conflict."),
    410: ("Gone", "URI no longer exists and has been permanently removed."),
    411: ("Length Required", "Client must specify Content-Length."),
    412: ("Precondition Failed", "Precondition in headers is false."),
    413: ("Request Entity Too Large", "Entity is too large."),
    414: ("Request-URI Too Long", "URI is too long."),
    415: ("Unsupported Media Type", "Entity body in unsupported format."),
    416: ("Requested Range Not Satisfiable", "Cannot satisfy request range."),
    417: ("Expectation Failed", "Expect condition could not be satisfied."),
    500: ("Internal Server Error", "Server got itself in trouble"),
    501: ("Not Implemented", "Server does not support this operation"),
    502: ("Bad Gateway", "Invalid responses from another server/proxy."),
    503: ("Service Unavailable", "The server cannot process the request due to a high load"),
    504: ("Gateway Timeout", "The gateway server did not receive a timely response"),
    505: ("HTTP Version Not Supported", "Cannot fulfill request."),
}