Make State.as_dict return everything not initially there

1 files changed, 5 insertions, 1 deletions

diff --git a/vmdb/state.py b/vmdb/state.py
index d9f3840..d64244d 100644
--- a/vmdb/state.py
+++ b/vmdb/state.py
@@ -18,9 +18,13 @@
 
 class State(object):
 
+    def __init__(self):
+        self._attrs = {}  # make sure this attribute exists
+        self._attrs = self.as_dict()
+
     def as_dict(self):
         return {
             key: getattr(self, key)
             for key in dir(self)
-            if not key.startswith('_') and not key == 'as_dict'
+            if not key in self._attrs
         }