summaryrefslogtreecommitdiffstats
path: root/ipaclient/plugins/certprofile.py
blob: f36f27128860c3cba51e209986df86eeed7fb0e0 (plain)
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
#
# Copyright (C) 2015  FreeIPA Contributors see COPYING for license
#

from ipaclient.frontend import MethodOverride
from ipalib import util
from ipalib.parameters import File
from ipalib.plugable import Registry
from ipalib.text import _

register = Registry()


@register(override=True)
class certprofile_show(MethodOverride):
    def forward(self, *keys, **options):
        if 'out' in options:
            util.check_writable_file(options['out'])

        result = super(certprofile_show, self).forward(*keys, **options)
        if 'out' in options and 'config' in result['result']:
            with open(options['out'], 'wb') as f:
                f.write(result['result'].pop('config'))
            result['summary'] = (
                _("Profile configuration stored in file '%(file)s'")
                % dict(file=options['out'])
            )

        return result


@register(override=True)
class certprofile_import(MethodOverride):
    def get_options(self):
        for option in super(certprofile_import, self).get_options():
            if option.name == 'file':
                option = option.clone_retype(option.name, File)
            yield option


@register(override=True)
class certprofile_mod(MethodOverride):
    def get_options(self):
        for option in super(certprofile_mod, self).get_options():
            if option.name == 'file':
                option = option.clone_retype(option.name, File)
            yield option