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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
# Unix SMB/CIFS implementation.
#
# claim value type management
#
# Copyright (C) Catalyst.Net Ltd. 2023
#
# Written by Rob van der Linde <rob@catalyst.net.nz>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import samba.getopt as options
from samba.netcmd import Command, CommandError, Option, SuperCommand
from samba.netcmd.domain.models import ValueType
from samba.netcmd.domain.models.exceptions import ModelError
class cmd_domain_claim_value_type_list(Command):
"""List claim values types on the domain."""
synopsis = "%prog -H <URL> [options]"
takes_optiongroups = {
"sambaopts": options.SambaOptions,
"credopts": options.CredentialsOptions,
"hostopts": options.HostOptions,
}
takes_options = [
Option("--json", help="Output results in JSON format.",
dest="output_format", action="store_const", const="json"),
]
def run(self, hostopts=None, sambaopts=None, credopts=None,
output_format=None):
ldb = self.ldb_connect(hostopts, sambaopts, credopts)
# Value types grouped by display name.
try:
value_types = {value_type.display_name: value_type.as_dict()
for value_type in ValueType.query(ldb)}
except ModelError as e:
raise CommandError(e)
# Using json output format gives more detail.
if output_format == "json":
self.print_json(value_types)
else:
for value_type in value_types.keys():
self.outf.write(f"{value_type}\n")
class cmd_domain_claim_value_type_view(Command):
"""View a single claim value type on the domain."""
synopsis = "%prog -H <URL> [options]"
takes_optiongroups = {
"sambaopts": options.SambaOptions,
"credopts": options.CredentialsOptions,
"hostopts": options.HostOptions,
}
takes_options = [
Option("--name",
help="Display name of claim value type to view (required).",
dest="name", action="store", type=str, required=True),
]
def run(self, hostopts=None, sambaopts=None, credopts=None, name=None):
ldb = self.ldb_connect(hostopts, sambaopts, credopts)
try:
value_type = ValueType.get(ldb, display_name=name)
except ModelError as e:
raise CommandError(e)
# Check if value type exists first.
if value_type is None:
raise CommandError(f"Value type {name} not found.")
# Display vale type as JSON.
self.print_json(value_type.as_dict())
class cmd_domain_claim_value_type(SuperCommand):
"""Manage claim value types on the domain."""
subcommands = {
"list": cmd_domain_claim_value_type_list(),
"view": cmd_domain_claim_value_type_view(),
}
|