bin/pooler.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
   6 #    $Id$
   7 #
   8 #    This program is free software: you can redistribute it and/or modify
   9 #    it under the terms of the GNU General Public License as published by
  10 #    the Free Software Foundation, either version 3 of the License, or
  11 #    (at your option) any later version.
  12 #
  13 #    This program is distributed in the hope that it will be useful,
  14 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #    GNU General Public License for more details.
  17 #
  18 #    You should have received a copy of the GNU General Public License
  19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21 ##############################################################################
  22
  23 pool_dic = {}
  24
  25 def get_db_and_pool(db_name, force_demo=False, status=None, update_module=False):
  26     if not status:
  27         status={}
  28
  29     db = get_db_only(db_name)
  30
  31     if db_name in pool_dic:
  32         pool = pool_dic[db_name]
  33     else:
  34         import addons
  35         import osv.osv
  36         pool = osv.osv.osv_pool()
  37         pool_dic[db_name] = pool
  38
  39         try:
  40             addons.load_modules(db, force_demo, status, update_module)
  41         except Exception, e:
  42             del pool_dic[db_name]
  43             raise
  44
  45         cr = db.cursor()
  46         try:
  47             pool.init_set(cr, False)
  48             cr.commit()
  49         finally:
  50             cr.close()
  51
  52         import report
  53         report.interface.register_all(db)
  54         pool.get('ir.cron')._poolJobs(db.dbname)
  55     return db, pool
  56
  57
  58 def restart_pool(db_name, force_demo=False, status=None, update_module=False):
  59     if db_name in pool_dic:
  60         pool_dic[db_name].get('ir.cron').cancel(db_name)
  61         del pool_dic[db_name]
  62     return get_db_and_pool(db_name, force_demo, status, update_module=update_module)
  63
  64
  65 def get_db_only(db_name):
  66     # ATTENTION:
  67     # do not put this import outside this function
  68     # sql_db must not be loaded before the logger is initialized.
  69     # sql_db import psycopg2.tool which create a default logger if there is not.
  70     # this resulting of having the logs outputed twice...
  71     import sql_db
  72     db = sql_db.db_connect(db_name)
  73     return db
  74
  75
  76 def get_db(db_name):
  77     return get_db_and_pool(db_name)[0]
  78
  79
  80 def get_pool(db_name, force_demo=False, status=None, update_module=False):
  81     pool = get_db_and_pool(db_name, force_demo, status, update_module)[1]
  82     return pool
  83
  84 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
  85