configurationengine/source/cone/public/parsecontext.py
author m2lahtel
Tue, 10 Aug 2010 14:29:28 +0300
changeset 3 e7e0ae78773e
permissions -rw-r--r--
ConE 1.2.11 release

#
# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
# All rights reserved.
# This component and the accompanying materials are made available
# under the terms of "Eclipse Public License v1.0"
# which accompanies this distribution, and is available
# at the URL "http://www.eclipse.org/legal/epl-v10.html".
#
# Initial Contributors:
# Nokia Corporation - initial contribution.
#
# Contributors:
#
# Description: 
#

import threading
import logging

import cone.public.utils

class ParseContext(object):
    """
    Parse context class for handling exceptions and problems encountered
    when parsing a file (ConfML, ImplML or anything included from those).
    
    The context contains the path to the file that is currently being parsed,
    but the file path can also be overridden when calling a handler method
    (e.g. an ImplML file includes some other file, which in contains a problem).
    
    Sub-classes should override _handle_exception() and _handle_problem(),
    NOT handle_exception() and handle_problem().
    """
    def __init__(self):
        #: The file that is currently being parsed.
        self.current_file = None
    
    def handle_exception(self, exception, file_path=None):
        """
        Handle an exception that occurred while parsing a file.
        
        @param exception: The exception that occurred.
        @param file_path: Override for the name of the file that was being parsed.
            If this is None, self.current_file is used.
        """
        if file_path is None:
            file_path = self.current_file
        
        self._handle_exception(exception, file_path)
    
    def handle_problem(self, problem, file_path=None):
        """
        Handle a Problem that occurred while parsing a file.
        
        @param problem: The problem object.
        @param file_path: Override for the name of the file that was being parsed.
            If this is None, self.current_file is used.
        """
        if file_path is None:
            file_path = self.current_file
        
        self._handle_problem(problem, file_path)
    
    def _handle_exception(self, exception, file_path):
        """
        Called to handle an exception that occurred while parsing a file.
        
        Note that this should always be called in an exception handler,
        so you can safely use e.g. traceback.format_exc().
        
        @param exception: The exception that occurred.
        @param file_path: The file that was being parsed.
        """
        cone.public.utils.log_exception(
            logging.getLogger('cone'),
            "Error parsing '%s': %s" % (file_path, str(exception)))
    
    def _handle_problem(self, problem, file_path):
        """
        Called when parsing a file yields a problem.
        @param problem: The problem object.
        @param file_path: Override for the name of the file that was being parsed.
            If this is None, self.current_file is used.
        """
        problem.file = file_path
        problem.log(logging.getLogger('cone'))

# Thread-local storage for the contexts
_contexts = threading.local()

def _init_contexts():
    """
    Initialize thread-local contexts if they are not initialized yet.
    """
    if not hasattr(_contexts, 'implml'):
        _contexts.implml = None
        _contexts.confml = None
        _contexts.implml_default = ParseContext()
        _contexts.confml_default = ParseContext()

def get_implml_context():
    """
    Get the current ImplML parse context.
    """
    _init_contexts()
    if _contexts.implml is not None:
        return _contexts.implml
    else:
        return _contexts.implml_default

def set_implml_context(context):
    """
    Set the current ImplML parse context.
    @param context: The context to set. If None, the default context
        will be used.
    """
    _init_contexts()
    _contexts.implml = context

def get_confml_context():
    """
    Get the current ConfML parse context.
    """
    _init_contexts()
    if _contexts.confml is not None:
        return _contexts.confml
    else:
        return _contexts.confml_default

def set_confml_context(context):
    """
    Set the current ConfML parse context.
    @param context: The context to set. If None, the default context
        will be used.
    """
    _init_contexts()
    _contexts.confml = context