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("从实例获取类失败")