Show More
@@ -6,6 +6,8 b' try:' | |||||
6 | except NameError: |
|
6 | except NameError: | |
7 | pass |
|
7 | pass | |
8 |
|
8 | |||
|
9 | import os | |||
|
10 | ||||
9 | import requests |
|
11 | import requests | |
10 | import getpass |
|
12 | import getpass | |
11 | import json |
|
13 | import json | |
@@ -102,3 +104,101 b' def get_pulls_list(project, github_api=3):' | |||||
102 | if github_api == 2 : |
|
104 | if github_api == 2 : | |
103 | return json.loads(response.text)['pulls'] |
|
105 | return json.loads(response.text)['pulls'] | |
104 | return json.loads(response.text) |
|
106 | return json.loads(response.text) | |
|
107 | ||||
|
108 | # encode_multipart_formdata is from urllib3.filepost | |||
|
109 | # The only change is to iter_fields, to enforce S3's required key ordering | |||
|
110 | ||||
|
111 | def iter_fields(fields): | |||
|
112 | fields = fields.copy() | |||
|
113 | for key in ('key', 'acl', 'Filename', 'success_action_status', 'AWSAccessKeyId', | |||
|
114 | 'Policy', 'Signature', 'Content-Type', 'file'): | |||
|
115 | yield (key, fields.pop(key)) | |||
|
116 | for (k,v) in fields.items(): | |||
|
117 | yield k,v | |||
|
118 | ||||
|
119 | def encode_multipart_formdata(fields, boundary=None): | |||
|
120 | """ | |||
|
121 | Encode a dictionary of ``fields`` using the multipart/form-data mime format. | |||
|
122 | ||||
|
123 | :param fields: | |||
|
124 | Dictionary of fields or list of (key, value) field tuples. The key is | |||
|
125 | treated as the field name, and the value as the body of the form-data | |||
|
126 | bytes. If the value is a tuple of two elements, then the first element | |||
|
127 | is treated as the filename of the form-data section. | |||
|
128 | ||||
|
129 | Field names and filenames must be unicode. | |||
|
130 | ||||
|
131 | :param boundary: | |||
|
132 | If not specified, then a random boundary will be generated using | |||
|
133 | :func:`mimetools.choose_boundary`. | |||
|
134 | """ | |||
|
135 | # copy requests imports in here: | |||
|
136 | from io import BytesIO | |||
|
137 | from requests.packages.urllib3.filepost import ( | |||
|
138 | choose_boundary, six, writer, b, get_content_type | |||
|
139 | ) | |||
|
140 | body = BytesIO() | |||
|
141 | if boundary is None: | |||
|
142 | boundary = choose_boundary() | |||
|
143 | ||||
|
144 | for fieldname, value in iter_fields(fields): | |||
|
145 | body.write(b('--%s\r\n' % (boundary))) | |||
|
146 | ||||
|
147 | if isinstance(value, tuple): | |||
|
148 | filename, data = value | |||
|
149 | writer(body).write('Content-Disposition: form-data; name="%s"; ' | |||
|
150 | 'filename="%s"\r\n' % (fieldname, filename)) | |||
|
151 | body.write(b('Content-Type: %s\r\n\r\n' % | |||
|
152 | (get_content_type(filename)))) | |||
|
153 | else: | |||
|
154 | data = value | |||
|
155 | writer(body).write('Content-Disposition: form-data; name="%s"\r\n' | |||
|
156 | % (fieldname)) | |||
|
157 | body.write(b'Content-Type: text/plain\r\n\r\n') | |||
|
158 | ||||
|
159 | if isinstance(data, int): | |||
|
160 | data = str(data) # Backwards compatibility | |||
|
161 | if isinstance(data, six.text_type): | |||
|
162 | writer(body).write(data) | |||
|
163 | else: | |||
|
164 | body.write(data) | |||
|
165 | ||||
|
166 | body.write(b'\r\n') | |||
|
167 | ||||
|
168 | body.write(b('--%s--\r\n' % (boundary))) | |||
|
169 | ||||
|
170 | content_type = b('multipart/form-data; boundary=%s' % boundary) | |||
|
171 | ||||
|
172 | return body.getvalue(), content_type | |||
|
173 | ||||
|
174 | ||||
|
175 | def post_download(project, filename, name=None, description=""): | |||
|
176 | """Upload a file to the GitHub downloads area""" | |||
|
177 | if name is None: | |||
|
178 | name = os.path.basename(filename) | |||
|
179 | with open(filename, 'rb') as f: | |||
|
180 | filedata = f.read() | |||
|
181 | ||||
|
182 | url = "https://api.github.com/repos/{project}/downloads".format(project=project) | |||
|
183 | ||||
|
184 | payload = json.dumps(dict(name=name, size=len(filedata), | |||
|
185 | description=description)) | |||
|
186 | response = requests.post(url, data=payload, headers=make_auth_header()) | |||
|
187 | response.raise_for_status() | |||
|
188 | reply = json.loads(response.content) | |||
|
189 | s3_url = reply['s3_url'] | |||
|
190 | ||||
|
191 | fields = dict( | |||
|
192 | key=reply['path'], | |||
|
193 | acl=reply['acl'], | |||
|
194 | success_action_status=201, | |||
|
195 | Filename=reply['name'], | |||
|
196 | AWSAccessKeyId=reply['accesskeyid'], | |||
|
197 | Policy=reply['policy'], | |||
|
198 | Signature=reply['signature'], | |||
|
199 | file=(reply['name'], filedata), | |||
|
200 | ) | |||
|
201 | fields['Content-Type'] = reply['mime_type'] | |||
|
202 | data, content_type = encode_multipart_formdata(fields) | |||
|
203 | s3r = requests.post(s3_url, data=data, headers={'Content-Type': content_type}) | |||
|
204 | return s3r |
General Comments 0
You need to be logged in to leave comments.
Login now