github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

pref: 修改 inventory

pull/8997/head
ibuler 2022-10-28 16:25:16 +08:00
parent bd9afaba1e
commit 084dcc7b44
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/ops/ansible/inventory.py
View File

@ -9,16 +9,17 @@ __all__ = ['JMSInventory']
class JMSInventory: class JMSInventory:
def __init__(self, manager, assets=None, account_policy='smart', account_prefer='root,administrator'): def __init__(self, manager, assets=None, account_policy='smart',
account_prefer='root,administrator', host_callback=None):
""" """
:param assets: :param assets:
:param account_prefer: account username name if not set use account_policy :param account_prefer: account username name if not set use account_policy
:param account_policy: smart, privileged_must, privileged_first :param account_policy: smart, privileged_must, privileged_first
""" """
self.manager = manager
self.assets = self.clean_assets(assets) self.assets = self.clean_assets(assets)
self.account_prefer = account_prefer self.account_prefer = account_prefer
self.account_policy = account_policy self.account_policy = account_policy
self.host_callback = host_callback
@staticmethod @staticmethod
def clean_assets(assets): def clean_assets(assets):
@ -106,7 +107,8 @@ class JMSInventory:
'jms_asset': { 'jms_asset': {
'id': str(asset.id), 'name': asset.name, 'address': asset.address, 'id': str(asset.id), 'name': asset.name, 'address': asset.address,
'type': asset.type, 'category': asset.category, 'type': asset.type, 'category': asset.category,
'protocol': asset.protocol, 'port': asset.port,'database': '', 'protocol': asset.protocol, 'port': asset.port,
'category_property': asset.category_property,
'protocols': [{'name': p.name, 'port': p.port} for p in protocols], 'protocols': [{'name': p.name, 'port': p.port} for p in protocols],
}, },
'jms_account': { 'jms_account': {
@ -118,9 +120,6 @@ class JMSInventory:
ansible_connection = ansible_config.get('ansible_connection', 'ssh') ansible_connection = ansible_config.get('ansible_connection', 'ssh')
host.update(ansible_config) host.update(ansible_config)
if platform.category == 'database':
host['jms_asset']['database'] = asset.database.db_name
gateway = None gateway = None
if asset.domain: if asset.domain:
gateway = asset.domain.select_gateway() gateway = asset.domain.select_gateway()
@ -176,8 +175,8 @@ class JMSInventory:
if not automation.ansible_enabled: if not automation.ansible_enabled:
host['error'] = _('Ansible disabled') host['error'] = _('Ansible disabled')
if self.manager.host_callback is not None: if self.host_callback is not None:
host = self.manager.host_callback( host = self.host_callback(
host, asset=asset, account=account, host, asset=asset, account=account,
platform=platform, automation=automation, platform=platform, automation=automation,
path_dir=path_dir path_dir=path_dir