Creating Import Hooks

[Jonathan Gardner]

On Feb 17, 10:48 pm, Sreejith K <sreejith... at gmail.com> wrote:
> Hi everyone,
>
> I need to implement custom import hooks for an application (http://www.python.org/dev/peps/pep-0302/). I want to restrict an application
> to import certain modules (say socket module). Google app engine is
> using a module hook to do this (HardenedModulesHook in google/
> appengine/tools/dev_appserver.py). But I want to allow that
> application to use an sdk module (custom) which imports and uses
> socket module. But the module hook restricts the access by sdk.
> Finding out, which file is importing a module give a solution?? ie. If
> the application is importing socket module, I want to restrict it. But
> if the sdk module is importing socket I want to allow it. Is there any
> way I can do this ?
>
> Application
> ========
> import sdk
> import socket               # I dont want to allow this (need to raise
> ImportError)
>
> SDK
> ====
> import socket               # need to allow this


SDK
===
import socket

App
===
import SDK
import sys
socket = sys.modules['socket']