bin/pooler.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 # Copyright (c) 2004-2008 Tiny SPRL (http://tiny.be) All Rights Reserved.
   5 #
   6 # $Id$
   7 #
   8 # WARNING: This program as such is intended to be used by professional
   9 # programmers who take the whole responsability of assessing all potential
  10 # consequences resulting from its eventual inadequacies and bugs
  11 # End users who are looking for a ready-to-use solution with commercial
  12 # garantees and support are strongly adviced to contract a Free Software
  13 # Service Company
  14 #
  15 # This program is Free Software; you can redistribute it and/or
  16 # modify it under the terms of the GNU General Public License
  17 # as published by the Free Software Foundation; either version 2
  18 # of the License, or (at your option) any later version.
  19 #
  20 # This program is distributed in the hope that it will be useful,
  21 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 # GNU General Public License for more details.
  24 #
  25 # You should have received a copy of the GNU General Public License
  26 # along with this program; if not, write to the Free Software
  27 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  28 ###############################################################################
  29
  30 import sql_db
  31 import osv.osv
  32 import tools
  33 import addons
  34 import netsvc
  35
  36 db_dic = {}
  37 pool_dic = {}
  38
  39
  40 def get_db_and_pool(db_name, force_demo=False, status=None, update_module=False):
  41     if not status:
  42         status={}
  43     if db_name in db_dic:
  44         db = db_dic[db_name]
  45     else:
  46         logger = netsvc.Logger()
  47         logger.notifyChannel('pooler', netsvc.LOG_INFO, 'Connecting to %s' % (db_name.lower()))
  48         db = sql_db.db_connect(db_name)
  49         db_dic[db_name] = db
  50
  51     if db_name in pool_dic:
  52         pool = pool_dic[db_name]
  53     else:
  54         pool = osv.osv.osv_pool()
  55         pool_dic[db_name] = pool
  56         addons.load_modules(db, force_demo, status, update_module)
  57         cr = db.cursor()
  58         pool.init_set(cr, False)
  59         cr.commit()
  60         cr.close()
  61
  62         if not update_module:
  63             import report
  64             report.interface.register_all(db)
  65             pool.get('ir.cron')._poolJobs(db.dbname)
  66     return db, pool
  67
  68
  69 def restart_pool(db_name, force_demo=False, update_module=False):
  70 #   del db_dic[db_name]
  71     del pool_dic[db_name]
  72     return get_db_and_pool(db_name, force_demo, update_module=update_module)
  73
  74
  75 def close_db(db_name):
  76     if db_name in db_dic:
  77         db_dic[db_name].truedb.close()
  78         del db_dic[db_name]
  79     if db_name in pool_dic:
  80         del pool_dic[db_name]
  81
  82
  83 def get_db_only(db_name):
  84     if db_name in db_dic:
  85         db = db_dic[db_name]
  86     else:
  87         db = sql_db.db_connect(db_name)
  88         db_dic[db_name] = db
  89     return db
  90
  91
  92 def get_db(db_name):
  93 #   print "get_db", db_name
  94     return get_db_and_pool(db_name)[0]
  95
  96
  97 def get_pool(db_name, force_demo=False, status=None, update_module=False):
  98 #   print "get_pool", db_name
  99     pool = get_db_and_pool(db_name, force_demo, status, update_module)[1]
 100 #   addons.load_modules(db_name, False)
 101 #   if not pool.obj_list():
 102 #       pool.instanciate()
 103 #   print "pool", pool
 104     return pool
 105 #   return get_db_and_pool(db_name)[1]
 106
 107
 108 def init():
 109     global db
 110 #   db = get_db_only(tools.config['db_name'])
 111     sql_db.init()
 112
 113
 114 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 115