mirror of https://github.com/jumpserver/jumpserver
36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
import os.path
|
|
import uuid
|
|
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from ops.const import CreateMethods
|
|
from ops.exception import PlaybookNoValidEntry
|
|
from orgs.mixins.models import JMSOrgBaseModel
|
|
|
|
|
|
class Playbook(JMSOrgBaseModel):
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
name = models.CharField(max_length=128, verbose_name=_('Name'), null=True)
|
|
path = models.FileField(upload_to='playbooks/')
|
|
creator = models.ForeignKey('users.User', verbose_name=_("Creator"), on_delete=models.SET_NULL, null=True)
|
|
comment = models.CharField(max_length=1024, default='', verbose_name=_('Comment'), null=True, blank=True)
|
|
create_method = models.CharField(max_length=128, choices=CreateMethods.choices, default=CreateMethods.blank,
|
|
verbose_name=_('CreateMethod'))
|
|
vcs_url = models.CharField(max_length=1024, default='', verbose_name=_('VCS URL'), null=True, blank=True)
|
|
|
|
@property
|
|
def entry(self):
|
|
work_dir = self.work_dir
|
|
valid_entry = ('main.yml', 'main.yaml', 'main')
|
|
for f in os.listdir(work_dir):
|
|
if f in valid_entry:
|
|
return os.path.join(work_dir, f)
|
|
raise PlaybookNoValidEntry
|
|
|
|
@property
|
|
def work_dir(self):
|
|
work_dir = os.path.join(settings.DATA_DIR, "ops", "playbook", self.id.__str__())
|
|
return work_dir
|