mirror of https://github.com/jumpserver/jumpserver
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import uuid
|
|
import os
|
|
from django.conf import settings
|
|
from django.db import models
|
|
|
|
|
|
class CeleryTask(models.Model):
|
|
WAITING = "waiting"
|
|
RUNNING = "running"
|
|
FINISHED = "finished"
|
|
LOG_DIR = os.path.join(settings.PROJECT_DIR, 'data', 'celery')
|
|
|
|
STATUS_CHOICES = (
|
|
(WAITING, WAITING),
|
|
(RUNNING, RUNNING),
|
|
(FINISHED, FINISHED),
|
|
)
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4)
|
|
name = models.CharField(max_length=1024)
|
|
status = models.CharField(max_length=128, choices=STATUS_CHOICES, db_index=True)
|
|
log_path = models.CharField(max_length=256, blank=True, null=True)
|
|
date_published = models.DateTimeField(auto_now_add=True)
|
|
date_start = models.DateTimeField(null=True)
|
|
date_finished = models.DateTimeField(null=True)
|
|
|
|
def __str__(self):
|
|
return "{}: {}".format(self.name, self.id)
|
|
|
|
def is_finished(self):
|
|
return self.status == self.FINISHED
|
|
|
|
@property
|
|
def full_log_path(self):
|
|
if not self.log_path:
|
|
return None
|
|
return os.path.join(self.LOG_DIR, self.log_path)
|