mravi/uncloud-mravi
1
0
Fork 0
forked from uncloud/uncloud
Code Issues Pull requests Projects Releases Wiki Activity

fix migrations the ugly way

Browse source 
Signed-off-by: Nico Schottelius <nico@nico-notebook.schottelius.org>
This commit is contained in:
Ahmed Bilal 2020-02-26 11:31:17 +01:00
commit 0c7ca1147a
28 changed files with 524 additions and 432 deletions
Download patch file Download diff file Expand all files Collapse all files

37
uncloud/opennebula/management/commands/syncvm.py
View file

@ -8,15 +8,10 @@ from xmlrpc.client import ServerProxy as RPCClient
from django.core.management.base import BaseCommand
from django.contrib.auth import get_user_model
from xmltodict import parse
from ungleich_common.ldap.ldap_manager import LdapManager
from opennebula.models import VM as VMModel
def find_user_based_on_email(users, email):
for user in users:
if email in user.mail.values:
return user
from django_auth_ldap.backend import LDAPBackend
class Command(BaseCommand):
@ -26,39 +21,29 @@ class Command(BaseCommand):
pass
def handle(self, *args, **options):
ldap_server_uri = secrets.LDAP_SERVER_URI.split(',')[0]
ldap_manager = LdapManager(
server=ldap_server_uri,
admin_dn=secrets.LDAP_ADMIN_DN,
admin_password=secrets.LDAP_ADMIN_PASSWORD,
)
users = ldap_manager.get('') # Get all users
with RPCClient(secrets.OPENNEBULA_URL) as rpc_client:
success, response, *_ = rpc_client.one.vmpool.infoextended(
secrets.OPENNEBULA_USER_PASS, -2, -1, -1, -1
)
if success:
vms = json.loads(json.dumps(parse(response)))['VM_POOL']['VM']
unknown_user_with_email = set()
unknown_user = set()
backend = LDAPBackend()
for vm in vms:
vm_id = vm['ID']
vm_owner_email = vm['UNAME']
vm_owner = vm['UNAME']
user = backend.populate_user(username=vm_owner)
user = find_user_based_on_email(users, vm_owner_email)
if not user:
unknown_user_with_email.add(vm_owner_email)
unknown_user.add(vm_owner)
else:
try:
user_in_db = get_user_model().objects.get(email=vm_owner_email)
except get_user_model().DoesNotExist:
user_in_db = get_user_model().objects.create_user(username=user.uid, email=vm_owner_email)
VMModel.objects.update_or_create(
id=f'opennebula{vm_id}',
defaults={'data': vm, 'owner': user_in_db}
vmid=vm_id,
defaults={'data': vm, 'owner': user}
)
print('User with email but not found in ldap:', unknown_user_with_email)
print('User not found in ldap:', unknown_user)
else:
print(response)
print(secrets.OPENNEBULA_USER_PASS)