import ihooks

import os

import __builtin__

import imp

import sys



from imp import C_EXTENSION, PY_SOURCE, PY_COMPILED

from imp import C_BUILTIN, PY_FROZEN, PKG_DIRECTORY

BUILTIN_MODULE = C_BUILTIN

FROZEN_MODULE = PY_FROZEN



rootpkg_dir = "ROOT"



class PackageFS:

    def __init__(self):

        self.prefix_path = r"f:	estscript"



    def printMsg(self, msg):

        print "PackageFS -- ", msg

        

    def CombinePath(self, filenameinPkg):

         i = filenameinPkg.find(rootpkg_dir)

         pos = i + 1 + len(rootpkg_dir)

         filename = filenameinPkg[pos:]

         realpath = self.prefix_path + "\" + filename

         print  "filename = ", filename

         print  "realpath = ", realpath

         return realpath

    def isdir(self,path):

        realpath = self.CombinePath(path)

        self.printMsg("isdir")

        print "path = ", path

        print "realpath = ", realpath

        return os.path.isdir(realpath)



    def path_isfile(self, path):

        self.printMsg("path_isfile")        

        realpath = self.CombinePath(path)

        return os.path.isfile(realpath)



    def openfile(self, x):

        self.printMsg("openfile")                

        parms = list(x)

        path = parms[0]

        realpath = self.CombinePath(path)

        parms[0] = realpath

        print " realpath = ", realpath

        print "*(tuple(parms))", tuple(parms)

        

        

        return open(*(tuple(parms)))

    def listdir(self, path):

        self.printMsg("listdir")

        realpath = self.CombinePath(path)

        return os.listdir(realpath)

        

    

class XyyLoader(ihooks.ModuleLoader):

    pass

class XyyHook(ihooks.Hooks):

    def __init__(self):

        self.pck_fs = PackageFS()

    def printMsg(self, msg):

        print "XyyHook -- ", msg

        

    def IsInPackage(self, filename):

        if (rootpkg_dir in filename):

            return True

        else:

            return False

   

    def path_exists(self, x):

        self.printMsg("path_exists")        

        if (self.IsInPackage(x)):

           return self.pck_fs.exists(x)

        else:

           return os.path.exists(x)

    def path_isdir(self, x):

        self.printMsg("path_isdir")

        if (self.IsInPackage(x)):

            re = self.pck_fs.isdir(x)

            print "x is path? ", re

            return self.pck_fs.isdir(x)

        else:

            return os.path.isdir(x)

    def path_isfile(self, x):

        self.printMsg("path_isfile")

        if (self.IsInPackage(x)):

            return self.pck_fs.isfile(x)

        else:

            return os.path.isfile(x)

    def openfile(self, *x):

        self.printMsg("openfile")

        print "*"*20, "x = ", x

        print "*"*20, "x[0] = ", x[0]

        path,right = x

        if (self.IsInPackage(path)):

            print "call package openfile path = ", path

            return self.pck_fs.openfile((path, right))

        else:

            print "call fs openfile path = "  , path

            return open(*x)

    def listdir(self, x):

        self.printMsg("listdir")

        if (self.IsInPackage(x)):

            return self.pck_fs.listdir(x)

        else:

            return os.listdir(x)



class XyyImporter(ihooks.ModuleImporter):

    pass







xyyhook = XyyHook()

xyyLoader = XyyLoader(xyyhook)

importer = ihooks.ModuleImporter(xyyLoader)

ihooks.install(importer)