moved new fact implementation to fact app

awx/fact/models/__init__.py

@@ -0,0 +1,6 @@
+# Copyright (c) 2015 Ansible, Inc.
+# All Rights Reserved
+
+from __future__ import absolute_import
+
+from .fact import * # noqa

awx/fact/models/fact.py

@@ -0,0 +1,95 @@
+from mongoengine import Document, DynamicDocument, DateTimeField, ReferenceField, StringField
+
+class FactHost(Document):
+    hostname = StringField(max_length=100, required=True, unique=True)
+
+    # TODO: Consider using hashed index on hostname. django-mongo may not support this but
+    # executing raw js will
+    meta = {
+        'indexes': [
+            'hostname' 
+        ]
+    }
+
+    @staticmethod
+    def get_host_id(hostname):
+        host = FactHost.objects.get(hostname=hostname)
+        if host:
+            return host.id
+        return None
+
+class Fact(DynamicDocument):
+    timestamp = DateTimeField(required=True)
+    host = ReferenceField(FactHost, required=True)
+    module = StringField(max_length=50, required=True)
+    # fact = <anything>
+
+    # TODO: Consider using hashed index on host. django-mongo may not support this but
+    # executing raw js will
+    meta = {
+        'indexes': [
+            '-timestamp',
+            'host'
+        ]
+    }
+
+    @staticmethod
+    def add_fact(timestamp, fact, host, module):
+        fact_obj = Fact(timestamp=timestamp, host=host, module=module, fact=fact)
+        fact_obj.save()
+        version_obj = FactVersion(timestamp=timestamp, host=host, module=module, fact=fact_obj)
+        version_obj.save()
+        return (fact_obj, version_obj)
+
+    # TODO: if we want to relax the need to include module...
+    # If module not specified then filter query may return more than 1 result.
+    # Thus, the resulting facts must somehow be unioned/concated/ or kept as an array.
+    @staticmethod
+    def get_host_version(hostname, timestamp, module):
+        try:
+            host = FactHost.objects.get(hostname=hostname)
+        except FactHost.DoesNotExist:
+            return None
+
+        kv = {
+            'host' : host.id,
+            'timestamp__lte': timestamp,
+            'module': module,
+        }
+
+        try:
+            facts = Fact.objects.filter(**kv)
+            if not facts:
+                return None
+            return facts[0]
+        except Fact.DoesNotExist:
+            return None
+
+    @staticmethod
+    def get_host_timeline(hostname, module):
+        try:
+            host = FactHost.objects.get(hostname=hostname)
+        except FactHost.DoesNotExist:
+            return None
+
+        kv = {
+            'host': host.id,
+            'module': module,
+        }
+
+        return FactVersion.objects.filter(**kv).values_list('timestamp')
+
+class FactVersion(Document):
+    timestamp = DateTimeField(required=True)
+    host = ReferenceField(FactHost, required=True)
+    module = StringField(max_length=50, required=True)  
+    fact = ReferenceField(Fact, required=True)
+    # TODO: Consider using hashed index on module. django-mongo may not support this but
+    # executing raw js will
+    meta = {
+        'indexes': [
+            '-timestamp',
+            'module'
+        ]
+    }
+