mirror of https://github.com/jumpserver/jumpserver
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
import os
|
|
import tempfile
|
|
import shutil
|
|
from typing import List
|
|
|
|
from django.conf import settings
|
|
|
|
from assets.models import Asset
|
|
|
|
|
|
class BasePlaybookGenerator:
|
|
def __init__(self, assets: list[Asset], strategy, ansible_connection='ssh'):
|
|
self.assets = assets
|
|
self.strategy = strategy
|
|
self.playbook_dir = self.temp_folder_path()
|
|
|
|
def generate(self):
|
|
self.prepare_playbook_dir()
|
|
self.generate_inventory()
|
|
self.generate_playbook()
|
|
|
|
def prepare_playbook_dir(self):
|
|
pass
|
|
|
|
def generate_inventory(self):
|
|
pass
|
|
|
|
def generate_playbook(self):
|
|
pass
|
|
|
|
@property
|
|
def base_dir(self):
|
|
tmp_dir = os.path.join(settings.PROJECT_DIR, 'tmp')
|
|
path = os.path.join(tmp_dir, self.strategy)
|
|
return path
|
|
|
|
def temp_folder_path(self):
|
|
return tempfile.mkdtemp(dir=self.base_dir)
|
|
|
|
def del_temp_folder(self):
|
|
shutil.rmtree(self.playbook_dir)
|
|
|
|
def generate_temp_playbook(self):
|
|
src = self.src_filepath
|
|
dst = os.path.join(self.temp_folder, self.strategy)
|
|
shutil.copytree(src, dst)
|
|
return dst
|