GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

separate get_imported_module

Browse Source
This commit is contained in:
debugtalk
2017-08-31 10:48:15 +08:00
parent 74c97a9ccb
commit ad58e2915f
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ate/context.py
View File

@@ -1,5 +1,4 @@
import copy
import importlib
import os
import re
import sys
@@ -46,7 +45,7 @@ class Context(object):
""" import required modules dynamicly
"""
for module_name in modules:
globals()[module_name] = importlib.import_module(module_name)
globals()[module_name] = utils.get_imported_module(module_name)
def bind_functions(self, function_binds, level="testcase"):
""" Bind named functions within the context
@@ -70,7 +69,8 @@ class Context(object):
"""
sys.path.insert(0, os.getcwd())
for module_name in modules:
imported_functions_dict = utils.get_module_functions(module_name)
imported_module = utils.get_imported_module(module_name)
imported_functions_dict = utils.filter_module_functions(imported_module)
self.__update_context_functions_config(level, imported_functions_dict)
def bind_variables(self, variable_binds, level="testcase"):

12
ate/utils.py
View File

@@ -255,9 +255,13 @@ def is_function(tup):
name, item = tup
return isinstance(item, types.FunctionType)
def get_module_functions(module_name):
""" import module and return filtered functions
def get_imported_module(module_name):
""" import module and return imported module
"""
imported = importlib.import_module(module_name)
module_functions_dict = dict(filter(is_function, vars(imported).items()))
return importlib.import_module(module_name)
def filter_module_functions(module):
""" filter functions from import module
"""
module_functions_dict = dict(filter(is_function, vars(module).items()))
return module_functions_dict