# Copyright (c) 2013 CEF Python, see the Authors file. # All rights reserved. Licensed under BSD 3-clause license. # Project website: https://github.com/cztomczak/cefpython include "cefpython.pyx" import weakref # ----------------------------------------------------------------------------- # PyStringVisitor # ----------------------------------------------------------------------------- cdef object g_userStringVisitors = weakref.WeakValueDictionary() cdef int g_userStringVisitorMaxId = 0 cdef int StoreUserStringVisitor(object userStringVisitor) except *: global g_userStringVisitorMaxId global g_userStringVisitors g_userStringVisitorMaxId += 1 g_userStringVisitors[g_userStringVisitorMaxId] = userStringVisitor return g_userStringVisitorMaxId cdef PyStringVisitor GetUserStringVisitor(int stringVisitorId): global g_userStringVisitors cdef object userStringVisitor cdef PyStringVisitor pyStringVisitor if stringVisitorId in g_userStringVisitors: userStringVisitor = g_userStringVisitors[stringVisitorId] pyStringVisitor = PyStringVisitor(userStringVisitor) return pyStringVisitor cdef class PyStringVisitor: cdef object userStringVisitor def __init__(self, object userStringVisitor): self.userStringVisitor = userStringVisitor cdef object GetCallback(self, str funcName): if self.userStringVisitor and ( hasattr(self.userStringVisitor, funcName) and ( callable(getattr(self.userStringVisitor, funcName)))): return getattr(self.userStringVisitor, funcName) # ----------------------------------------------------------------------------- # C++ StringVisitor # ----------------------------------------------------------------------------- cdef CefRefPtr[CefStringVisitor] CreateStringVisitor( object userStringVisitor) except *: if not userStringVisitor: raise Exception("userStringVisitor object missing") if not hasattr(userStringVisitor, "Visit"): raise Exception("userStringVisitor object is missing Visit() method") cdef int stringVisitorId = StoreUserStringVisitor(userStringVisitor) cdef CefRefPtr[CefStringVisitor] cefStringVisitor = ( new StringVisitor(stringVisitorId)) return cefStringVisitor cdef public void StringVisitor_Visit( int stringVisitorId, const CefString& string ) except * with gil: cdef str pyString cdef PyStringVisitor userStringVisitor cdef object callback try: pyString = CefToPyString(string) userStringVisitor = GetUserStringVisitor(stringVisitorId) if userStringVisitor: callback = userStringVisitor.GetCallback("Visit") if callback: callback(value=pyString) except: (exc_type, exc_value, exc_trace) = sys.exc_info() sys.excepthook(exc_type, exc_value, exc_trace)