ok
Direktori : /opt/alt/python37/lib/python3.7/site-packages/lve_utils/hooks/ |
Current File : //opt/alt/python37/lib/python3.7/site-packages/lve_utils/hooks/plesk_package_rename_hook.py |
#!/usr/bin/python # coding=utf-8 # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2022 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT from __future__ import absolute_import import os from functools import wraps from clcommon.public_hooks.lib import ModifyPackageHook import cldetectlib from clcontrollib import detect_panelclass from clveconfig.ve_config import get_xml_config, save_xml OLD_ADMIN_KEY = 'OLD_ADMIN_TEMPLATE' NEW_ADMIN_KEY = 'NEW_ADMIN_TEMPLATE' OLD_DOMAIN_KEY = 'OLD_DOMAIN_TEMPLATE' NEW_DOMAIN_KEY = 'NEW_DOMAIN_TEMPLATE' def is_plesk_or_exit(f): """ Check that function runs on plesk, or return None """ @wraps(f) def inner(*args, **kwargs): if not cldetectlib.is_plesk(): return None else: return f(*args, **kwargs) return inner class PleskRenameHook(ModifyPackageHook): """ Sync plesk DB package id with ve.cfg package name """ @is_plesk_or_exit def post_modify_package(self, *args, **kwargs): """ Synchronize plesk DB and ve.cfg package names by plesk DB id """ panel = detect_panelclass() package_id = os.environ.get(OLD_ADMIN_KEY) new_id = os.environ.get(NEW_ADMIN_KEY) if not package_id: package_id = os.environ.get(OLD_DOMAIN_KEY) new_id = os.environ.get(NEW_DOMAIN_KEY) assert package_id and package_id == new_id # this id should be equal in case of modify\rename new_package_name = panel.get_package_name_by_id_from_plesk_db(int(package_id)) xml, _ = get_xml_config(False) packages = xml.getElementsByTagName('package') try: el_to_rename = next(filter( # pylint: disable=W1639 lambda x: x.getAttribute('plesk_id') == package_id, packages )) except StopIteration: return el_to_rename.setAttribute('id', new_package_name) save_xml(xml)