491 lines
19 KiB
Python
491 lines
19 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
import base64
|
||
|
import json
|
||
|
import logging
|
||
|
import os
|
||
|
import shutil
|
||
|
import subprocess
|
||
|
import tempfile
|
||
|
import threading
|
||
|
import traceback
|
||
|
from xml.etree import ElementTree as ET
|
||
|
import zipfile
|
||
|
|
||
|
from psycopg2 import sql
|
||
|
from pytz import country_timezones
|
||
|
from functools import wraps
|
||
|
from contextlib import closing
|
||
|
from decorator import decorator
|
||
|
|
||
|
import psycopg2
|
||
|
|
||
|
import odoo
|
||
|
from odoo import SUPERUSER_ID
|
||
|
from odoo.exceptions import AccessDenied
|
||
|
import odoo.release
|
||
|
import odoo.sql_db
|
||
|
import odoo.tools
|
||
|
from odoo.sql_db import db_connect
|
||
|
from odoo.release import version_info
|
||
|
from odoo.tools import find_pg_tool, exec_pg_environ
|
||
|
|
||
|
_logger = logging.getLogger(__name__)
|
||
|
|
||
|
class DatabaseExists(Warning):
|
||
|
pass
|
||
|
|
||
|
|
||
|
def check_db_management_enabled(method):
|
||
|
def if_db_mgt_enabled(method, self, *args, **kwargs):
|
||
|
if not odoo.tools.config['list_db']:
|
||
|
_logger.error('Database management functions blocked, admin disabled database listing')
|
||
|
raise AccessDenied()
|
||
|
return method(self, *args, **kwargs)
|
||
|
return decorator(if_db_mgt_enabled, method)
|
||
|
|
||
|
#----------------------------------------------------------
|
||
|
# Master password required
|
||
|
#----------------------------------------------------------
|
||
|
|
||
|
def check_super(passwd):
|
||
|
if passwd and odoo.tools.config.verify_admin_password(passwd):
|
||
|
return True
|
||
|
raise odoo.exceptions.AccessDenied()
|
||
|
|
||
|
# This should be moved to odoo.modules.db, along side initialize().
|
||
|
def _initialize_db(id, db_name, demo, lang, user_password, login='admin', country_code=None, phone=None):
|
||
|
try:
|
||
|
db = odoo.sql_db.db_connect(db_name)
|
||
|
with closing(db.cursor()) as cr:
|
||
|
# TODO this should be removed as it is done by Registry.new().
|
||
|
odoo.modules.db.initialize(cr)
|
||
|
odoo.tools.config['load_language'] = lang
|
||
|
cr.commit()
|
||
|
|
||
|
registry = odoo.modules.registry.Registry.new(db_name, demo, None, update_module=True)
|
||
|
|
||
|
with closing(registry.cursor()) as cr:
|
||
|
env = odoo.api.Environment(cr, SUPERUSER_ID, {})
|
||
|
|
||
|
if lang:
|
||
|
modules = env['ir.module.module'].search([('state', '=', 'installed')])
|
||
|
modules._update_translations(lang)
|
||
|
|
||
|
if country_code:
|
||
|
country = env['res.country'].search([('code', 'ilike', country_code)])[0]
|
||
|
env['res.company'].browse(1).write({'country_id': country_code and country.id, 'currency_id': country_code and country.currency_id.id})
|
||
|
if len(country_timezones.get(country_code, [])) == 1:
|
||
|
users = env['res.users'].search([])
|
||
|
users.write({'tz': country_timezones[country_code][0]})
|
||
|
if phone:
|
||
|
env['res.company'].browse(1).write({'phone': phone})
|
||
|
if '@' in login:
|
||
|
env['res.company'].browse(1).write({'email': login})
|
||
|
|
||
|
# update admin's password and lang and login
|
||
|
values = {'password': user_password, 'lang': lang}
|
||
|
if login:
|
||
|
values['login'] = login
|
||
|
emails = odoo.tools.email_split(login)
|
||
|
if emails:
|
||
|
values['email'] = emails[0]
|
||
|
env.ref('base.user_admin').write(values)
|
||
|
|
||
|
cr.execute('SELECT login, password FROM res_users ORDER BY login')
|
||
|
cr.commit()
|
||
|
except Exception as e:
|
||
|
_logger.exception('CREATE DATABASE failed:')
|
||
|
|
||
|
def _create_empty_database(name):
|
||
|
db = odoo.sql_db.db_connect('postgres')
|
||
|
with closing(db.cursor()) as cr:
|
||
|
chosen_template = odoo.tools.config['db_template']
|
||
|
cr.execute("SELECT datname FROM pg_database WHERE datname = %s",
|
||
|
(name,), log_exceptions=False)
|
||
|
if cr.fetchall():
|
||
|
raise DatabaseExists("database %r already exists!" % (name,))
|
||
|
else:
|
||
|
# database-altering operations cannot be executed inside a transaction
|
||
|
cr.rollback()
|
||
|
cr._cnx.autocommit = True
|
||
|
|
||
|
# 'C' collate is only safe with template0, but provides more useful indexes
|
||
|
collate = sql.SQL("LC_COLLATE 'C'" if chosen_template == 'template0' else "")
|
||
|
cr.execute(
|
||
|
sql.SQL("CREATE DATABASE {} ENCODING 'unicode' {} TEMPLATE {}").format(
|
||
|
sql.Identifier(name), collate, sql.Identifier(chosen_template)
|
||
|
))
|
||
|
|
||
|
# TODO: add --extension=trigram,unaccent
|
||
|
try:
|
||
|
db = odoo.sql_db.db_connect(name)
|
||
|
with db.cursor() as cr:
|
||
|
cr.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm")
|
||
|
if odoo.tools.config['unaccent']:
|
||
|
cr.execute("CREATE EXTENSION IF NOT EXISTS unaccent")
|
||
|
# From PostgreSQL's point of view, making 'unaccent' immutable is incorrect
|
||
|
# because it depends on external data - see
|
||
|
# https://www.postgresql.org/message-id/flat/201012021544.oB2FiTn1041521@wwwmaster.postgresql.org#201012021544.oB2FiTn1041521@wwwmaster.postgresql.org
|
||
|
# But in the case of Odoo, we consider that those data don't
|
||
|
# change in the lifetime of a database. If they do change, all
|
||
|
# indexes created with this function become corrupted!
|
||
|
cr.execute("ALTER FUNCTION unaccent(text) IMMUTABLE")
|
||
|
except psycopg2.Error as e:
|
||
|
_logger.warning("Unable to create PostgreSQL extensions : %s", e)
|
||
|
|
||
|
# restore legacy behaviour on pg15+
|
||
|
try:
|
||
|
db = odoo.sql_db.db_connect(name)
|
||
|
with db.cursor() as cr:
|
||
|
cr.execute("GRANT CREATE ON SCHEMA PUBLIC TO PUBLIC")
|
||
|
except psycopg2.Error as e:
|
||
|
_logger.warning("Unable to make public schema public-accessible: %s", e)
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_create_database(db_name, demo, lang, user_password='admin', login='admin', country_code=None, phone=None):
|
||
|
""" Similar to exp_create but blocking."""
|
||
|
_logger.info('Create database `%s`.', db_name)
|
||
|
_create_empty_database(db_name)
|
||
|
_initialize_db(id, db_name, demo, lang, user_password, login, country_code, phone)
|
||
|
return True
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_duplicate_database(db_original_name, db_name, neutralize_database=False):
|
||
|
_logger.info('Duplicate database `%s` to `%s`.', db_original_name, db_name)
|
||
|
odoo.sql_db.close_db(db_original_name)
|
||
|
db = odoo.sql_db.db_connect('postgres')
|
||
|
with closing(db.cursor()) as cr:
|
||
|
# database-altering operations cannot be executed inside a transaction
|
||
|
cr._cnx.autocommit = True
|
||
|
_drop_conn(cr, db_original_name)
|
||
|
cr.execute(sql.SQL("CREATE DATABASE {} ENCODING 'unicode' TEMPLATE {}").format(
|
||
|
sql.Identifier(db_name),
|
||
|
sql.Identifier(db_original_name)
|
||
|
))
|
||
|
|
||
|
registry = odoo.modules.registry.Registry.new(db_name)
|
||
|
with registry.cursor() as cr:
|
||
|
# if it's a copy of a database, force generation of a new dbuuid
|
||
|
env = odoo.api.Environment(cr, SUPERUSER_ID, {})
|
||
|
env['ir.config_parameter'].init(force=True)
|
||
|
if neutralize_database:
|
||
|
odoo.modules.neutralize.neutralize_database(cr)
|
||
|
|
||
|
from_fs = odoo.tools.config.filestore(db_original_name)
|
||
|
to_fs = odoo.tools.config.filestore(db_name)
|
||
|
if os.path.exists(from_fs) and not os.path.exists(to_fs):
|
||
|
shutil.copytree(from_fs, to_fs)
|
||
|
return True
|
||
|
|
||
|
def _drop_conn(cr, db_name):
|
||
|
# Try to terminate all other connections that might prevent
|
||
|
# dropping the database
|
||
|
try:
|
||
|
# PostgreSQL 9.2 renamed pg_stat_activity.procpid to pid:
|
||
|
# http://www.postgresql.org/docs/9.2/static/release-9-2.html#AEN110389
|
||
|
pid_col = 'pid' if cr._cnx.server_version >= 90200 else 'procpid'
|
||
|
|
||
|
cr.execute("""SELECT pg_terminate_backend(%(pid_col)s)
|
||
|
FROM pg_stat_activity
|
||
|
WHERE datname = %%s AND
|
||
|
%(pid_col)s != pg_backend_pid()""" % {'pid_col': pid_col},
|
||
|
(db_name,))
|
||
|
except Exception:
|
||
|
pass
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_drop(db_name):
|
||
|
if db_name not in list_dbs(True):
|
||
|
return False
|
||
|
odoo.modules.registry.Registry.delete(db_name)
|
||
|
odoo.sql_db.close_db(db_name)
|
||
|
|
||
|
db = odoo.sql_db.db_connect('postgres')
|
||
|
with closing(db.cursor()) as cr:
|
||
|
# database-altering operations cannot be executed inside a transaction
|
||
|
cr._cnx.autocommit = True
|
||
|
_drop_conn(cr, db_name)
|
||
|
|
||
|
try:
|
||
|
cr.execute(sql.SQL('DROP DATABASE {}').format(sql.Identifier(db_name)))
|
||
|
except Exception as e:
|
||
|
_logger.info('DROP DB: %s failed:\n%s', db_name, e)
|
||
|
raise Exception("Couldn't drop database %s: %s" % (db_name, e))
|
||
|
else:
|
||
|
_logger.info('DROP DB: %s', db_name)
|
||
|
|
||
|
fs = odoo.tools.config.filestore(db_name)
|
||
|
if os.path.exists(fs):
|
||
|
shutil.rmtree(fs)
|
||
|
return True
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_dump(db_name, format):
|
||
|
with tempfile.TemporaryFile(mode='w+b') as t:
|
||
|
dump_db(db_name, t, format)
|
||
|
t.seek(0)
|
||
|
return base64.b64encode(t.read()).decode()
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def dump_db_manifest(cr):
|
||
|
pg_version = "%d.%d" % divmod(cr._obj.connection.server_version / 100, 100)
|
||
|
cr.execute("SELECT name, latest_version FROM ir_module_module WHERE state = 'installed'")
|
||
|
modules = dict(cr.fetchall())
|
||
|
manifest = {
|
||
|
'odoo_dump': '1',
|
||
|
'db_name': cr.dbname,
|
||
|
'version': odoo.release.version,
|
||
|
'version_info': odoo.release.version_info,
|
||
|
'major_version': odoo.release.major_version,
|
||
|
'pg_version': pg_version,
|
||
|
'modules': modules,
|
||
|
}
|
||
|
return manifest
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def dump_db(db_name, stream, backup_format='zip'):
|
||
|
"""Dump database `db` into file-like object `stream` if stream is None
|
||
|
return a file object with the dump """
|
||
|
|
||
|
_logger.info('DUMP DB: %s format %s', db_name, backup_format)
|
||
|
|
||
|
cmd = [find_pg_tool('pg_dump'), '--no-owner', db_name]
|
||
|
env = exec_pg_environ()
|
||
|
|
||
|
if backup_format == 'zip':
|
||
|
with tempfile.TemporaryDirectory() as dump_dir:
|
||
|
filestore = odoo.tools.config.filestore(db_name)
|
||
|
if os.path.exists(filestore):
|
||
|
shutil.copytree(filestore, os.path.join(dump_dir, 'filestore'))
|
||
|
with open(os.path.join(dump_dir, 'manifest.json'), 'w') as fh:
|
||
|
db = odoo.sql_db.db_connect(db_name)
|
||
|
with db.cursor() as cr:
|
||
|
json.dump(dump_db_manifest(cr), fh, indent=4)
|
||
|
cmd.insert(-1, '--file=' + os.path.join(dump_dir, 'dump.sql'))
|
||
|
subprocess.run(cmd, env=env, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT, check=True)
|
||
|
if stream:
|
||
|
odoo.tools.osutil.zip_dir(dump_dir, stream, include_dir=False, fnct_sort=lambda file_name: file_name != 'dump.sql')
|
||
|
else:
|
||
|
t=tempfile.TemporaryFile()
|
||
|
odoo.tools.osutil.zip_dir(dump_dir, t, include_dir=False, fnct_sort=lambda file_name: file_name != 'dump.sql')
|
||
|
t.seek(0)
|
||
|
return t
|
||
|
else:
|
||
|
cmd.insert(-1, '--format=c')
|
||
|
stdout = subprocess.Popen(cmd, env=env, stdin=subprocess.DEVNULL, stdout=subprocess.PIPE).stdout
|
||
|
if stream:
|
||
|
shutil.copyfileobj(stdout, stream)
|
||
|
else:
|
||
|
return stdout
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_restore(db_name, data, copy=False):
|
||
|
def chunks(d, n=8192):
|
||
|
for i in range(0, len(d), n):
|
||
|
yield d[i:i+n]
|
||
|
data_file = tempfile.NamedTemporaryFile(delete=False)
|
||
|
try:
|
||
|
for chunk in chunks(data):
|
||
|
data_file.write(base64.b64decode(chunk))
|
||
|
data_file.close()
|
||
|
restore_db(db_name, data_file.name, copy=copy)
|
||
|
finally:
|
||
|
os.unlink(data_file.name)
|
||
|
return True
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def restore_db(db, dump_file, copy=False, neutralize_database=False):
|
||
|
assert isinstance(db, str)
|
||
|
if exp_db_exist(db):
|
||
|
_logger.warning('RESTORE DB: %s already exists', db)
|
||
|
raise Exception("Database already exists")
|
||
|
|
||
|
_logger.info('RESTORING DB: %s', db)
|
||
|
_create_empty_database(db)
|
||
|
|
||
|
filestore_path = None
|
||
|
with tempfile.TemporaryDirectory() as dump_dir:
|
||
|
if zipfile.is_zipfile(dump_file):
|
||
|
# v8 format
|
||
|
with zipfile.ZipFile(dump_file, 'r') as z:
|
||
|
# only extract known members!
|
||
|
filestore = [m for m in z.namelist() if m.startswith('filestore/')]
|
||
|
z.extractall(dump_dir, ['dump.sql'] + filestore)
|
||
|
|
||
|
if filestore:
|
||
|
filestore_path = os.path.join(dump_dir, 'filestore')
|
||
|
|
||
|
pg_cmd = 'psql'
|
||
|
pg_args = ['-q', '-f', os.path.join(dump_dir, 'dump.sql')]
|
||
|
|
||
|
else:
|
||
|
# <= 7.0 format (raw pg_dump output)
|
||
|
pg_cmd = 'pg_restore'
|
||
|
pg_args = ['--no-owner', dump_file]
|
||
|
|
||
|
r = subprocess.run(
|
||
|
[find_pg_tool(pg_cmd), '--dbname=' + db, *pg_args],
|
||
|
env=exec_pg_environ(),
|
||
|
stdout=subprocess.DEVNULL,
|
||
|
stderr=subprocess.STDOUT,
|
||
|
)
|
||
|
if r.returncode != 0:
|
||
|
raise Exception("Couldn't restore database")
|
||
|
|
||
|
registry = odoo.modules.registry.Registry.new(db)
|
||
|
with registry.cursor() as cr:
|
||
|
env = odoo.api.Environment(cr, SUPERUSER_ID, {})
|
||
|
if copy:
|
||
|
# if it's a copy of a database, force generation of a new dbuuid
|
||
|
env['ir.config_parameter'].init(force=True)
|
||
|
if neutralize_database:
|
||
|
odoo.modules.neutralize.neutralize_database(cr)
|
||
|
|
||
|
if filestore_path:
|
||
|
filestore_dest = env['ir.attachment']._filestore()
|
||
|
shutil.move(filestore_path, filestore_dest)
|
||
|
|
||
|
_logger.info('RESTORE DB: %s', db)
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_rename(old_name, new_name):
|
||
|
odoo.modules.registry.Registry.delete(old_name)
|
||
|
odoo.sql_db.close_db(old_name)
|
||
|
|
||
|
db = odoo.sql_db.db_connect('postgres')
|
||
|
with closing(db.cursor()) as cr:
|
||
|
# database-altering operations cannot be executed inside a transaction
|
||
|
cr._cnx.autocommit = True
|
||
|
_drop_conn(cr, old_name)
|
||
|
try:
|
||
|
cr.execute(sql.SQL('ALTER DATABASE {} RENAME TO {}').format(sql.Identifier(old_name), sql.Identifier(new_name)))
|
||
|
_logger.info('RENAME DB: %s -> %s', old_name, new_name)
|
||
|
except Exception as e:
|
||
|
_logger.info('RENAME DB: %s -> %s failed:\n%s', old_name, new_name, e)
|
||
|
raise Exception("Couldn't rename database %s to %s: %s" % (old_name, new_name, e))
|
||
|
|
||
|
old_fs = odoo.tools.config.filestore(old_name)
|
||
|
new_fs = odoo.tools.config.filestore(new_name)
|
||
|
if os.path.exists(old_fs) and not os.path.exists(new_fs):
|
||
|
shutil.move(old_fs, new_fs)
|
||
|
return True
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_change_admin_password(new_password):
|
||
|
odoo.tools.config.set_admin_password(new_password)
|
||
|
odoo.tools.config.save(['admin_passwd'])
|
||
|
return True
|
||
|
|
||
|
@check_db_management_enabled
|
||
|
def exp_migrate_databases(databases):
|
||
|
for db in databases:
|
||
|
_logger.info('migrate database %s', db)
|
||
|
odoo.tools.config['update']['base'] = True
|
||
|
odoo.modules.registry.Registry.new(db, force_demo=False, update_module=True)
|
||
|
return True
|
||
|
|
||
|
#----------------------------------------------------------
|
||
|
# No master password required
|
||
|
#----------------------------------------------------------
|
||
|
|
||
|
@odoo.tools.mute_logger('odoo.sql_db')
|
||
|
def exp_db_exist(db_name):
|
||
|
## Not True: in fact, check if connection to database is possible. The database may exists
|
||
|
try:
|
||
|
db = odoo.sql_db.db_connect(db_name)
|
||
|
with db.cursor():
|
||
|
return True
|
||
|
except Exception:
|
||
|
return False
|
||
|
|
||
|
def list_dbs(force=False):
|
||
|
if not odoo.tools.config['list_db'] and not force:
|
||
|
raise odoo.exceptions.AccessDenied()
|
||
|
|
||
|
if not odoo.tools.config['dbfilter'] and odoo.tools.config['db_name']:
|
||
|
# In case --db-filter is not provided and --database is passed, Odoo will not
|
||
|
# fetch the list of databases available on the postgres server and instead will
|
||
|
# use the value of --database as comma seperated list of exposed databases.
|
||
|
res = sorted(db.strip() for db in odoo.tools.config['db_name'].split(','))
|
||
|
return res
|
||
|
|
||
|
chosen_template = odoo.tools.config['db_template']
|
||
|
templates_list = tuple(set(['postgres', chosen_template]))
|
||
|
db = odoo.sql_db.db_connect('postgres')
|
||
|
with closing(db.cursor()) as cr:
|
||
|
try:
|
||
|
cr.execute("select datname from pg_database where datdba=(select usesysid from pg_user where usename=current_user) and not datistemplate and datallowconn and datname not in %s order by datname", (templates_list,))
|
||
|
res = [odoo.tools.ustr(name) for (name,) in cr.fetchall()]
|
||
|
except Exception:
|
||
|
_logger.exception('Listing databases failed:')
|
||
|
res = []
|
||
|
return res
|
||
|
|
||
|
def list_db_incompatible(databases):
|
||
|
""""Check a list of databases if they are compatible with this version of Odoo
|
||
|
|
||
|
:param databases: A list of existing Postgresql databases
|
||
|
:return: A list of databases that are incompatible
|
||
|
"""
|
||
|
incompatible_databases = []
|
||
|
server_version = '.'.join(str(v) for v in version_info[:2])
|
||
|
for database_name in databases:
|
||
|
with closing(db_connect(database_name).cursor()) as cr:
|
||
|
if odoo.tools.table_exists(cr, 'ir_module_module'):
|
||
|
cr.execute("SELECT latest_version FROM ir_module_module WHERE name=%s", ('base',))
|
||
|
base_version = cr.fetchone()
|
||
|
if not base_version or not base_version[0]:
|
||
|
incompatible_databases.append(database_name)
|
||
|
else:
|
||
|
# e.g. 10.saas~15
|
||
|
local_version = '.'.join(base_version[0].split('.')[:2])
|
||
|
if local_version != server_version:
|
||
|
incompatible_databases.append(database_name)
|
||
|
else:
|
||
|
incompatible_databases.append(database_name)
|
||
|
for database_name in incompatible_databases:
|
||
|
# release connection
|
||
|
odoo.sql_db.close_db(database_name)
|
||
|
return incompatible_databases
|
||
|
|
||
|
|
||
|
def exp_list(document=False):
|
||
|
if not odoo.tools.config['list_db']:
|
||
|
raise odoo.exceptions.AccessDenied()
|
||
|
return list_dbs()
|
||
|
|
||
|
def exp_list_lang():
|
||
|
return odoo.tools.scan_languages()
|
||
|
|
||
|
def exp_list_countries():
|
||
|
list_countries = []
|
||
|
root = ET.parse(os.path.join(odoo.tools.config['root_path'], 'addons/base/data/res_country_data.xml')).getroot()
|
||
|
for country in root.find('data').findall('record[@model="res.country"]'):
|
||
|
name = country.find('field[@name="name"]').text
|
||
|
code = country.find('field[@name="code"]').text
|
||
|
list_countries.append([code, name])
|
||
|
return sorted(list_countries, key=lambda c: c[1])
|
||
|
|
||
|
def exp_server_version():
|
||
|
""" Return the version of the server
|
||
|
Used by the client to verify the compatibility with its own version
|
||
|
"""
|
||
|
return odoo.release.version
|
||
|
|
||
|
#----------------------------------------------------------
|
||
|
# db service dispatch
|
||
|
#----------------------------------------------------------
|
||
|
|
||
|
def dispatch(method, params):
|
||
|
g = globals()
|
||
|
exp_method_name = 'exp_' + method
|
||
|
if method in ['db_exist', 'list', 'list_lang', 'server_version']:
|
||
|
return g[exp_method_name](*params)
|
||
|
elif exp_method_name in g:
|
||
|
passwd = params[0]
|
||
|
params = params[1:]
|
||
|
check_super(passwd)
|
||
|
return g[exp_method_name](*params)
|
||
|
else:
|
||
|
raise KeyError("Method not found: %s" % method)
|