~ubuntu-branches/ubuntu/natty/quickly/natty

# -*- coding: utf-8 -*-

# Copyright 2009 Canonical Ltd.

# Author 2009 Didier Roche

#

# This file is part of Quickly

#

#This program is free software: you can redistribute it and/or modify it 

#under the terms of the GNU General Public License version 3, as published 

#by the Free Software Foundation.

 

#This program is distributed in the hope that it will be useful, but 

#WITHOUT ANY WARRANTY; without even the implied warranties of 

#MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A 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 this program.  If not, see <http://www.gnu.org/licenses/>.

 

import os

import sys

 

import gettext

from gettext import gettext as _

 

from quickly import tools

 

global_config = {} # retrieve from /etc/quickly

project_config = {} # retrieved from project/.quickly

 

def loadConfig(global_config=None):

    """ load configuration from /etc/quickly or .quickly file"""

 

    if global_config:

        quickly_file_path = '/etc/quickly'

        config = global_config

    else:

        # retrieve .quickly file

        try:

            quickly_file_path = tools.get_root_project_path()  + '/.quickly'

            config = project_config

        except tools.project_path_not_found:

            print _("ERROR: Can't load configuration in current path or its parent ones.")

            sys.exit(1)

        

        

    try:

        fileconfig = file(quickly_file_path, 'rb')

        for line in fileconfig: 

            fields = line.split('#')[0] # Suppress commentary after the value in configuration file and in full line

            fields = fields.split('=') # Separate variable from value

            # normally, we have two fields in "fields"

            if len(fields) == 2:

                config[fields[0].strip()] = fields[1].strip() 

        fileconfig.close()

    except (OSError, IOError), e:

        print _("ERROR: Can't load configuration in %s: %s" % (quickly_file_path, e))

        sys.exit(1)

 

 

def saveConfig(global_config=None, config_file_path=None):

    """ save the configuration file from config dictionnary in project or global quuickly file

 

    config_file_path is optional (needed by the create command, for instance).

    getcwd() is taken by default.    

    

    keep commentaries and layout from original file """

 

    if global_config:

        quickly_file_path = '/etc/quickly'

        config = global_config

    else:

        # retrieve .quickly file

        try:

            quickly_file_path = tools.get_root_project_path(config_file_path) + '/.quickly'

        # if no .quickly, create it using config_file_path or cwd

        except tools.project_path_not_found:

            if config_file_path is not None:

                quickly_file_path = os.path.abspath(config_file_path) + '/.quickly'

            else:

                quickly_file_path = os.getcwd() + "/.quickly"

            print _("WARNING: No .quickly file found. Initiate a new one")

        config = project_config

    

    try:

        filedest = file(quickly_file_path + '.swp', 'w')

        try:

            fileconfig = file(quickly_file_path, 'rb')

            remaingconfigtosave = config.copy()

            for line in fileconfig:

                fields = line.split('#')[0] # Suppress commentary after the value in configuration file and in full line

                fieldsafter = line.split('#')[1:]

                fields = fields.split('=') # Separate variable from value

                # normally, we have two fields in "fields" and it should be used by config tabular

                if len(fields) == 2 and fields[0].strip() in remaingconfigtosave:

                    line = fields[0].strip() + " = " + str(remaingconfigtosave.pop(fields[0].strip()))

                    if len(fieldsafter) > 0:

                        line = line + " #" + "#".join(fieldsafter) # fieldsafter already contains \n

                    else:

                        line = line + "\n"

                filedest.write(line) # commentaries or empty lines, anything other things which is not useful will be printed unchanged

            # write remaining data if some (new data not in the old config file).

            filedest.write("".join(elem + " = " + str(remaingconfigtosave[elem]) + '\n' for elem in remaingconfigtosave)) #\n here for last element (and not be added, when no iteration to do)

#            print "\n".join(elem + " = " + remaingconfigtosave[elem] for elem in remaingconfigtosave)

            fileconfig.close()

        except (OSError, IOError), e:      

            # write config file from scratch (no previous file found)

            filedest.write("\n".join(elem + " = " + str(config[elem]) for elem in config) + "\n")

        finally:

            filedest.close()

            os.rename(filedest.name, quickly_file_path)

    except AttributeError, e:

        print _("ERROR: Can't save configuration in %s" % quickly_file_path)

        return 1

    return 0