from django.db import models

class Website(models.Model):
    name = models.CharField(max_length=100, unique=True)
    base_url = models.URLField()
    description = models.TextField(blank=True, null=True)
    article_list_url = models.URLField(blank=True, null=True)
    article_selector = models.CharField(max_length=255, blank=True, null=True)
    content_selector = models.CharField(max_length=255, blank=True, null=True)
    enabled = models.BooleanField(default=True)

    def __str__(self):
        return self.name


class Article(models.Model):
    website = models.ForeignKey(Website, on_delete=models.CASCADE)
    title = models.CharField(max_length=300)
    url = models.URLField(unique=True)
    pub_date = models.DateTimeField(null=True, blank=True)
    content = models.TextField()  # html内容
    created_at = models.DateTimeField(auto_now_add=True)
    media_files = models.JSONField(default=list, blank=True)

    def __str__(self):
        return self.title