Merge pull request #8 from jsocol/master

updates from source

.gitignore

@@ -10,4 +10,8 @@ dist
 .pydevproject
 .settings
 .tox/
+.DS_Store
 docs/_*
+
+# Visual Studio Code config files
+.vscode

waffle/managers.py

@@ -11,6 +11,9 @@
 class BaseManager(models.Manager):
     KEY_SETTING = ''
 
+    def get_by_natural_key(self, name):
+        return self.get(name=name)
+
     def create(self, *args, **kwargs):
         ret = super(BaseManager, self).create(*args, **kwargs)
         cache_key = get_setting(self.KEY_SETTING)

waffle/models.py

@@ -31,6 +31,9 @@ class Meta(object):
     def __str__(self):
         return self.name
 
+    def natural_key(self):
+        return (self.name,)
+
     @classmethod
     def _cache_key(cls, name):
         return keyfmt(get_setting(cls.SINGLE_CACHE_KEY), name)

waffle/tests/test_models.py

@@ -0,0 +1,20 @@
+from __future__ import unicode_literals
+
+from django.test import TestCase
+
+from waffle.models import Flag, Switch, Sample
+
+
+class ModelsTests(TestCase):
+    def test_natural_keys(self):
+        flag = Flag.objects.create(name='test-flag')
+        switch = Switch.objects.create(name='test-switch')
+        sample = Sample.objects.create(name='test-sample', percent=0)
+
+        self.assertEqual(flag.natural_key(), ('test-flag',))
+        self.assertEqual(switch.natural_key(), ('test-switch',))
+        self.assertEqual(sample.natural_key(), ('test-sample',))
+
+        self.assertEqual(Flag.objects.get_by_natural_key('test-flag'), flag)
+        self.assertEqual(Switch.objects.get_by_natural_key('test-switch'), switch)
+        self.assertEqual(Sample.objects.get_by_natural_key('test-sample'), sample)