cefpython/src/cookie.pyx at cefpython57 · 2php/cefpython · GitHub

Repository navigation

History

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

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

314

315

316

317

318

319

320

321

322

323

324

325

326

327

328

329

330

331

332

333

334

335

336

# Project website: https://github.com/cztomczak/cefpython

include "cefpython.pyx"

from task cimport *

# ------------------------------------------------------------------------------

# Tests

# ------------------------------------------------------------------------------

#cdef Cookie cookie = Cookie()

#cookie.SetName("asd1")

#print("cookie.cefCookie: %s" % cookie.cefCookie)

#print("cookie.GetName(): %s" % cookie.GetName())

#print("cookie.GetCreation(): %s" % cookie.GetCreation())

#cookie.SetCreation(datetime.datetime(2013,5,23))

#print("cookie.GetCreation(): %s" % cookie.GetCreation())

#print("cookie: %s" % cookie.Get())

# ------------------------------------------------------------------------------

# Globals

# ------------------------------------------------------------------------------

# noinspection PyUnresolvedReferences

cdef PyCookieManager g_globalCookieManager = None

# See StoreUserCookieVisitor().

import weakref

cdef object g_userCookieVisitors = weakref.WeakValueDictionary()

cdef int g_userCookieVisitorMaxId = 0

# ------------------------------------------------------------------------------

# Cookie

# ------------------------------------------------------------------------------

# noinspection PyUnresolvedReferences

ctypedef Cookie PyCookie

cdef PyCookie CreatePyCookie(CefCookie cefCookie):

cdef PyCookie pyCookie = Cookie()

pyCookie.cefCookie = cefCookie

return pyCookie

cdef class Cookie:

cdef CefCookie cefCookie

cpdef py_void Set(self, dict cookie):

for key in cookie:

if key == "name":

self.SetName(cookie[key])

elif key == "value":

self.SetValue(cookie[key])

elif key == "domain":

self.SetDomain(cookie[key])

elif key == "path":

self.SetPath(cookie[key])

elif key == "secure":

self.SetSecure(cookie[key])

elif key == "httpOnly":

self.SetHttpOnly(cookie[key])

elif key == "creation":

self.SetCreation(cookie[key])

elif key == "lastAccess":

self.SetLastAccess(cookie[key])

elif key == "hasExpires":

self.SetHasExpires(cookie[key])

elif key == "expires":

self.SetExpires(cookie[key])

else:

raise Exception("Invalid key: %s" % key)

cpdef dict Get(self):

return {

"name": self.GetName(),

"value": self.GetValue(),

"domain": self.GetDomain(),

"path": self.GetPath(),

"secure": self.GetSecure(),

"httpOnly": self.GetHttpOnly(),

"creation": self.GetCreation(),

"lastAccess": self.GetLastAccess(),

"hasExpires": self.GetHasExpires(),

"expires": self.GetExpires(),

}

cpdef py_void SetName(self, py_string name):

# This works:

# | CefString(&self.cefCookie.name).FromString(name)

# This does not work:

# | cdef CefString cefString = CefString(&self.cefCookie.name)

# | PyToCefString(name, cefString)

# Because it's a Copy Constructor, it does not reference the

# same underlying cef_string_t, instead it copies the value.

# "T a(b)" - direct initialization (not supported by cython)

# "T a = b" - copy initialization

# But this works:

# | cdef CefString* cefString = new CefString(&self.cefCookie.name)

# | PyToCefStringPointer(name, cefString)

# | del cefString

# Solution: use Attach() method to pass reference to cef_string_t.

cdef CefString cefString

cefString.Attach(&self.cefCookie.name, False)

PyToCefString(name, cefString)

cpdef str GetName(self):

cdef CefString cefString

cefString.Attach(&self.cefCookie.name, False)

return CefToPyString(cefString)

cpdef py_void SetValue(self, py_string value):

cdef CefString cefString

cefString.Attach(&self.cefCookie.value, False)

PyToCefString(value, cefString)

cpdef str GetValue(self):

cdef CefString cefString

cefString.Attach(&self.cefCookie.value, False)

return CefToPyString(cefString)

cpdef py_void SetDomain(self, py_string domain):

cdef CefString cefString

cefString.Attach(&self.cefCookie.domain, False)

PyToCefString(domain, cefString)

cpdef str GetDomain(self):

cdef CefString cefString

