import json from xmlrpc.client import ServerProxy as RPCClient from django_auth_ldap.backend import LDAPBackend from django.core.management.base import BaseCommand from django.conf import settings from xmltodict import parse from opennebula.models import VM as VMModel class Command(BaseCommand): help = 'Syncronize VM information from OpenNebula' def add_arguments(self, parser): pass def handle(self, *args, **options): with RPCClient(settings.OPENNEBULA_URL) as rpc_client: success, response, *_ = rpc_client.one.vmpool.infoextended( settings.OPENNEBULA_USER_PASS, -2, -1, -1, -1 ) if success: vms = json.loads(json.dumps(parse(response)))['VM_POOL']['VM'] unknown_user = set() backend = LDAPBackend() for vm in vms: vm_id = vm['ID'] vm_owner = vm['UNAME'] user = backend.populate_user(username=vm_owner) if not user: unknown_user.add(vm_owner) else: VMModel.objects.update_or_create( vmid=vm_id, defaults={'data': vm, 'owner': user} ) print('User not found in ldap:', unknown_user) else: print(response)