Source code for anybox.recipe.odoo.runtime.session

"""Utilities to start a server process."""
import warnings
import sys
import os
import logging
from distutils.version import Version

    import openerp
except ImportError:
    warnings.warn("This must be imported with a buildout openerp recipe "
                  "driven sys.path", RuntimeWarning)
        from openerp.cli import server as startup
    except ImportError:
        from .backports.cli import server as startup
    from import config
    from openerp import SUPERUSER_ID
    from import parse_version

from optparse import OptionParser  # we support python >= 2.6

logger = logging.getLogger(__name__)

DEFAULT_VERSION_PARAMETER = 'buildout.db_version'


[docs]class OpenERPVersion(Version): """OpenERP idea of version, wrapped in a class. This is based on :meth:``, and Provides straight-ahead comparison with tuples of integers, or distutils Version classes. """
[docs] def parse(self, incoming): if isinstance(incoming, OpenERPVersion): self.vstring = incoming.vstring self.components = incoming.components else: self.vstring = incoming self.components = parse_version(incoming)
def __str__(self): return self.vstring def __repr__(self): return 'OpenERPVersion(%r)' % str(self) def __cmp__(self, other): if isinstance(other, tuple): other = '.'.join(str(s) for s in other) elif not isinstance(other, self.__class__): other = str(other) # Works with distutils' Version classes other = self.__class__(other) return cmp(self.components, other.components)
[docs]class Session(object): """A class to give server-level access to one database. There should be exactly one instance of this class per process. It can be used for any kind of script involving OpenERP API, and provides facilities for upgrade scripts (see also :mod:anybox.recipe.odoo.runtime.upgrade) Before actual use, call :meth:`open`. Then you'll have useful attributes and methods reminiscent of the unit test classes: * :attr:`cr`: a cursor * :attr:`uid`: user id * :attr:`registry`: access to model objects * :attr:`is_initialization`: True if and only if the database was not initialized before the call to :meth:`open` Example application code::"my_db") admin = session.registry('res_users').browse(, session.uid, 1) (...) session.close() Transaction management is up to user code Upgrade scripts writers should check the version handling properties: * :meth:`db_version` * :meth:`package_version` Instantiation is done by passing the path to OpenERP main configuration file and the path of the buildout directory. Usually, instantiation code is written by the recipe in the body of the executable "OpenERP scripts" it produces. Script writers provide a callable that takes a :class:`.Session` object argument and declare it as a console script entry point in their distribution. End users can reference such entry points in their buildout configurations to have buildout produce the actual executable. See :doc:`/scripts` for details. Upgrade scripts are a special case of that process, in which the entry point is actually provided by the recipe and rewraps a user-level source script. Later versions of the recipe may find a way to pass the whole buildout configuration (recall that this is to be used in a separate process in which the buildout configuration has not been parsed). """ def __init__(self, conffile, buildout_dir, parse_config=True): self.buildout_dir = buildout_dir self.openerp_config_file = conffile self._registry = = None if parse_config: config.parse_config(['-c', conffile])
[docs] def ready(self): return self._registry is not None
[docs] def open(self, db=None, with_demo=False): """Load the database Loading an empty database in OpenERP has the side effect of installing the ``base`` module. Whether to loading demo data or not has therefore to be decided right away. :param db: database name. If not specified, the same cascading of defaults as OpenERP mainstream will be applied: configuration file, psycopg2/lipq defaults. :param with_demo: controls the loading of demo data for all module installations triggered by this call to :meth:`open` and further uses of :meth:`load_modules` on this :class:`Session` instance: * if ``True``, demo data will uniformly be loaded * if ``False``, no demo data will be loaded * if ``None``, demo data will be loaded according to the value of ``without_demo`` in configuration In all cases, the behaviour will stay consistent until the next call of ``open()``, but the implementation does not protect against any race conditions in OpenERP internals. """ if db is None: db = config['db_name'] if not db: db = '' # expected value expected by OpenERP to start defaulting. cnx = openerp.sql_db.db_connect(db) cr = cnx.cursor() self.is_initialization = not(openerp.modules.db.is_initialized(cr)) cr.close() startup.check_root_user() startup.check_postgres_user() openerp.netsvc.init_logger() saved_without_demo = config['without_demo'] if with_demo is None: with_demo = config['without_demo'] config['without_demo'] = not with_demo self.with_demo = with_demo self._registry = openerp.modules.registry.RegistryManager.get( db, update_module=False) config['without_demo'] = saved_without_demo self.init_cursor() self.uid = SUPERUSER_ID self.init_environments()
[docs] def init_environments(self): """Enter the environments context manager, but don't leave it Automatically called by :meth:`open` and registry altering methods. See :class:``openerp.api.Environment`` for explanations about environments. For OpenERP/Odoo versions prior to the new style API merge, this is a no-op. This thread-local ``environments`` is initialized and cleaned with each request in the normal usage of the framework. That's why is is provided as a context manager. Therefore, user code probably needs in some case to clean it to avoid side effects. This can be done by calling :meth:`clean_environments`. """ try: gen_factory = openerp.api.Environment.manage except AttributeError: return self._environments_gen_context = gen_factory().gen
[docs] def clean_environments(self, reinit=True): """Cleans the thread-local environment. See :meth:`init_environments` for more details. This method does nothing if the environments have not been initialized. :param bool reinit: if ``True``, :meth:`init_environments` will be called again after cleaning """ try: gen_context = self._environments_gen_context except AttributeError: return try: except StopIteration: pass else: logger.warn("clean_environments: we had the context manager, but " "it had not been called. This suggest low-leve " "tampering with it that should be more cautious. " "Proceeding with cleansing.") try: except StopIteration: pass else: raise RuntimeError("Called the environments context manager " "twice and it's not finished. " "This is really unexpected.") del self._environments_gen_context if reinit: self.init_environments() # A later version might read that from buildout configuration.
_version_parameter_name = DEFAULT_VERSION_PARAMETER @property
[docs] def version_file_path(self): """Absolute path of the flat file storing the package version. For now this is not configurable, a later version might read it from buildout configuration. """ return os.path.join(self.buildout_dir, DEFAULT_VERSION_FILE)
[docs] def parse_version_string(self, vstring): """Stable method for downstream code needing to instantiate a version. This method returns an appropriate version instance, without any dependency on where to import the class from. Especially useful for applications whose life started before this set of utilities has been used : this helps building an usable default. """ return OpenERPVersion(vstring)
@property def db_version(self): """Settable property for version stored in DB of the whole buildout. This can be thought as the latest version to which the DB has been upgraded to. A simple caching system to avoid querying the DB multiple times is implemented. """ db_version = getattr(self, '_db_version', None) if db_version is not None: return db_version db_version = self.registry('ir.config_parameter').get_param(, self.uid, self._version_parameter_name) if not db_version: # as usual OpenERP thinks its simpler to use False as None # restoring sanity ASAP db_version = None else: db_version = OpenERPVersion(db_version) self._db_version = db_version return db_version @db_version.setter
[docs] def db_version(self, version): self.registry('ir.config_parameter').set_param(, self.uid, self._version_parameter_name, str(version)) self._db_version = OpenERPVersion(version)
[docs] def package_version(self): """Property reading the version file from buildout directory. Comments introduced with a hash are accepted. Only the first significant line is taken into account. """ pkg_version = getattr(self, '_pkg_version', None) if pkg_version is not None: return pkg_version try: with open(self.version_file_path) as f: for line in f: line = line.split('#', 1)[0].strip() if not line: continue self._pkg_version = OpenERPVersion(line) return self._pkg_version except IOError:"No version file could be read, " "package version considered to be None")
[docs] def update_modules_list(self): """Update the list of available OpenERP modules, like the UI allows to. This is necessary prior to install of any new module. """ self.registry('ir.module.module').update_list(, self.uid)
[docs] def init_cursor(self): db = getattr(self._registry, 'db', None) if db is None: # current trunk (future v8) = self._registry.cursor() else: # In OpenERP < 8, Registry.cursor() object is # a context manager providing auto closing, # but we don't want to control the whole lifespan # of the cursor. = db.cursor()
[docs] def registry(self, model): """Lookup model by name and return a ready-to-work instance.""" return self._registry.get(model)
[docs] def rollback(self): self.clean_environments()
[docs] def close(self): """Close the cursor and forget about the current database. The session is thus ready to open another database. """ dbname = self.clean_environments() openerp.modules.registry.RegistryManager.delete(dbname)
[docs] def update_modules(self, modules, db=None): """Update the prescribed modules in the database. :param db: Database name. If not specified, it is assumed to have already been opened with :meth:`open`, e.g, for a prior read of :meth:`db_version`. If it is specified, then the session in particular opens that db and will use it afterwards whether another one was already opened or not. :param modules: any iterable of module names. Not installed modules will be ignored The special name ``'all'`` triggers the update of all installed modules. """ if db is None: if is None: raise ValueError("update_modules needs either the session to " "be opened or an explicit database name") db = if is not None: self.close() for module in modules: config['update'][module] = 1 self._registry = openerp.modules.registry.RegistryManager.get( db, update_module=True) config['update'].clear() self.init_cursor() self.clean_environments()
[docs] def install_modules(self, modules, db=None, update_modules_list=True, open_with_demo=False): """Install the modules in the database. Has the side effect of closing the current cursor, committing if and only if the list of modules is updated. Demo data loading is handled consistently with the decision taken by :meth:`open`. :param db: Database name. If not specified, it is assumed to have already been opened with :meth:`open`, e.g, for a prior read of :meth:`db_version`. If it is specified, then the session in particular opens that db and will use it afterwards whether another one was already opened or not. :param modules: any iterable of module names. :param update_modules_list: if True, will update the module lists *and commit* before the install begins. :param open_with_demo: if ``db`` is not None, will be passed to :meth:`open`. """ already_open = is not None if db is None: if not already_open: raise ValueError("install_modules needs either the session to " "be opened or an explicit database name") db = elif update_modules_list and not ( already_open and == db):, with_demo=open_with_demo) if update_modules_list: self.update_modules_list() if is not None: self.close() saved_without_demo = config['without_demo'] # with update_modules_list=False, an explicitely named DB would not # have gone through open() yet. config['without_demo'] = not getattr(self, 'with_demo', open_with_demo) for module in modules: config['init'][module] = 1 self._registry = openerp.modules.registry.RegistryManager.get( db, update_module=True, force_demo=self.with_demo) config['init'].clear() config['without_demo'] = saved_without_demo self.init_cursor() self.clean_environments()
[docs] def handle_command_line_options(self, to_handle): """Handle prescribed command line options and eat them. Anything before first occurrence of ``--`` on the command-line is taken into account and removed from ``sys.argv``. Help messages: If -h or --help is specified and -- is not, the help for the wrapper will be printed, and the -h/--help option kept in sys.argv. If -h or --help is specified before --, the help for this wrapper will be printed and options after -- will be kept in sys.argv. if -h or --help is specified after --, it will be ignored at this stage, and kept in sys.argv (in most cases triggering help print for the wrapped script). """ parser = OptionParser( usage="%(prog)s [OpenERP options] -- other arguments", description="This is a script rewrapped by OpenERP buildout " "recipe to add OpenERP-related options on the command " "line prior to other arguments.") if '-d' in to_handle: parser.add_option('-d', '--db-name', help="Name of the database to work on. " "If not specified, the database from " "configuration files will be used") try: sep = sys.argv.index('--') except ValueError: if '-h' in sys.argv or '--help' in sys.argv: # in case of call myscript -h --, only the wrapper help # will be printed parser.epilog = ("Help message from the wrapped script, " "if any, will follow.") parser.print_help() print return our_argv = [] sep = None else: our_argv = sys.argv[1:sep] options, args = parser.parse_args(our_argv) if sep is not None: del sys.argv[1:sep+1] if '-d' in to_handle: if options.db_name:"Opening database %r", options.db_name) else:"No database specified, using the one specified " "in buildout configuration.")