From f2b638f5fa8644e596349f09ddd79fc0c9e5ec40 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Mon, 18 Jul 2016 15:57:49 +0000
Subject: Committing latest changes to anon collection stuff, but still need to
 fix some issues

---
 wqflask/wqflask/collect.py                      | 23 +++++++++++++----------
 wqflask/wqflask/templates/collections/view.html |  2 +-
 wqflask/wqflask/user_manager.py                 |  9 +++++++--
 3 files changed, 21 insertions(+), 13 deletions(-)

(limited to 'wqflask')

diff --git a/wqflask/wqflask/collect.py b/wqflask/wqflask/collect.py
index d5ed89cd..733a4df0 100644
--- a/wqflask/wqflask/collect.py
+++ b/wqflask/wqflask/collect.py
@@ -56,7 +56,7 @@ class AnonCollection(object):
     """User is not logged in"""
     def __init__(self, collection_name):
         anon_user = user_manager.AnonUser()
-        self.key = "anon_collection:v1:{}".format(anon_user.anon_id)
+        self.key = anon_user.key
         self.name = collection_name
         self.id = uuid.uuid4()
         self.created_timestamp = datetime.datetime.utcnow().strftime('%b %d %Y %I:%M%p')
@@ -82,12 +82,13 @@ class AnonCollection(object):
         #assert collection_name == "Default", "Unexpected collection name for anonymous user"
         #print("params[traits]:", params['traits'])
         self.traits = list(process_traits(params['traits']))
-        print("traits is:", self.traits)
-        print("self.key is:", self.key)
+        #print("traits is:", self.traits)
+        #print("self.key is:", self.key)
         #len_before = len(Redis.smembers(self.key))
         existing_collections = Redis.get(self.key)
         print("EXISTING COLLECTIONS:", existing_collections)
         if existing_collections != "None":
+            print("EXISTING COLLECTION NOT NONE")
             collections_dict = json.loads(existing_collections)
             #print("EXISTING COLLECTIONS:", collections_dict)
             if self.id in collections_dict.keys():
@@ -108,7 +109,7 @@ class AnonCollection(object):
             collections_dict = {str(self.id) : new_collection_dict}
             
         Redis.set(self.key, json.dumps(collections_dict))
-        print("COLLECTIONS_DICT:", Redis.get(self.key))
+        #print("COLLECTIONS_DICT:", Redis.get(self.key))
         #Redis.sadd(self.key, *list(traits))
         #Redis.expire(self.key, 60 * 60 * 24 * 5)
         #print("currently in redis:", Redis.smembers(self.key))
@@ -208,7 +209,7 @@ def collections_add():
                                collections = user_collections,
                                )
     else:
-        anon_collections = user_manager.AnonUser().get_collections()
+        anon_collections = list(user_manager.AnonUser().get_collections().keys())
         return render_template("collections/add.html",
                                    traits = traits,
                                    collections = anon_collections,
@@ -221,7 +222,7 @@ def collections_add():
 @app.route("/collections/new")
 def collections_new():
     params = request.args
-    print("request.args in collections_new are:", params)
+    #print("request.args in collections_new are:", params)
 
     collection_name = params['new_collection']
     
@@ -239,7 +240,8 @@ def collections_new():
         if g.user_session.logged_in:
             return UserCollection().add_traits(params, collection_name)
         else:
-            return AnonCollection().add_traits(params, collection_name)
+            #print("PARAMS ADD TO COLLECTION:", params)
+            return AnonCollection().add_traits(params)
     else:
         print("ELSE")
         CauseAnError
@@ -257,7 +259,7 @@ def process_traits(unprocessed_traits):
         #print("data is:", data)
         #print("hmac is:", hmac)
         assert hmac==user_manager.actual_hmac_creation(data), "Data tampering?"
-        traits.add(str(data))
+        traits.add                                                                                               (str(data))
     return traits
 
 def create_new(collection_name):
@@ -389,9 +391,10 @@ def view_collection():
 
     if "uc_id" in params:
         collection_info = dict(trait_obs=trait_obs,
-                           uc = uc)
+                               uc = uc)
     else:
-        collection_info = dict(trait_obs=trait_obs)
+        collection_info = dict(trait_obs=trait_obs,
+                               collection_name=this_collection['name'])
     if "json" in params:
         print("json_version:", json_version)
         return json.dumps(json_version)
diff --git a/wqflask/wqflask/templates/collections/view.html b/wqflask/wqflask/templates/collections/view.html
index 288207e7..4b1752cd 100644
--- a/wqflask/wqflask/templates/collections/view.html
+++ b/wqflask/wqflask/templates/collections/view.html
@@ -14,7 +14,7 @@
         <h2>{{ uc.name }}</h2>
         <h3>{{ 'This collection has {}.'.format(numify(trait_obs|count, "record", "records")) }}</h3>
     {% else %}
-        <h2>Your Collection</h2>
+        <h2> {{ collection_name }}</h2>
         <h3>{{ 'This collection has {}.'.format(numify(trait_obs|count, "record", "records")) }}</h3>
     {% endif %}
     </div>
diff --git a/wqflask/wqflask/user_manager.py b/wqflask/wqflask/user_manager.py
index f0655ff4..c7b3bdb4 100644
--- a/wqflask/wqflask/user_manager.py
+++ b/wqflask/wqflask/user_manager.py
@@ -75,6 +75,7 @@ class AnonUser(object):
             logger.debug("creating new cookie")
             self.anon_id, self.cookie = create_signed_cookie()
         self.key = "anon_collection:v5:{}".format(self.anon_id)
+        print("THE KEY IS:", self.key)
 
         @after.after_this_request
         def set_cookie(response):
@@ -94,8 +95,12 @@ class AnonUser(object):
             
     def get_collections(self):
         collections = Redis.get(self.key)
-        print("COLLECTIONS:", collections)
-        return collections
+        if collections == "None":
+            return {}
+        else:
+            return json.loads(collections)
+        #print("COLLECTIONS:", collections)
+        #return collections
 
 
 def verify_cookie(cookie):
-- 
cgit v1.2.3