+# SPDX-License-Identifier: GPL-2.0-only
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+
import bt2
import os
class TestIter(bt2._UserMessageIterator):
def __init__(self, config, output_port):
- inputs = output_port.user_data['inputs']
- sc = output_port.user_data['sc']
+ inputs = output_port.user_data["inputs"]
+ sc = output_port.user_data["sc"]
tc = sc.trace_class
t = tc()
s = t.create_stream(sc, name=self._make_stream_name(inputs))
comp_cls_name = self._component.__class__.__name__
return (
comp_cls_name
- + ': '
- + ', '.join(sorted([os.path.basename(str(x)) for x in inputs]))
+ + ": "
+ + ", ".join(sorted([os.path.basename(str(x)) for x in inputs]))
)
def __next__(self):
tc = self._create_trace_class()
sc = tc.create_stream_class()
- self._add_output_port('out', {'inputs': params['inputs'], 'sc': sc})
+ self._add_output_port("out", {"inputs": params["inputs"], "sc": sc})
@bt2.plugin_component_class
@staticmethod
def _user_query(priv_query_exec, obj, params, method_obj):
- if obj == 'babeltrace.support-info':
- if params['type'] == 'file':
- name = os.path.basename(str(params['input']))
-
- if name.startswith('aaa'):
- return {'weight': 1, 'group': 'aaa'}
- elif name.startswith('bbb'):
- return {'weight': 0.5, 'group': 'bbb'}
- elif name.startswith('ccc'):
+ if obj == "babeltrace.support-info":
+ if params["type"] == "file":
+ name = os.path.basename(str(params["input"]))
+
+ if name.startswith("aaa"):
+ return {"weight": 1, "group": "aaa"}
+ elif name.startswith("bbb"):
+ return {"weight": 0.5, "group": "bbb"}
+ elif name.startswith("ccc"):
# Try two different ways of returning 1 (an int and a float).
- if name[3] == '1':
+ if name[3] == "1":
return 1
else:
return 1.0
@staticmethod
def _user_query(priv_query_exec, obj, params, method_obj):
- if obj == 'babeltrace.support-info':
- if params['type'] == 'directory':
- name = os.path.basename(str(params['input']))
- return 1 if name == 'some-dir' else 0
+ if obj == "babeltrace.support-info":
+ if params["type"] == "directory":
+ name = os.path.basename(str(params["input"]))
+ return 1 if name == "some-dir" else 0
else:
return 0
else:
@staticmethod
def _user_query(priv_query_exec, obj, params, method_obj):
- if obj == 'babeltrace.support-info':
+ if obj == "babeltrace.support-info":
return (
1.0
- if params['type'] == 'string' and params['input'] == 'ABCDE'
+ if params["type"] == "string" and params["input"] == "ABCDE"
else 0.0
)
else: