add AutoGeneratedRecord class

usc-isi-i2 · Dec 7, 2018 · f6be3fe · f6be3fe
1 parent 086fde1
commit f6be3fe
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/rltk/__init__.py b/rltk/__init__.py
@@ -1,4 +1,5 @@
-from rltk.record import Record, cached_property, generate_record_property_cache, validate_record, remove_raw_object
+from rltk.record import Record, AutoGeneratedRecord,\
+    cached_property, generate_record_property_cache, validate_record, remove_raw_object
 from rltk.dataset import Dataset
 from rltk.io import *
 from rltk.similarity import *

diff --git a/rltk/record.py b/rltk/record.py
@@ -122,3 +122,18 @@ def get_property_names(cls: type):
             continue
         keys.append(prop_name)
     return keys
+
+
+@remove_raw_object
+class AutoGeneratedRecord(Record):
+
+    def __init__(self, raw_object):
+        super().__init__(raw_object)
+        for k, v in raw_object.items():
+            if k == 'id':
+                continue
+            setattr(self, k, v)
+
+    @cached_property
+    def id(self):
+        return self.raw_object['id']