mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-26 07:22:27 +00:00
perf: DBeaver连接时不检查更新、不提示创建实例数据库、不弹出下载驱动框
This commit is contained in:
@@ -4,6 +4,9 @@ import win32api
|
||||
import shutil
|
||||
import subprocess
|
||||
|
||||
from xml.etree import ElementTree
|
||||
from xml.sax import SAXException
|
||||
|
||||
from common import wait_pid, BaseApplication
|
||||
|
||||
|
||||
@@ -21,16 +24,88 @@ class AppletApplication(BaseApplication):
|
||||
self.port = self.asset.get_protocol_port(self.protocol)
|
||||
self.db = self.asset.spec_info.db_name
|
||||
self.name = '%s-%s-%s' % (self.host, self.db, int(time.time()))
|
||||
self.app_work_path = self.get_app_work_path()
|
||||
self.pid = None
|
||||
self.app = None
|
||||
|
||||
def launch(self):
|
||||
@staticmethod
|
||||
def get_app_work_path():
|
||||
win_user_name = win32api.GetUserName()
|
||||
return r'C:\Users\%s\AppData\Roaming\DBeaverData' % win_user_name
|
||||
|
||||
@staticmethod
|
||||
def _read_config(config_file):
|
||||
default_config = {}
|
||||
if not os.path.exists(config_file):
|
||||
return default_config
|
||||
|
||||
with open(config_file, 'r') as f:
|
||||
for line in f.readlines():
|
||||
try:
|
||||
config_key, config_value = line.split('=')
|
||||
except ValueError:
|
||||
continue
|
||||
default_config[config_key] = config_value
|
||||
return default_config
|
||||
|
||||
@staticmethod
|
||||
def _write_config(config_file, config):
|
||||
with open(config_file, 'w')as f:
|
||||
for key, value in config.items():
|
||||
f.write(f'{key}={value}\n')
|
||||
|
||||
@staticmethod
|
||||
def _merge_driver_xml(src_path, dest_path):
|
||||
tree1 = ElementTree.parse(dest_path)
|
||||
tree2 = ElementTree.parse(src_path)
|
||||
|
||||
for child2 in tree2.getroot():
|
||||
found = False
|
||||
for child1 in tree1.getroot():
|
||||
if child1.tag == child2.tag and child1.attrib == child2.attrib:
|
||||
found = True
|
||||
break
|
||||
if not found:
|
||||
tree1.getroot().append(child2)
|
||||
tree1.write(dest_path)
|
||||
|
||||
def init_driver(self):
|
||||
src_driver = os.path.join(os.path.dirname(self.path), 'drivers')
|
||||
dest_driver = r'C:\Users\%s\AppData\Roaming\DBeaverData\drivers' % win_user_name
|
||||
dest_driver = os.path.join(self.app_work_path, 'drivers')
|
||||
if not os.path.exists(dest_driver):
|
||||
shutil.copytree(src_driver, dest_driver, dirs_exist_ok=True)
|
||||
|
||||
def init_driver_config(self):
|
||||
driver_yml_path = os.path.join(
|
||||
self.app_work_path, 'workspace6', '.metadata', '.config',
|
||||
)
|
||||
driver_yml_file = os.path.join(driver_yml_path, 'drivers.xml')
|
||||
try:
|
||||
self._merge_driver_xml('./config/drivers.xml', driver_yml_file)
|
||||
except (SAXException, FileNotFoundError):
|
||||
os.makedirs(driver_yml_path, exist_ok=True)
|
||||
shutil.copy('./config/drivers.xml', driver_yml_file)
|
||||
|
||||
def init_other_config(self):
|
||||
config_path = os.path.join(
|
||||
self.app_work_path, 'workspace6', '.metadata',
|
||||
'.plugins', 'org.eclipse.core.runtime', '.settings',
|
||||
)
|
||||
os.makedirs(config_path, exist_ok=True)
|
||||
config_file = os.path.join(config_path, 'org.jkiss.dbeaver.core.prefs')
|
||||
|
||||
config = self._read_config(config_file)
|
||||
config['ui.auto.update.check'] = 'false'
|
||||
config['sample.database.canceled'] = 'true'
|
||||
config['tipOfTheDayInitializer.notFirstRun'] = 'true'
|
||||
config['ui.show.tip.of.the.day.on.startup'] = 'false'
|
||||
self._write_config(config_file, config)
|
||||
|
||||
def launch(self):
|
||||
self.init_driver()
|
||||
self.init_driver_config()
|
||||
self.init_other_config()
|
||||
|
||||
def _get_exec_params(self):
|
||||
driver = getattr(self, 'driver', self.protocol)
|
||||
params_string = f'name={self.name}|' \
|
||||
|
Reference in New Issue
Block a user