Added get() method to objects (#3)

remodel/models.py

@@ -86,6 +86,12 @@ def delete(self):
         for field in self.fields.related:
             delattr(self.fields, field)
 
+    def get(self, key, default=None):
+        try:
+            return getattr(self.fields, key)
+        except AttributeError:
+            return default
+
     def __getitem__(self, key):
         try:
             return getattr(self.fields, key)

tests/test_models.py

@@ -286,3 +286,24 @@ def test_already_deleted(self):
             a.delete()
 
     # TODO: Add tests for confirming that related objects have no reference left to the deleted object
+
+
+class GetTests(BaseTestCase):
+    def setUp(self):
+        super(GetTests, self).setUp()
+
+        class Artist(Model):
+            pass
+        self.Artist = Artist
+
+    def test_existing_field(self):
+        a = self.Artist(name='Andrei')
+        assert a.get('name') == 'Andrei'
+
+    def test_inexisting_field_with_default(self):
+        a = self.Artist()
+        assert a.get('name', 'Andrei') == 'Andrei'
+
+    def test_inexisting_field_without_default(self):
+        a = self.Artist()
+        assert a.get('name') is None