56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
import random
|
|
|
|
from django.db import models
|
|
from django.contrib.postgres.fields import JSONField
|
|
from django.contrib.postgres.fields.jsonb import KeyTextTransform
|
|
from django.core.cache import cache
|
|
|
|
|
|
class EntryManager(models.Manager):
|
|
|
|
def dump(self, timestamp=None, host=None, lines=100, **kwargs):
|
|
qs = Entry.objects.all()
|
|
if host:
|
|
qs = qs.filter(host__in=host)
|
|
qs = qs.filter(**self.parse_options(**kwargs))
|
|
if lines:
|
|
qs = qs.order_by('-timestamp')[:int(lines)][::-1]
|
|
return qs
|
|
|
|
def parse_options(self, **kwargs):
|
|
# todo mode='contains' require jsonb KeyTextTransform and probably annotations
|
|
options = {}
|
|
for k, v in kwargs.items():
|
|
if k.upper() == 'csrfmiddlewaretoken':
|
|
continue
|
|
options['data__' + k.upper() + '__in'] = v
|
|
return options
|
|
|
|
def hosts(self):
|
|
value = cache.get('hosts')
|
|
if value is None:
|
|
value = self.model.objects.all().values_list('host', flat=True).distinct()
|
|
cache.set('hosts', value, 600 * random.random())
|
|
return value
|
|
|
|
def units(self):
|
|
value = cache.get('units')
|
|
if value is None:
|
|
value = self.model.objects.all().annotate(unit=KeyTextTransform('_SYSTEMD_UNIT', 'data')).values_list('unit', flat=True).distinct()
|
|
cache.set('units', value, 600 * random.random())
|
|
return value
|
|
|
|
|
|
class Entry(models.Model):
|
|
timestamp = models.DateTimeField(auto_now_add=True, db_index=True)
|
|
host = models.CharField(max_length=128, db_index=True)
|
|
data = JSONField()
|
|
objects = EntryManager()
|
|
|
|
def __str__(self):
|
|
return '%s %s %s' % (self.timestamp, self.host, self.data)
|
|
|
|
@property
|
|
def unit(self):
|
|
return self.data.get('_SYSTEMD_UNIT', '').replace('.service', '')
|