Spaces:
Sleeping
Sleeping
File size: 1,907 Bytes
0310410 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | from django.db import models
from django.contrib.auth.models import User
class Screener(models.Model):
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="screeners")
name = models.CharField(max_length=255)
filters = models.JSONField(default=dict, blank=True)
columns = models.JSONField(default=list, blank=True)
selected_symbols = models.JSONField(default=list, blank=True)
cached_results = models.JSONField(null=True, blank=True, default=None)
last_run_at = models.DateTimeField(null=True, blank=True)
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
class Meta:
ordering = ["-updated_at"]
@staticmethod
def _normalize_unique_symbols(values):
if not isinstance(values, list):
return []
normalized = []
seen = set()
for raw in values:
symbol = str(raw or "").strip().upper()
if not symbol or symbol in seen:
continue
seen.add(symbol)
normalized.append(symbol)
return normalized
def save(self, *args, **kwargs):
# Enforce a canonical selected symbol list on every save path.
self.selected_symbols = self._normalize_unique_symbols(self.selected_symbols)
super().save(*args, **kwargs)
def __str__(self):
return f"{self.user.username}: {self.name}"
class ScreenerSymbolCache(models.Model):
symbol = models.CharField(max_length=32, unique=True)
data = models.JSONField(default=dict, blank=True)
last_fetched_at = models.DateTimeField()
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
class Meta:
ordering = ["-last_fetched_at"]
def __str__(self):
return self.symbol
|