diff --git a/__init__.py b/__init__.py index 8a9d1b7..10e5f45 100644 --- a/__init__.py +++ b/__init__.py @@ -1 +1,3 @@ -default_app_config = 'dbsettings.apps.SettingsConfig' \ No newline at end of file +from dbsettings.functions import * + +default_app_config = 'dbsettings.apps.SettingsConfig' diff --git a/functions.py b/functions.py new file mode 100644 index 0000000..71c895f --- /dev/null +++ b/functions.py @@ -0,0 +1,17 @@ +from dbsettings.models import Setting + +def getValue(key, default=None): + try: + return Setting.objects.get(key=key).value + except: + if default is None: + raise KeyError("No such setting: %s" % key) + else: + return default + +def setValue(key, value): + obj = Setting.objects.get_or_create(key=key)[0] # pylint: disable=E1101 + obj.value = value + obj.save() + return True + diff --git a/tests.py b/tests.py deleted file mode 100644 index 7ce503c..0000000 --- a/tests.py +++ /dev/null @@ -1,3 +0,0 @@ -from django.test import TestCase - -# Create your tests here. diff --git a/views.py b/views.py index 54663dd..6eed39d 100644 --- a/views.py +++ b/views.py @@ -1,17 +1 @@ -from django.shortcuts import render -from dbsettings.models import Setting - -def getValue(key, default=None): - try: - return Setting.objects.get(key=key).value - except: - if default is None: - raise KeyError("No such setting: %s" % key) - else: - return default - -def setValue(key, value): - obj = Setting.objects.get_or_create(key=key)[0] # pylint: disable=E1101 - obj.value = value - obj.save() - return True +from dbsettings.functions import *