1 files changed, 99 insertions, 0 deletions
diff --git a/ansible_collections/kaytus/ksmanage/plugins/modules/power_budget_info.py b/ansible_collections/kaytus/ksmanage/plugins/modules/power_budget_info.py
new file mode 100644
index 000000000..d2631de2f
--- /dev/null
+++ b/ansible_collections/kaytus/ksmanage/plugins/modules/power_budget_info.py
@@ -0,0 +1,99 @@
+#!/usr/bin/python
+# -*- coding:utf-8 -*-
+
+# Copyright(C) 2023 Kaytus Inc. All Rights Reserved.
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+
+from __future__ import (absolute_import, division, print_function)
+
+__metaclass__ = type
+
+DOCUMENTATION = '''
+---
+module: power_budget_info
+version_added: "1.0.0"
+author:
+    - WangBaoshan (@ieisystem)
+short_description: Get power budget information
+description:
+   - Get power budget information on kaytus Server.
+notes:
+   - Supports C(check_mode).
+options: {}
+extends_documentation_fragment:
+    - kaytus.ksmanage.ksmanage
+'''
+
+EXAMPLES = '''
+- name: Power budget test
+  hosts: ksmanage
+  connection: local
+  gather_facts: false
+  vars:
+    ksmanage:
+      host: "{{ ansible_ssh_host }}"
+      username: "{{ username }}"
+      password: "{{ password }}"
+
+  tasks:
+
+  - name: "Get power budget information"
+    kaytus.ksmanage.power_budget_info:
+      provider: "{{ ksmanage }}"
+'''
+
+RETURN = '''
+message:
+    description: Messages returned after module execution.
+    returned: always
+    type: str
+state:
+    description: Status after module execution.
+    returned: always
+    type: str
+changed:
+    description: Check to see if a change was made on the device.
+    returned: always
+    type: bool
+'''
+
+from ansible.module_utils.basic import AnsibleModule
+from ansible_collections.kaytus.ksmanage.plugins.module_utils.ksmanage import (ksmanage_argument_spec, get_connection)
+
+
+class Power(object):
+    def __init__(self, argument_spec):
+        self.spec = argument_spec
+        self.module = None
+        self.init_module()
+        self.results = dict()
+
+    def init_module(self):
+        """Init module object"""
+
+        self.module = AnsibleModule(
+            argument_spec=self.spec, supports_check_mode=True)
+
+    def run_command(self):
+        self.module.params['subcommand'] = 'getpowerbudget'
+        self.results = get_connection(self.module)
+
+    def show_result(self):
+        """Show result"""
+        self.module.exit_json(**self.results)
+
+    def work(self):
+        """Worker"""
+        self.run_command()
+        self.show_result()
+
+
+def main():
+    argument_spec = dict()
+    argument_spec.update(ksmanage_argument_spec)
+    power_obj = Power(argument_spec)
+    power_obj.work()
+
+
+if __name__ == '__main__':
+    main()