ansible-collections/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2025-04-05 18:20:30 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 2
community.general/plugins/modules/proxmox_backup_schedule.py
mayabi f09c2a10bc rollback
2025-03-30 13:57:12 +03:30

264 lines
8.5 KiB
Python
Raw Blame History

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Copyright (c) 2025 Marzieh Raoufnezhad <raoufnezhad@gmail.com>
# Copyright (c) 2025 Maryam Mayabi <mayabi.ahm at gmail.com>
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later
from __future__ import absolute_import, division, print_function
__metaclass__ = type
DOCUMENTATION = """
---
module: proxmox_backup_schedule
short_description: Schedule VM backups and removing them
version_added: 10.5.0
description: The module modifies backup jobs such as set or delete C(vmid).
author:
- "Marzieh Raoufnezhad (@raoufnezhad) <raoufnezhad@gmail.com>"
- "Maryam Mayabi (@mmayabi) <mayabi.ahm@gmail.com>"
options:
vm_name:
description:
- The name of the Proxmox VM.
- Mutually exclusive with O(vm_id).
type: str
vm_id:
description:
- The ID of the Proxmox VM.
- Mutually exclusive with O(vm_name).
type: str
backup_id:
description: The backup job ID.
type: str
state:
description:
- If V(present), the module will update backup job with new VM ID.
- If V(absent), the module will remove the VM ID from all backup jobs where the VM ID has existed.
required: true
choices: ["present", "absent"]
type: str
extends_documentation_fragment:
- community.general.proxmox.documentation
- community.general.attributes
- community.general.attributes.info_module
- community.general.proxmox.actiongroup_proxmox
"""
EXAMPLES = """
- name: Scheduling VM Backups based on VM name
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_name: 'VM Name'
backup_id: 'backup-b2adffdc-316e'
state: 'present'
- name: Scheduling VM Backups based on VM ID
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_id: 'VM ID'
backup_id: 'backup-b2adffdc-316e'
state: 'present'
- name: Removing backup setting based on VM name
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_name: 'VM Name'
state: 'absent'
- name: Removing backup setting based on VM ID
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_id: 'VM ID'
state: 'absent'
- name: Removing backup setting based on VM name from specific backup job
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_name: 'VM Name'
backup_id: 'backup-b2adffdc-316e'
state: 'absent'
- name: Removing backup setting based on VM ID from specific backup job
proxmox_backup_schedule:
api_user: 'myUser@pam'
api_password: '*******'
api_host: '192.168.20.20'
vm_id: 'VM ID'
backup_id: 'backup-b2adffdc-316e'
state: 'absent'
"""
RETURN = """
---
backup_schedule:
description:
- If V(present), the backup_schedule will return True after adding the VM ID to the backup job.
- If V(absent), the backup_schedule will return True after removing it from the backup job.
returned: always, but can be empty
type: raw
"""
from ansible.module_utils.basic import AnsibleModule, missing_required_lib
from ansible_collections.community.general.plugins.module_utils.proxmox import (
proxmox_auth_argument_spec, ProxmoxAnsible, HAS_PROXMOXER, PROXMOXER_IMP_ERR)
class ProxmoxSetVMBackupAnsible(ProxmoxAnsible):
# Getting backup sections
def get_cluster_bklist(self):
try:
backupSections = self.proxmox_api.cluster.backup.get()
except Exception as e:
self.module.fail_json(msg="Getting backup sections failed: %s" % e)
return backupSections
def get_cluster_specific_bkjobid(self, backup_id):
try:
specificBackupID = self.proxmox_api.cluster.backup.get(backup_id)
except Exception as e:
self.module.fail_json(msg="Getting specific backup ID failed: %s" % e)
return specificBackupID
def set_vmid_backup(self, backup_id, bk_id_vmids):
try:
self.proxmox_api.cluster.backup.put(backup_id, vmid=bk_id_vmids)
except Exception as e:
self.module.fail_json(msg="Setting vmid backup failed: %s" % e)
return
def get_vms_list(self):
try:
vms = self.proxmox_api.cluster.resources.get(type='vm')
except Exception as e:
self.module.fail_json(msg="Getting vms info from cluster failed: %s" % e)
return vms
# convert vm name to vm ID
def vmname_2_vmid(self, vmname):
vmInfo = self.get_vms_list()
vms = [vm for vm in vmInfo if vm['name'] == vmname]
return (vms[0]['vmid'])
# add vmid to backup job
def backup_present(self, vm_id, backup_id):
bk_id_info = self.get_cluster_specific_bkjobid(backup_id)
# If bk_id_info is a list, get the first item (assuming there's only one backup job returned)
if isinstance(bk_id_info, list):
bk_id_info = bk_id_info[0] # Access the first item in the list
vms_id = bk_id_info['vmid'].split(',')
if str(vm_id) not in vms_id:
bk_id_vmids = bk_id_info['vmid'] + ',' + str(vm_id)
self.set_vmid_backup(backup_id, bk_id_vmids)
return True
else:
return False
# delete vmid from backup job
def backup_absent(self, vm_id, backup_id):
if backup_id:
bk_id_info = self.get_cluster_specific_bkjobid(backup_id)
if isinstance(bk_id_info, list):
bk_id_info = bk_id_info[0] # Access the first item in the list
vmids = bk_id_info['vmid'].split(',')
if str(vm_id) in vmids:
if len(vmids) > 1:
vmids.remove(str(vm_id))
new_vmids = ','.join(map(str, vmids))
self.set_vmid_backup(bk_id_info['id'], new_vmids)
return True
else:
self.module.fail_json(msg="No more than one vmid is assigned to %s. You just can remove job." % bk_id_info['id'])
return False
else:
bkID_delvm = []
backupList = self.get_cluster_bklist()
for backupItem in backupList:
vmids = list(backupItem['vmid'].split(','))
if str(vm_id) in vmids:
if len(vmids) > 1:
vmids.remove(str(vm_id))
new_vmids = ','.join(map(str, vmids))
self.set_vmid_backup(backupItem['id'], new_vmids)
bkID_delvm.append(backupItem['id'])
else:
self.module.fail_json(msg="No more than one vmid is assigned to %s. You just can remove job." % backupItem['id'])
if bkID_delvm:
return True
else:
return False
# main function
def main():
# Define module args
args = proxmox_auth_argument_spec()
backup_schedule_args = dict(
vm_name=dict(type='str'),
vm_id=dict(type='str'),
backup_id=dict(type='str'),
state=dict(choices=['present', 'absent'], required=True)
)
args.update(backup_schedule_args)
module = AnsibleModule(
argument_spec=args,
mutually_exclusive=[('vm_id', 'vm_name')],
supports_check_mode=True
)
# Define (init) result value
result = dict(
changed=False,
message=''
)
# Check if proxmoxer exist
if not HAS_PROXMOXER:
module.fail_json(msg=missing_required_lib('proxmoxer'), exception=PROXMOXER_IMP_ERR)
# Start to connect to proxmox to get backup data
proxmox = ProxmoxSetVMBackupAnsible(module)
vm_name = module.params['vm_name']
vm_id = module.params['vm_id']
backup_id = module.params['backup_id']
state = module.params['state']
if vm_name:
vm_id = proxmox.vmname_2_vmid(vm_name)
if state == 'present':
result['backup_schedule'] = proxmox.backup_present(vm_id, backup_id)
if state == 'absent':
result['backup_schedule'] = proxmox.backup_absent(vm_id, backup_id)
if result['backup_schedule']:
result['changed'] = True
result['message'] = 'The backup schedule has been changed successfully.'
else:
result['message'] = 'The backup schedule did not change anything.'
module.exit_json(**result)
if __name__ == '__main__':
main()
Reference in a new issue View git blame Copy permalink