You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.7 KiB
58 lines
1.7 KiB
# Copyright © 2011-2024 Splunk, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import xml.etree.ElementTree as ET
|
|
from .utils import parse_xml_data
|
|
|
|
|
|
class InputDefinition:
|
|
"""``InputDefinition`` encodes the XML defining inputs that Splunk passes to
|
|
a modular input script.
|
|
|
|
**Example**::
|
|
|
|
i = InputDefinition()
|
|
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.metadata = {}
|
|
self.inputs = {}
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, InputDefinition):
|
|
return False
|
|
return self.metadata == other.metadata and self.inputs == other.inputs
|
|
|
|
@staticmethod
|
|
def parse(stream):
|
|
"""Parse a stream containing XML into an ``InputDefinition``.
|
|
|
|
:param stream: stream containing XML to parse.
|
|
:return: definition: an ``InputDefinition`` object.
|
|
"""
|
|
definition = InputDefinition()
|
|
|
|
# parse XML from the stream, then get the root node
|
|
root = ET.parse(stream).getroot()
|
|
|
|
for node in root:
|
|
if node.tag == "configuration":
|
|
# get config for each stanza
|
|
definition.inputs = parse_xml_data(node, "stanza")
|
|
else:
|
|
definition.metadata[node.tag] = node.text
|
|
|
|
return definition
|