Improving endpoint rotation logic (#545)

lemur/sources/models.py

@@ -6,11 +6,12 @@
 .. moduleauthor:: Kevin Glisson <kglisson@netflix.com>
 """
 from sqlalchemy.orm import relationship
-from sqlalchemy import Column, Integer, String, Text, DateTime, Boolean
+from sqlalchemy import Column, Integer, String, Text, Boolean
 from sqlalchemy_utils import JSONType
 from lemur.database import db
 
 from lemur.plugins.base import plugins
+from sqlalchemy_utils import ArrowType
 
 
 class Source(db.Model):
@@ -21,7 +22,7 @@ class Source(db.Model):
     description = Column(Text())
     plugin_name = Column(String(32))
     active = Column(Boolean, default=True)
-    last_run = Column(DateTime)
+    last_run = Column(ArrowType)
     endpoints = relationship("Endpoint", back_populates="source")
 
     @property

lemur/sources/service.py

@@ -141,6 +141,7 @@ def sync_endpoints(source):
 
         policy['ciphers'] = policy_ciphers
         endpoint['policy'] = endpoint_service.get_or_create_policy(**policy)
+        endpoint['source'] = source
 
         if not exists:
             endpoint_service.create(**endpoint)