0x2620/phantasmobile
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

only allow /m/

Browse source
This commit is contained in:
j 2025-02-22 12:04:28 -05:00
commit 250fa43c19
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
app/item/models.py
View file

@ -7,6 +7,7 @@ import lxml.html
from django.conf import settings from django.conf import settings
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from django.core.exceptions import ValidationError
from django.db import models from django.db import models
from django.db.models.functions import ExtractWeek, ExtractYear from django.db.models.functions import ExtractWeek, ExtractYear
from django.urls import reverse from django.urls import reverse
@ -48,6 +49,10 @@ class Item(models.Model):
self.get_hue() self.get_hue()
super().save(*args, **kwargs) super().save(*args, **kwargs)
def clean(self):
if '/m/' not in self.url:
raise ValidationError("You can only use mobile urls(/m/)")
def __str__(self): def __str__(self):
return '%s (%s)' % (self.title, self.url) return '%s (%s)' % (self.title, self.url)
@ -101,12 +106,14 @@ class Item(models.Model):
return week, archive return week, archive
def get_week(self): def get_week(self):
return int(self.published.strftime('%W')) return int(self.published.strftime('%W')) if self.published else None
def get_year(self): def get_year(self):
return int(self.published.strftime('%Y')) return int(self.published.strftime('%Y')) if self.published else None
def get_monday(self): def get_monday(self):
if not self.published:
return None
d = '%s-W%s' % (self.get_year(), self.get_week()) d = '%s-W%s' % (self.get_year(), self.get_week())
return datetime.strptime(d + '-1', "%Y-W%W-%w").strftime('%Y-%m-%d') return datetime.strptime(d + '-1', "%Y-W%W-%w").strftime('%Y-%m-%d')