From 33f368910f856a76dcbacb00e8f90c1d173a47bd Mon Sep 17 00:00:00 2001
From: Ian Gulliver <ian@flamingcow.tv>
Date: Fri, 19 Jun 2015 22:43:20 -0700
Subject: [PATCH] Better method for tracking reconnect of polling clients.

---
 api.py        | 2 +-
 lib/models.py | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/api.py b/api.py
index b953618..2eb412b 100644
--- a/api.py
+++ b/api.py
@@ -75,7 +75,7 @@ def Poll(google_user, client, client_address, instance_id, args):
 
   return {
     'result': 'ok',
-    'new': instance.newly_created,
+    'instance_generation': instance.generation,
     'events': events,
   }
 
diff --git a/lib/models.py b/lib/models.py
index 875a768..c2d1dad 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -18,6 +18,7 @@ import hashlib
 import logging
 import random
 import struct
+import uuid
 
 from google.appengine.api import channel
 from google.appengine.api import users
@@ -95,6 +96,7 @@ class Instance(db.Model):
   active = db.BooleanProperty(required=True, default=False)
   polling = db.BooleanProperty(required=True, default=False)
   last_poll = db.DateTimeProperty(required=True, auto_now=True)
+  generation = db.StringProperty(required=True)
 
   @classmethod
   def FromID(cls, instance_id):
@@ -107,11 +109,12 @@ class Instance(db.Model):
     def _FindOrCreate():
       entity = cls.get_by_key_name(instance_id)
       if entity:
-        entity.newly_created = False
         return entity
-      entity = cls(key_name=instance_id, **kwargs)
+      entity = cls(
+          key_name=instance_id,
+          generation=str(uuid.uuid4()),
+          **kwargs)
       entity.put()
-      entity.newly_created = True
       return entity
     return db.run_in_transaction(_FindOrCreate)