summaryrefslogtreecommitdiffstats
path: root/ansible_collections/cloud/common/plugins/lookup/turbo_demo.py
blob: 4d69b0d014895dbfe5b013131c43f78d1301fd19 (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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Copyright: (c) 2021, Aubin Bikouo (@abikouo)
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)

DOCUMENTATION = r"""
name: turbo_demo
author:
  - Aubin Bikouo (@abikouo)

short_description: A demo for lookup plugins on cloud.common
description:
  - return the parent process of the running process
options:
  playbook_vars:
    description: list of playbook variables to add in the output.
    type: list
    elements: str
"""

EXAMPLES = r"""
"""

RETURN = r"""
"""


import os
import sys
import traceback

from ansible_collections.cloud.common.plugins.plugin_utils.turbo.lookup import (
    TurboLookupBase as LookupBase,
)


def counter():
    counter.i += 1
    return counter.i


# NOTE: workaround to avoid a warning with ansible-doc
if True:  # pylint: disable=using-constant-test
    counter.i = 0


async def execute(terms, variables, playbook_vars):
    result = []
    result.append(f"running from pid: {os.getpid()}")
    if playbook_vars is not None:
        result += [
            variables["vars"].get(x) for x in playbook_vars if x in variables["vars"]
        ]
    if terms:
        result += terms

    for id, stack in list(sys._current_frames().items()):
        for fname, line_id, name, line in traceback.extract_stack(stack):
            if fname == __file__:
                continue

    result.append(f"turbo_demo_counter: {counter()}")
    return result


class LookupModule(LookupBase):
    async def _run(self, terms, variables=None, playbook_vars=None):
        result = await execute(terms, variables, playbook_vars)
        return result

    run = _run if not hasattr(LookupBase, "run_on_daemon") else LookupBase.run_on_daemon