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
|
import logging
from Bcfg2.Bcfg2Py3k import StringIO
from Bcfg2.Server.Plugins.Cfg import SETUP
from Bcfg2.Server.Plugins.Cfg.CfgGenshiGenerator import CfgGenshiGenerator
from Bcfg2.Server.Plugins.Cfg.CfgEncryptedGenerator import decrypt, \
CfgEncryptedGenerator
logger = logging.getLogger(__name__)
try:
from genshi.template import TemplateLoader, loader
except ImportError:
# CfgGenshiGenerator will raise errors if genshi doesn't exist
pass
def crypted_loader(filename):
loadfunc = loader.directory(os.path.dirname(filename))
filepath, filename, fileobj, uptodate = loadfunc(filename)
return (filepath, filename, StringIO(decrypt(fileobj.read())), uptodate)
class CfgEncryptedGenshiGenerator(CfgGenshiGenerator, CfgEncryptedGenerator):
__extensions__ = ['genshi.crypt', 'crypt.genshi']
def __init__(self, fname, spec, encoding):
CfgEncryptedGenerator.__init__(self, fname, spec, encoding)
CfgGenshiGenerator.__init__(self, fname, spec, encoding)
self.loader = TemplateLoader([crypted_loader])
def handle_event(self, event):
CfgEncryptedGenerator.handle_event(self, event)
CfgGenshiGenerator.handle_event(self, event)
def get_data(self, entry, metadata):
CfgGenshiGenerator.get_data(self, entry, metadata)
|