Allow selecting custom OAuthLibCore implementation

Author: eofs

oauth2_provider/views/mixins.py

@@ -46,6 +46,16 @@ def get_validator_class(cls):
         else:
             return cls.validator_class
 
+    @classmethod
+    def get_oauthlib_core_class(cls):
+        """
+        Return the OAuthLibCore implementation class to use
+        """
+        if cls.oauthlib_core_class is None:
+            return OAuthLibCore
+        else:
+            return cls.oauthlib_core_class
+
     @classmethod
     def get_server(cls):
         """
@@ -63,7 +73,8 @@ def get_oauthlib_core(cls):
         """
         if not hasattr(cls, '_oauthlib_core'):
             server = cls.get_server()
-            cls._oauthlib_core = OAuthLibCore(server)
+            core_class = cls.get_oauthlib_core_class()
+            cls._oauthlib_core = core_class(server)
         return cls._oauthlib_core
 
     def validate_authorization_request(self, request):