#!/usr/bin/env python
"""Loads and manages the bot settings."""

from botly.db import Db


class Settings(Db):
    
    def __init__(self):
        self.settings = {}

    def get_string(self, key, default='None'):
        """Returns the string value if it exists."""
        if not key in self.settings: 
            self.settings[key] = self._load_setting(key, default)
        return self.settings[key]

    def get_int(self, key, default='0'):
        """Returns the int value if it exists."""
        # Default default is a string because passed to get_string.
        # It's converted back to int below
        v = self.get_string(key, default=default)
        return int(v)

    def _load_setting(self, key, default):
        assert self.is_loaded(), "Database is not loaded."
        setting = Db.dbxml.xpath('/BotDb/Settings/' + key)
        if setting:
            return setting[0].text
        else:
            return default