cefString.Attach(&self.cefCookie.domain, False)

return CefToPyString(cefString)

cpdef py_void SetPath(self, py_string path):

cdef CefString cefString

cefString.Attach(&self.cefCookie.path, False)

PyToCefString(path, cefString)

cpdef str GetPath(self):

cdef CefString cefString

cefString.Attach(&self.cefCookie.path, False)

return CefToPyString(cefString)

cpdef py_void SetSecure(self, py_bool secure):

# Need to wrap it with bool() to get rid of the C++ compiler

# warnings: "cefpython.cpp(24740) : warning C4800: 'int' :

# forcing value to bool 'true' or 'false' (performance warning)".

self.cefCookie.secure = bool(secure)

cpdef py_bool GetSecure(self):

return self.cefCookie.secure

cpdef py_void SetHttpOnly(self, py_bool httpOnly):

self.cefCookie.httponly = bool(httpOnly)

cpdef py_bool GetHttpOnly(self):

return self.cefCookie.httponly

cpdef py_void SetCreation(self, object creation):

DatetimeToCefTimeT(creation, self.cefCookie.creation)

cpdef object GetCreation(self):

return CefTimeTToDatetime(self.cefCookie.creation)

cpdef py_void SetLastAccess(self, object lastAccess):

DatetimeToCefTimeT(lastAccess, self.cefCookie.last_access)

cpdef object GetLastAccess(self):

return CefTimeTToDatetime(self.cefCookie.last_access)

cpdef py_void SetHasExpires(self, py_bool hasExpires):

self.cefCookie.has_expires = bool(hasExpires)

cpdef py_bool GetHasExpires(self):

return self.cefCookie.has_expires

cpdef py_void SetExpires(self, object expires):

DatetimeToCefTimeT(expires, self.cefCookie.expires)

cpdef object GetExpires(self):

return CefTimeTToDatetime(self.cefCookie.expires)

# ------------------------------------------------------------------------------

# CookieManager

# ------------------------------------------------------------------------------

class CookieManager:

@staticmethod

def GetGlobalManager():

global g_globalCookieManager

cdef CefRefPtr[CefCookieManager] cefCookieManager

if not g_globalCookieManager:

cefCookieManager = CefCookieManager_GetGlobalManager(

<CefRefPtr[CefCompletionCallback]?>NULL)

g_globalCookieManager = CreatePyCookieManager(cefCookieManager)

return g_globalCookieManager

@staticmethod

def CreateManager(py_string path, py_bool persistSessionCookies=False):

cdef CefRefPtr[CefCookieManager] cefCookieManager

cefCookieManager = CefCookieManager_CreateManager(

PyToCefStringValue(path), bool(persistSessionCookies),

<CefRefPtr[CefCompletionCallback]?>NULL)

if <void*>cefCookieManager != NULL and cefCookieManager.get():

return CreatePyCookieManager(cefCookieManager)

return None

# ------------------------------------------------------------------------------

# PyCookieManager

# ------------------------------------------------------------------------------

cdef PyCookieManager CreatePyCookieManager(

CefRefPtr[CefCookieManager] cefCookieManager):

cdef PyCookieManager pyCookieManager = PyCookieManager()

pyCookieManager.cefCookieManager = cefCookieManager

return pyCookieManager

cdef class PyCookieManager:

cdef CefRefPtr[CefCookieManager] cefCookieManager

cpdef py_void SetSupportedSchemes(self, list schemes):

cdef cpp_vector[CefString] schemesVector

for scheme in schemes:

schemesVector.push_back(PyToCefStringValue(scheme))

self.cefCookieManager.get().SetSupportedSchemes(schemesVector,

<CefRefPtr[CefCompletionCallback]?>NULL)

cdef py_void ValidateUserCookieVisitor(self, object userCookieVisitor):

if userCookieVisitor and hasattr(userCookieVisitor, "Visit") and (

callable(getattr(userCookieVisitor, "Visit"))):

# OK.

return

raise Exception("CookieVisitor object is missing Visit() method")

cpdef py_bool VisitAllCookies(self, object userCookieVisitor):

self.ValidateUserCookieVisitor(userCookieVisitor)

cdef int cookieVisitorId = StoreUserCookieVisitor(userCookieVisitor)

cdef CefRefPtr[CefCookieVisitor] cefCookieVisitor = (

<CefRefPtr[CefCookieVisitor]?>new CookieVisitor(

cookieVisitorId))

