plstackapi/core/api/keys.py

   1 from plstackapi.openstack.client import OpenStackClient
   2 from plstackapi.openstack.driver import OpenStackDriver
   3 from plstackapi.core.api.auth import auth_check
   4 from plstackapi.core.models import Key, User
   5
   6 def lookup_user(fields):
   7     user = None
   8     if 'user' in fields:
   9         if isinstance(fields['user'], int):
  10             users = User.objects.filter(id=fields['user'])
  11         else:
  12             users = User.objects.filter(email=fields['user'])
  13         if users:
  14             user = users[0]
  15     if not user:
  16         raise Exception, "No such user: %s" % fields['user']
  17     return user
  18
  19 def add_key(auth, fields):
  20     driver = OpenStackDriver(client = auth_check(auth))
  21     user = lookup_user(fields)
  22     if user: fields['user'] = user
  23     key = Key(**fields)
  24     nova_fields = {'name': key.name,
  25                    'key': key.key}
  26     nova_key = driver.create_keypair(**nova_fields)
  27     key.save()
  28     return key
  29
  30 def update_key(auth, id, **fields):
  31     return
  32
  33 def delete_key(auth, filter={}):
  34     driver = OpenStackDriver(client = auth_check(auth))
  35     keys = Key.objects.filter(**filter)
  36     for key in keys:
  37         driver.delete_keypair(name=key.name)
  38         key.delete()
  39     return 1
  40
  41 def get_keys(auth, filter={}):
  42     client = auth_check(auth)
  43     keys = Key.objects.filter(**filter)
  44     return keys
  45
  46
  47