Allow set and save to be chained

easysettings/common_base.py

@@ -259,6 +259,8 @@ def save(self, module, filename=None, **kwargs):
         with BackedUpWriter(self.filename) as f:
             module.dump(self.save_hook(self.data), f, **kwargs)
 
+        return self
+
     def save_hook(self, data):
         """ Called on self.data before JSON encoding, before saving.
             Can be overridden to modify self.data before encoding/saving.
@@ -283,6 +285,7 @@ def set(self, option, value):
                 value   : Value to set for the option/key.
         """
         self.data[option] = value
+        return self
 
     def set_defaults(self, default_config):
         """ Save a copy of keys/value-types from `default_config` to optionally

easysettings/json_settings.py

@@ -101,6 +101,8 @@ def save(self, filename=None, sort_keys=False):
     def setsave(self, option, value, filename=None, sort_keys=False):
         """ The same as calling .set() and then .save(). """
         super(JSONSettings, self).setsave(
+            option,
+            value,
             filename=filename,
             sort_keys=sort_keys,
         )

easysettings/test_settingsbase.py

@@ -475,6 +475,44 @@ def test_setattr(self):
             msg='setattr() did not change the real attribute.',
         )
 
+    def test_setsave(self):
+        """setsave() should work as a shortcut for set() and save()"""
+        settings = self.settings_cls.from_file(self.testfile)
+        settings.setsave('option5', 'value5', filename=self.testfile)
+
+        with open(self.testfile) as f:
+            rawdata = f.read()
+
+        self.assertEqual(
+            settings.option5,
+            'value5',
+            msg='setsave() did not set the attribute.',
+        )
+
+        self.assertTrue(
+            ('option5' in rawdata) and ('value5' in rawdata),
+            msg='Could not find new option in saved data!',
+        )
+
+    def test_chain_set_save(self):
+        """set() and save() can be chained"""
+        settings = self.settings_cls.from_file(self.testfile)
+        settings.set('option6', 'value6').save(filename=self.testfile)
+
+        with open(self.testfile) as f:
+            rawdata = f.read()
+
+        self.assertEqual(
+            settings.option6,
+            'value6',
+            msg='set().save() did not set the attribute.',
+        )
+
+        self.assertTrue(
+            ('option6' in rawdata) and ('value6' in rawdata),
+            msg='Could not find new option in saved data!',
+        )
+
 
 class JSONSettingsBaseTests(SettingsBaseTests):
     def test_encoder_decoder(self):

easysettings/toml_settings.py

@@ -106,4 +106,4 @@ def save(self, filename=None):
 
     def setsave(self, option, value, filename=None):
         """ The same as calling .set() and then .save(). """
-        super(TOMLSettings, self).setsave(filename=filename)
+        super(TOMLSettings, self).setsave(option, value, filename=filename)

easysettings/yaml_settings.py

@@ -101,4 +101,4 @@ def save(self, filename=None):
 
     def setsave(self, option, value, filename=None):
         """ The same as calling .set() and then .save(). """
-        super(YAMLSettings, self).setsave(filename=filename)
+        super(YAMLSettings, self).setsave(option, value, filename=filename)