34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""
|
|
Custom manager for ConfigValue objects.
|
|
"""
|
|
from django.db import models
|
|
from src import logger
|
|
|
|
class ConfigValueManager(models.Manager):
|
|
def get_configvalue(self, configname):
|
|
"""
|
|
Retrieve a configuration value.
|
|
"""
|
|
try:
|
|
return self.get(conf_key__iexact=configname).conf_value
|
|
except self.model.DoesNotExist:
|
|
if configname not in ["game_firstrun"]:
|
|
logger.log_errmsg("Unable to get config value for %s (does not exist).\n" % (
|
|
configname))
|
|
raise
|
|
|
|
def set_configvalue(self, configname, newvalue):
|
|
"""
|
|
Sets a configuration value with the specified name.
|
|
Returns the new value for the directive.
|
|
"""
|
|
try:
|
|
conf = self.get(conf_key=configname)
|
|
conf.conf_value = newvalue
|
|
conf.save()
|
|
# We'll do this instead of conf.conf_value, might save a DB query.
|
|
return newvalue
|
|
except self.model.DoesNotExist:
|
|
logger.log_errmsg("Unable to set config value for %s (does not exist):\n%s" % (
|
|
configname))
|
|
raise
|