mirror of
https://github.com/Rushilwiz/NewViewsNews.git
synced 2025-04-18 10:30:16 -04:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
# Create your models here.
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
from django.contrib.auth.models import User
|
|
from django.urls import reverse
|
|
|
|
from PIL import Image
|
|
|
|
from tinymce.models import HTMLField
|
|
|
|
# Create your models here.
|
|
|
|
|
|
class Article(models.Model):
|
|
A_L = 'authoritarian_left'
|
|
A_R = 'authoritarian_right'
|
|
L_L = 'libertarian_left'
|
|
L_R = 'libertarian_right'
|
|
POLITICAL_CHOICES = [
|
|
(A_L, ('Authoritarian Left')),
|
|
(A_R, ('Authoritarian Right')),
|
|
(L_L, ('Libertarian Left')),
|
|
(L_R, ('Libertarian Right')),
|
|
]
|
|
|
|
headline = models.CharField(max_length=100)
|
|
content = HTMLField()
|
|
date_published = models.DateTimeField(auto_now_add=True)
|
|
author = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
likes = models.IntegerField(default=0)
|
|
header = models.ImageField(default='default-header.jpg', upload_to='article-headers')
|
|
header_caption = models.CharField(max_length=100, default="")
|
|
tag = models.CharField(max_length=19,choices=POLITICAL_CHOICES,default=A_L)
|
|
|
|
def __str__(self):
|
|
return f"{self.author}'s article on {self.date_published}"
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('article-detail', kwargs={'pk': self.pk})
|
|
|
|
def save(self, *args, **kwargs):
|
|
super().save(*args, **kwargs)
|
|
|
|
img = Image.open(self.header.path)
|
|
|
|
if img.width > 1280 or img.height > 768:
|
|
output_size = (1280, 768)
|
|
img.thumbnail(output_size)
|
|
img.save(self.header.path)
|