return self.cefCookieManager.get().VisitAllCookies(

cefCookieVisitor)

cpdef py_bool VisitUrlCookies(self, py_string url,

py_bool includeHttpOnly, object userCookieVisitor):

self.ValidateUserCookieVisitor(userCookieVisitor)

cdef int cookieVisitorId = StoreUserCookieVisitor(userCookieVisitor)

cdef CefRefPtr[CefCookieVisitor] cefCookieVisitor = (

<CefRefPtr[CefCookieVisitor]?>new CookieVisitor(

cookieVisitorId))

return self.cefCookieManager.get().VisitUrlCookies(

PyToCefStringValue(url), bool(includeHttpOnly),

cefCookieVisitor)

cpdef py_void SetCookie(self, py_string url, PyCookie cookie):

assert isinstance(cookie, Cookie), "cookie object is invalid"

CefPostTask(TID_IO, CreateTask_SetCookie(

self.cefCookieManager.get(),

PyToCefStringValue(url), cookie.cefCookie,

<CefRefPtr[CefSetCookieCallback]?>NULL))

cpdef py_void DeleteCookies(self, py_string url, py_string cookie_name):

CefPostTask(TID_IO, CreateTask_DeleteCookies(

self.cefCookieManager.get(),

PyToCefStringValue(url), PyToCefStringValue(cookie_name),

<CefRefPtr[CefDeleteCookiesCallback]?>NULL))

cpdef py_bool SetStoragePath(self, py_string path,

py_bool persistSessionCookies=False):

return self.cefCookieManager.get().SetStoragePath(

PyToCefStringValue(path), bool(persistSessionCookies),

<CefRefPtr[CefCompletionCallback]?>NULL)

# ------------------------------------------------------------------------------

# PyCookieVisitor

# ------------------------------------------------------------------------------

cdef int StoreUserCookieVisitor(object userCookieVisitor) except *:

global g_userCookieVisitorMaxId

global g_userCookieVisitors

g_userCookieVisitorMaxId += 1

g_userCookieVisitors[g_userCookieVisitorMaxId] = userCookieVisitor

return g_userCookieVisitorMaxId

cdef PyCookieVisitor GetPyCookieVisitor(int cookieVisitorId):

global g_userCookieVisitors

cdef object userCookieVisitor

cdef PyCookieVisitor pyCookieVisitor

if cookieVisitorId in g_userCookieVisitors:

userCookieVisitor = g_userCookieVisitors[cookieVisitorId]

pyCookieVisitor = PyCookieVisitor(userCookieVisitor)

return pyCookieVisitor

cdef class PyCookieVisitor:

cdef object userCookieVisitor

def __init__(self, object userCookieVisitor):

self.userCookieVisitor = userCookieVisitor

cdef object GetCallback(self, str funcName):

if self.userCookieVisitor and (

hasattr(self.userCookieVisitor, funcName) and (

callable(getattr(self.userCookieVisitor, funcName)))):

return getattr(self.userCookieVisitor, funcName)

# ------------------------------------------------------------------------------

# C++ CookieVisitor

# ------------------------------------------------------------------------------

cdef public cpp_bool CookieVisitor_Visit(

int cookieVisitorId,

const CefCookie& cookie,

int count,

int total,

cpp_bool& deleteCookie

) except * with gil:

cdef PyCookieVisitor pyCookieVisitor

cdef object callback

cdef py_bool ret

cdef PyCookie pyCookie

cdef list pyDeleteCookie = [False]

try:

assert IsThread(TID_IO), "Must be called on the IO thread"

pyCookieVisitor = GetPyCookieVisitor(cookieVisitorId)

pyCookie = CreatePyCookie(cookie)

if pyCookieVisitor:

callback = pyCookieVisitor.GetCallback("Visit")

if callback:

ret = callback(

cookie=pyCookie,

count=count,

total=total,

delete_cookie_out=pyDeleteCookie)

(&deleteCookie)[0] = bool(pyDeleteCookie[0])

return bool(ret)

return False

except:

(exc_type, exc_value, exc_trace) = sys.exc_info()

sys.excepthook(exc_type, exc_value, exc_trace)

cefpython/src/cookie.pyx at cefpython57 · 2php/cefpython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

cookie.pyx

Latest commit

History

Breadcrumbs

cookie.pyx

File metadata and controls