Skip to content

Commit f4a7ad2

Browse files
committed
Merge function with default
2 parents eeb3659 + 065293d commit f4a7ad2

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

keyvaluestore/utils.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,22 @@
11
from keyvaluestore.models import KeyValueStore
2+
from keyvaluestore.exceptions import KeyNotExistingException
3+
24

35
def get_value_for_key(key):
46
return KeyValueStore.objects.get_value_for_key(key)
57

8+
9+
def get_value_or_default(key, default):
10+
try:
11+
value = get_value_for_key(key)
12+
except KeyNotExistingException:
13+
value = default
14+
return value
15+
16+
617
def set_key_value(key, value):
7-
obj,created = KeyValueStore.objects.get_or_create(key=key, defaults={'value': value})
18+
obj, created = KeyValueStore.objects.get_or_create(key=key, defaults={'value': value})
819
if not created:
920
obj.value = value
1021
obj.save()
11-
return True
22+
return True

0 commit comments

Comments
 (0)