From 8435ddccdfcd49d9946cbe2d48c01fbf280bcc66 Mon Sep 17 00:00:00 2001
From: jMyles <justin@justinholmes.com>
Date: Thu, 14 Dec 2017 11:20:57 -0800
Subject: [PATCH] Alice now makes a Policy instead of a PolicyGroup.

---
 nkms/policy/models.py | 24 +++++++++++-------------
 1 file changed, 11 insertions(+), 13 deletions(-)

diff --git a/nkms/policy/models.py b/nkms/policy/models.py
index def3efd61..0d71152f6 100644
--- a/nkms/policy/models.py
+++ b/nkms/policy/models.py
@@ -1,7 +1,7 @@
+import asyncio
 import binascii
 
 import msgpack
-import asyncio
 
 from nkms.characters import Alice, Bob, Ursula
 from nkms.crypto import api
@@ -59,19 +59,17 @@ class PolicyManagerForAlice(PolicyManager):
 
         ##### Temporary until we decide on an API for private key access
         alice_priv_enc = self.owner._crypto_power._power_ups[EncryptingPower].priv_key
-        re_enc_keys, pfrag = self.owner.generate_rekey_frags(alice_priv_enc, bob, m,
-                                                                     n)  # TODO: Access Alice's private key inside this method.
-        policies = []
-        for kfrag_id, kfrag in enumerate(re_enc_keys):
-            policy = Policy.from_alice(
-                alice=self.owner,
-                bob=bob,
-                kfrag=kfrag,
-                pfrag=pfrag,
-            )
-            policies.append(policy)
+        kfrags, pfrag = self.owner.generate_rekey_frags(alice_priv_enc, bob, m,
+                                                             n)  # TODO: Access Alice's private key inside this method.
+        policy = Policy.from_alice(
+            alice=self.owner,
+            bob=bob,
+            kfrags=kfrags,
+            pfrag=pfrag,
+            uri=uri,
+        )
 
-        return PolicyGroup(uri, self.owner, bob, policies)
+        return policy
 
 
 class PolicyGroup(object):