
    Vh	                     z    d dl mZmZmZ eZdZdZdZd dl	m
Z
 d dlmZmZ  G d de      Zd	 Zed
k(  r e        yy)    )absolute_importdivisionprint_functionaP  
---
module: gpu_info
version_added: "1.0.0"
author:
    - WangBaoshan (@ieisystem)
short_description: Get GPU information
description:
   - Get GPU information on kaytus Server.
   - Only the M6 models support this feature.
notes:
   - Supports C(check_mode).
options: {}
extends_documentation_fragment:
    - kaytus.ksmanage.ksmanage
a5  
- name: GPU test
  hosts: ksmanage
  connection: local
  gather_facts: false
  vars:
    ksmanage:
      host: "{{ ansible_ssh_host }}"
      username: "{{ username }}"
      password: "{{ password }}"

  tasks:

  - name: "Get gpu information"
    kaytus.ksmanage.gpu_info:
      provider: "{{ ksmanage }}"
a1  
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
)AnsibleModule)ksmanage_argument_specget_connectionc                   *    e Zd Zd Zd Zd Zd Zd Zy)CPUc                 ^    || _         d | _        | j                          t               | _        y N)specmoduleinit_moduledictresults)selfargument_specs     l/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/kaytus/ksmanage/plugins/modules/gpu_info.py__init__zCPU.__init__B   s%    !	v    c                 <    t        | j                  d      | _        y)zInit module objectT)r   supports_check_modeN)r   r   r   r   s    r   r   zCPU.init_moduleH   s     $))?r   c                 j    d| j                   j                  d<   t        | j                         | _        y )Ngetgpu
subcommand)r   paramsr   r   r   s    r   run_commandzCPU.run_commandN   s&    +3<(%dkk2r   c                 P     | j                   j                  di | j                   y)zShow resultN )r   	exit_jsonr   r   s    r   show_resultzCPU.show_resultR   s    --r   c                 D    | j                          | j                          y)WorkerN)r   r"   r   s    r   workzCPU.workV   s    r   N)__name__
__module____qualname__r   r   r   r"   r%   r    r   r   r
   r
   A   s    ?3.r   r
   c                  x    t               } | j                  t               t        |       }|j	                          y r   )r   updater   r
   r%   )r   cpu_objs     r   mainr,   \   s+    FM/0- GLLNr   __main__N)
__future__r   r   r   type__metaclass__DOCUMENTATIONEXAMPLESRETURNansible.module_utils.basicr   Aansible_collections.kaytus.ksmanage.plugins.module_utils.ksmanager   r   objectr
   r,   r&   r    r   r   <module>r7      sU    C B"$
 5 v& 6 zF r   