tryton/trytond/web_service/db.py

# -*- coding: utf-8 -*-
#This file is part of Tryton.  The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
import logging
import threading
from trytond.netsvc import Service
from trytond import security
from trytond.backend import Database
from trytond import pooler
from trytond import tools
import base64
import os
import sha


class DB(Service):

    def __init__(self, name="db"):
        super(DB, self).__init__(name)
        self.join_group("web-service")
        self.export_method(self.create)
        self.export_method(self.drop)
        self.export_method(self.dump)
        self.export_method(self.restore)
        self.export_method(self.list)
        self.export_method(self.list_lang)
        self.export_method(self.db_exist)
        self.export_method(self.change_admin_password)

    def create(self, password, db_name, lang, admin_password):
        security.check_super(password)
        res = False
        logger = logging.getLogger('web-service')

        database = Database().connect()
        cursor = database.cursor(autocommit=True)
        try:
            try:
                database.create(cursor, db_name)
                cursor.commit()
                cursor.close()
                database.close()

                database = Database(db_name).connect()
                cursor = database.cursor()
                database.init(cursor)
                cursor.commit()
                cursor.close()
                database.close()

                cursor = None
                database = None

                pool = pooler.get_pool(db_name, update_module=True, lang=[lang])
                database = Database(db_name).connect()
                cursor = database.cursor()

                #XXX replace with model write
                if lang != 'en_US':
                    cursor.execute('UPDATE ir_lang ' \
                            'SET translatable = True ' \
                            'WHERE code = %s', (lang,))
                cursor.execute('UPDATE res_user ' \
                        'SET language = ' \
                            '(SELECT id FROM ir_lang WHERE code = %s LIMIT 1) '\
                        'WHERE login <> \'root\'', (lang,))
                cursor.execute('UPDATE res_user ' \
                        'SET password = %s ' \
                        'WHERE login = \'admin\'',
                        (sha.new(admin_password).hexdigest(),))
                module_obj = pool.get('ir.module.module')
                if module_obj:
                    module_obj.update_list(cursor, 0)
                cursor.commit()
                res = True
            except:
                logger.error('CREATE DB: %s failed' % (db_name,))
                import traceback, sys
                tb_s = reduce(lambda x, y: x+y,
                        traceback.format_exception(*sys.exc_info()))
                logger.error('Exception in call: \n' + tb_s)
                raise
            else:
                logger.info('CREATE DB: %s' % (db_name,))
        finally:
            if cursor:
                cursor.close()
            if database:
                database.close()
        return res

    def drop(self, password, db_name):
        security.check_super(password)
        pooler.close_db(db_name)
        logger = logging.getLogger('web-service')

        database = Database().connect()
        cursor = database.cursor(autocommit=True)
        try:
            try:
                database.drop(cursor, db_name)
                cursor.commit()
            except:
                logger.error('DROP DB: %s failed' % (db_name,))
                import traceback, sys
                tb_s = reduce(lambda x, y: x+y,
                        traceback.format_exception(*sys.exc_info()))
                logger.error('Exception in call: \n' + tb_s)
                raise
            else:
                logger.info('DROP DB: %s' % (db_name))
        finally:
            cursor.close()
            database.close()
        return True

    def dump(self, password, db_name):
        logger = logging.getLogger('web-service')
        security.check_super(password)
        data = Database.dump(db_name)
        logger.info('DUMP DB: %s' % (db_name))
        return base64.encodestring(data)

    def restore(self, password, db_name, data):
        logger = logging.getLogger('web-service')
        security.check_super(password)
        if self.db_exist(db_name):
            raise Exception("Database already exists!")
        data = base64.decodestring(data)
        Database.restore(db_name, data)
        logger.info('RESTORE DB: %s' % (db_name))
        return True

    def db_exist(self, db_name):
        try:
            database = Database(db_name).connect()
            cursor = database.cursor()
            cursor.close()
            database.close()
            return True
        except:
            return False

    def list(self):
        database = Database().connect()
        try:
            cursor = database.cursor()
            res = database.list(cursor)
            cursor.close()
            database.close()
        except:
            res = []
        return res

    def change_admin_password(self, old_password, new_password):
        security.check_super(old_password)
        tools.CONFIG['admin_passwd'] = new_password
        tools.CONFIG.save()
        return True

    def list_lang(self):
        return [
            ('cs_CZ', 'Čeština'),
            ('de_DE', 'Deutsch'),
            ('en_US', 'English'),
            ('es_ES', 'Español'),
            ('fr_FR', 'Français'),
        ]