import shutil
import time

from Common import *
from loguru import logger
class UpdateSql():
    def __init__(self, install_path):
        self.service_name = 'ComacMariaDB'
        self.root_path = install_path
        self.mariadb_data_path = os.path.join(install_path, 'mariadb', 'data')
        self.mariadb_port = mariadb_port
        self.passowrd = mariadb_passowrd
        self.init_schema = mariadb_init_schema
        self.init_sql = os.path.join(install_path, 'mariadb', 'data', 'init.sql')

    def init_db(self):
        new_path = get_resource_path(os.path.join('datas', 'init.sql'))
        shutil.copy(str(new_path), self.mariadb_data_path)
        command2 = fr'{self.root_path}\mariadb\bin\mysql --no-defaults -u root -p{self.passowrd} -P {self.mariadb_port} {self.init_schema} < {self.init_sql}'
        with os.popen(command2) as stream:
            res2 = stream.read()
            logger.info(res2)
        os.remove(self.init_sql)
        pass

if __name__ == '__main__':
    current_dir = os.getcwd()
    logger.info("开始更新")
    UpdateSql(current_dir).init_db()
    logger.info("完成更新，3秒钟后退出")
    time.sleep(3)