This repository was archived by the owner on Aug 2, 2025. It is now read-only.
forked from cztomczak/cefpython
-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathrequest_cef1.pyx
More file actions
208 lines (188 loc) · 8.71 KB
/
request_cef1.pyx
File metadata and controls
208 lines (188 loc) · 8.71 KB
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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
# Copyright (c) 2012-2014 The CEF Python authors. All rights reserved.
# License: New BSD License.
# Website: http://code.google.com/p/cefpython/
class Request:
Flags = {
"None": cef_types.WUR_FLAG_NONE,
"SkipCache": cef_types.WUR_FLAG_SKIP_CACHE,
"AllowCachedCredentials": cef_types.WUR_FLAG_ALLOW_CACHED_CREDENTIALS,
"AllowCookies": cef_types.WUR_FLAG_ALLOW_COOKIES,
"ReportUploadProgress": cef_types.WUR_FLAG_REPORT_UPLOAD_PROGRESS,
"ReportLoadTiming": cef_types.WUR_FLAG_REPORT_LOAD_TIMING,
"ReportRawHeaders": cef_types.WUR_FLAG_REPORT_RAW_HEADERS,
}
def __init__(self):
# Request object is just a public API wrapper,
# the real Request object is named PyRequest.
raise Exception("Request object cannot be instantiated directly, "
"use static method Request.CreateRequest()")
@staticmethod
def CreateRequest():
cdef CefRefPtr[CefRequest] cefRequest = CefRequest_Create()
cdef PyRequest pyRequest = CreatePyRequest(cefRequest)
return pyRequest
cdef PyRequest CreatePyRequest(CefRefPtr[CefRequest] cefRequest):
# This can't be named "GetPyRequest()" as CefRequest has
# no unique identifier, so each time a different python object
# must be returned.
cdef PyRequest pyRequest = PyRequest()
pyRequest.cefRequest = cefRequest
return pyRequest
cdef class PyRequest:
cdef CefRefPtr[CefRequest] cefRequest
cdef CefRefPtr[CefRequest] GetCefRequest(self) except *:
if <void*>self.cefRequest != NULL and self.cefRequest.get():
return self.cefRequest
raise Exception("PyRequest.GetCefRequest() failed: "
"CefRequest was destroyed")
cpdef str GetUrl(self):
return CefToPyString(self.GetCefRequest().get().GetURL())
cpdef py_void SetUrl(self, py_string url):
cdef CefString cefUrl
PyToCefString(url, cefUrl)
self.GetCefRequest().get().SetURL(cefUrl)
cpdef str GetMethod(self):
return CefToPyString(self.GetCefRequest().get().GetMethod())
cpdef py_void SetMethod(self, py_string method):
cdef CefString cefMethod
PyToCefString(method, cefMethod)
self.GetCefRequest().get().SetMethod(cefMethod)
cpdef object GetPostData(self):
if self.GetMethod() != "POST":
return {}
cdef cpp_vector[CefRefPtr[CefPostDataElement]] elementVector
cdef CefRefPtr[CefPostData] postData = (
self.GetCefRequest().get().GetPostData())
if postData.get().GetElementCount() == 0:
return {}
postData.get().GetElements(elementVector)
cdef cpp_vector[CefRefPtr[CefPostDataElement]].iterator iterator = (
elementVector.begin())
cdef CefRefPtr[CefPostDataElement] postDataElement
cdef list retMultipart = []
cdef dict retUrlEncoded = {}
# pyData is really of type "str", but Cython will throw
# an error if we use that type: "Cannot convert 'bytes'
# object to str implicitly. This is not portable to Py3."
cdef object pyData
cdef size_t bytesCount
cdef void* voidData
cdef str pyFile
while iterator != elementVector.end():
postDataElement = deref(iterator)
if postDataElement.get().GetType() == cef_types.PDE_TYPE_EMPTY:
# May return an empty dict - retUrlEncoded.
pass
elif postDataElement.get().GetType() == cef_types.PDE_TYPE_BYTES:
bytesCount = postDataElement.get().GetBytesCount()
voidData = <void*>malloc(bytesCount)
postDataElement.get().GetBytes(bytesCount, voidData)
pyData = VoidPtrToString(voidData, bytesCount)
free(voidData)
if (pyData.startswith('--') or retMultipart):
# Content-Type: multipart/form-data
retMultipart.append(pyData)
else:
# Content-Type: application/x-www-form-urlencoded
retUrlEncoded.update(urlparse.parse_qsl(qs=pyData,
keep_blank_values=True))
elif postDataElement.get().GetType() == cef_types.PDE_TYPE_FILE:
pyFile = CefToPyString(postDataElement.get().GetFile())
retMultipart.append("@"+pyFile)
else:
raise Exception("Invalid type of CefPostDataElement")
preinc(iterator)
if retMultipart:
return retMultipart
else:
return retUrlEncoded
cpdef py_void SetPostData(self, object pyPostData):
cdef CefRefPtr[CefPostData] postData = (
self.GetCefRequest().get().GetPostData())
postData.get().RemoveElements()
cdef CefRefPtr[CefPostDataElement] postDataElement
cdef py_string pyElement
cdef CefString sfile
if type(pyPostData) == list:
for pyElement in pyPostData:
if pyElement.startswith('--'):
postDataElement = CefPostDataElement_Create()
postDataElement.get().SetToBytes(len(pyElement),
<char*>pyElement)
elif pyElement.startswith('@'):
postDataElement = CefPostDataElement_Create()
PyToCefString(pyElement[1:], sfile)
postDataElement.get().SetToFile(sfile)
elif not pyElement:
postDataElement = CefPostDataElement_Create()
postDataElement.get().SetToEmpty()
else:
raise Exception("Invalid element in postData: %s" % (
pyElement))
postData.get().AddElement(postDataElement)
elif type(pyPostData) == dict:
pyElement = urllib.urlencode(pyPostData)
pyElement = str(pyElement)
postDataElement = CefPostDataElement_Create()
postDataElement.get().SetToBytes(len(pyElement), <char*>pyElement)
postData.get().AddElement(postDataElement)
else:
raise Exception("Invalid type of postData, only dict|list allowed")
cpdef dict GetHeaderMap(self):
cdef list headerMultimap = self.GetHeaderMultimap()
cdef dict headerMap = {}
cdef tuple headerTuple
for headerTuple in headerMultimap:
key = headerTuple[0]
value = headerTuple[1]
headerMap[key] = value
return headerMap
cpdef list GetHeaderMultimap(self):
cdef cpp_multimap[CefString, CefString] cefHeaderMap
self.GetCefRequest().get().GetHeaderMap(cefHeaderMap)
cdef list pyHeaderMultimap = []
cdef cpp_multimap[CefString, CefString].iterator iterator = (
cefHeaderMap.begin())
cdef CefString cefKey
cdef CefString cefValue
cdef str pyKey
cdef str pyValue
while iterator != cefHeaderMap.end():
cefKey = deref(iterator).first
cefValue = deref(iterator).second
pyKey = CefToPyString(cefKey)
pyValue = CefToPyString(cefValue)
pyHeaderMultimap.append((pyKey, pyValue))
preinc(iterator)
return pyHeaderMultimap
cpdef py_void SetHeaderMap(self, dict headerMap):
assert len(headerMap) > 0, "headerMap param is empty"
cpdef list headerMultimap = []
cdef object key
for key in headerMap:
headerMultimap.append((str(key), str(headerMap[key])))
self.SetHeaderMultimap(headerMultimap)
cpdef py_void SetHeaderMultimap(self, list headerMultimap):
assert len(headerMultimap) > 0, "headerMultimap param is empty"
cdef cpp_multimap[CefString, CefString] cefHeaderMap
cdef CefString cefKey
cdef CefString cefValue
cdef cpp_pair[CefString, CefString] pair
cdef tuple headerTuple
for headerTuple in headerMultimap:
PyToCefString(str(headerTuple[0]), cefKey)
PyToCefString(str(headerTuple[1]), cefValue)
pair.first, pair.second = cefKey, cefValue
cefHeaderMap.insert(pair)
self.GetCefRequest().get().SetHeaderMap(cefHeaderMap)
cpdef int GetFlags(self) except *:
return self.GetCefRequest().get().GetFlags()
cpdef py_void SetFlags(self, int flags):
self.GetCefRequest().get().SetFlags(
<cef_types.cef_weburlrequest_flags_t?>flags)
cpdef str GetFirstPartyForCookies(self):
return CefToPyString(
self.GetCefRequest().get().GetFirstPartyForCookies())
cpdef py_void SetFirstPartyForCookies(self, py_string url):
self.GetCefRequest().get().SetFirstPartyForCookies(
PyToCefStringValue(url))