Changeset 218 for trunk/python/pyregfi

Timestamp:

03/30/11 11:17:26 (14 years ago)

Author:

tim

Message:

converted hive's .get_root() method into the 'root' property which is cached and better validated for user friendliness

File:

• trunk/python/pyregfi/__init__.py (modified) (4 diffs)

trunk/python/pyregfi/__init__.py

@@ -322 +322 @@
 
     def get_parent(self):
+        if self.is_root():
+            return None
         parent_base = regfi.regfi_get_parentkey(self._hive.file, self._base)
         if parent_base:
             return Key(self._hive, parent_base)
-
         return None
 
     def is_root(self):
-        return (self._hive.get_root() == self)
+        return (self._hive.root == self)
 
 
@@ -408 +409 @@
     file = None
     raw_file = None
-    
+    _root = None
+
     def __init__(self, fh):
         # The fileno method may not exist, or it may throw an exception
@@ -426 +428 @@
 
     def __getattr__(self, name):
+        if name == "root":
+            if self._root == None:
+                self._root = Key(self, regfi.regfi_get_rootkey(self.file))
+            return self._root
+
         return getattr(self.file.contents, name)
     
@@ -435 +442 @@
     def __iter__(self):
         return HiveIterator(self)
-
-    def get_root(self):
-        return Key(self, regfi.regfi_get_rootkey(self.file))