PythagoreModule.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Pythagore ("2.0")
# A Python IRC Bot
#
# PythagoreModule.py : Module class for Pythagore bot
#
# Copyright © 2007 Nicolas Dandrimont <Nicolas.Dandrimont@crans.org>
#
# This file is part of Pythagore.
#
# Pythagore is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License, version 2, as
# published by the Free Software Foundation.
#
# Pythagore is distributed in the hope it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# of FITNESS FOR ANY PARTICULAR PURPOSE. See the GNU General Public
# License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Pythagore; if not, write to the Free Software Foundation,
# Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA

import os

class PythagoreModule(object):
    """This is the class all Pythagore's Modules should subclass
    In fact it is quite empty..."""

    def __init__(self, pythagore):
        """Initialization takes a pythagore argument which is the current bot
        instance"""
        self.exports = {}
        self.bot = pythagore
        self.config = self.load_config()
        self.name = self.__class__.__name__

    def load_config(self):
        """Configuration initialization. Override this if you do not want
        plaintext config"""
        try:
            import yaml
        except ImportError:
            return {}

        name = self.__class__.__name__
        try:
            configfile = file(os.path.join("Config", name + ".yml"), 'r')
            config = yaml.safe_load(configfile)
            configfile.close()
        except IOError:
            print _("Config file for %(module)s not open !") % {'module': name}
            config = {}
        return config

    def save_config(self):
        """Configuration file save. Should be more or less failproof"""
        try:
            import yaml
        except ImportError:
            return

        name = self.__class__.__name__
        try:
            newconfigfile = file(os.path.join("Config", name + ".new.yml"), 'w')
            yaml.dump(self.config, newconfigfile, default_flow_style=False)
            os.rename(os.path.join("Config", name + ".new.yml"),
                      os.path.join("Config", name + ".yml"))
        except IOError:
            return