Python 导入与注册

背景

---

---

---

最近一直学习写一个POC扫描框架，但是不知道如何下手，正巧因为一些需要有朋友在研究POCSuite的实现原理，顺面蹭一些知识点，补一补Python基础的不足,为以后编写POC框架打地基。

导入

---

---

---

静态模块导入

---

---

import os#直接导入python自带模块或第三方安装后的模块
import mylib#直接导入自编写模块
import core.lib.api#在python默认查找路径中添加包中的某模块（本地、官方和安装的第三方库的路径）
import ..config.conf#相对路径中导入包中的模块

import sys
sys.path.append(path)
import xxx#添加路劲到python查找路径后导入模块
from xxx import xxx#从某包导入某模块或者从某模块导入莫对象
from xx.xx.xx import xxx#从某包的某包（模块）导入某模块或者某对象

以上都是常见的静态导入方法，下面来解释下导入原理，这里按照python2来介绍:

globals()和locals()

---

globals()返回一个包含全局对象的一个字典，包含创建的各种全局变量--自定义类及其实例、数据类型及其实例等等，locals()是局部的的



导入后

sys.modules

---

引用一段话来说明(加入globals(),并做检查了，而不是locals())：

Python中所有加载到内存的模块都放在sys.modules。当import一个模块时首先会在这个列表中查找是否已经加载了此模块，如果加载了则只是将模块的名字加入到正在调用import的模块的Local名字空间中。如果没有加载则从sys.path目录中按照模块名称查找模块文件，模块文件可以是py、pyc、pyd，找到后将模块载入内存，并加入到sys.modules中，并将名称导入到当前的Local名字空间。

动态导入

---

---

导入的模块在运行到一定程度才能确定

__import__(modName,globals(),locals(),[],-1)
"""
[]是formlist后面要加入某些模块的时候，在中间添加内容
-1是相对+绝对引入
0是绝对引入
1是相对引入
"""
from a import b
#equals
__import__(a,gloabls(),locals(),[b],-1)

注册

---

---

---

注册模块

---

---

def moduleRegister(regModule):
    filename = os.path.basename(regModule)
    filepath = os.path.dirname(regModule)
    log.debug("开始检查待注册模块%s"%regModule)
    if regModule.__name__ in globals():
        log.info("%s模块名已经注册"%regModule)
        raise Exception("模块名已经注册")
    try:
        log.debug("开始注册模块%s"%regModule)
        __import__(filename,globals(),locals(),[],-1)
        log.debug("成功注册模块%s"%regModule)
    except Exception,reason:
        raise Exception("模块注册失败")

注册类、实例对象

---

---

#POCSUITE
def registerPoc(pocClass):
    module = pocClass.__module__.split('.')[-1]
    if module in kb.registeredPocs:
        return

    kb.registeredPocs[module] = pocClass()

把类对或者实例对象加入全局变量的字典里

#引入依赖的外部对象、模块、包
import os
import sys

#定义全局类和函数
def registerCls(regMod,clsName):
    filename = os.path.basename(regMod)
    filepath = os.path.dirname(regMod)
    try:
        __import__(filename,globals(),locals(),[clsName],-1)
    except Exception,reason:
        raise Exception("模块注册失败")

def registerMod(regModule):
    filename = os.path.basename(regModule)
    filepath = os.path.dirname(regModule)
    try:
        __import__(filename,globals(),locals(),[],-1)
    except Exception,reason:
        try:
            sys.path.append(filepath)
        except Exception,reason:
            raise Exception("模块注册失败")

def getVarByMod(regMod,varName):
    try:
        return getattr(regMod,varName)
    except Exception,reason:
        raise Exception("从模块获取变量对象时失败")

def getFuncByMod(regMod,funcName):
    try:
        return getattr(regMod,funcName)
    except Exception,reason:
        raise Exception("从模块获取方法对象时失败")

def getClsByMod(regMod,clsName):
    try:
        return getattr(regMod,clsName)
    except Exception,reason:
        raise Exception("从模块获取类对象时失败")

def getClnByCls(clsObj,clnName):
    try:
        return getattr(clsObj,clnName)
    except Exception,reason:
        raise Exception("从类中获取属性失败")

def getFuncByCls(clsObj,funcName):
    try:
        return getattr(clsObj,clnName)
    except Exception,reason:
        raise Exception("从类中获取属性失败")

def createClsIst(clsObj,*args,**kwargs):
    try:
        return clsObj(*args,**kwargs)
    except Exception,reason:
        raise Exception("创建实例失败")

def executeFunc(funcObj,*args,**kwargs):
    try:
        return funcObj(*args,**kwargs)
    except Exception,reason:
        raise Exception("执行函数失败")

def getClsByIst(regObj):
    try:
        return regObj.__class__
    except Exception,reason:
        raise Exception("从实例获取类失败")