1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# -*- coding: utf8 -*-
"""
kaptan.handlers.pyfile_handler
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyright: (c) 2013 by the authors and contributors (See AUTHORS file).
:license: BSD, see LICENSE for more details.
"""
from __future__ import print_function, unicode_literals
import os
import sys
from . import BaseHandler
def import_pyfile(pathname, mod_name=''):
"""Import the contents of filepath as a Python module.
Parameters
----------
pathname: str
Path to the .py file to be imported as a module
mod_name: str
Name of the module when imported
Returns
-------
mod
The imported module
Raises
------
IOError
If file is not found
"""
if not os.path.isfile(pathname):
raise IOError('File {0} not found.'.format(pathname))
if sys.version_info[0] == 3 and sys.version_info[1] > 2: # Python >= 3.3
import importlib.machinery
loader = importlib.machinery.SourceFileLoader('', pathname)
mod = loader.load_module(mod_name)
else: # 2.6 >= Python <= 3.2
import imp
mod = imp.load_source(mod_name, pathname)
return mod
class PyFileHandler(BaseHandler):
def load(self, file_):
module = import_pyfile(file_)
data = dict()
for key in dir(module):
value = getattr(module, key)
if not key.startswith("__"):
data.update({key: value})
return data
def dump(self, file_):
raise NotImplementedError("Exporting python files is not supported.")
|
