1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
|
# This file is part of Tryton. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
from proteus import Model, Wizard
from proteus import config as pconfig
from .test_tryton import backup_db_cache, drop_create, restore_db_cache
__all__ = ['activate_modules', 'set_user']
def activate_modules(modules):
if isinstance(modules, str):
modules = [modules]
cache_name = '-'.join(modules)
if restore_db_cache(cache_name):
return _get_config()
drop_create()
cfg = _get_config()
Module = Model.get('ir.module')
records = Module.find([
('name', 'in', modules),
])
assert len(records) == len(modules)
Module.click(records, 'activate')
Wizard('ir.module.activate_upgrade').execute('upgrade')
backup_db_cache(cache_name)
return cfg
def _get_config():
cfg = pconfig.set_trytond()
cfg.pool.test = True
return cfg
def set_user(user, config=None):
if not config:
config = pconfig.get_config()
User = Model.get('res.user', config=config)
config.user = int(user)
config._context = User.get_preferences(True, {})
|