File size: 1,592 Bytes
44cae7c |
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 |
from datetime import datetime
from app.extensions import db
from slugify import slugify
import random
import string
class Article(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(200), nullable=False)
content = db.Column(db.Text, nullable=False)
summary = db.Column(db.Text)
slug = db.Column(db.String(200), unique=True, nullable=False)
created_at = db.Column(db.DateTime, default=datetime.utcnow)
updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
def __init__(self, *args, **kwargs):
if not kwargs.get('slug') and kwargs.get('title'):
base_slug = slugify(kwargs.get('title'))
slug = base_slug
counter = 1
# Keep trying until we find a unique slug
while Article.query.filter_by(slug=slug).first() is not None:
# Add random string for uniqueness
random_string = ''.join(random.choices(string.ascii_lowercase + string.digits, k=4))
slug = f"{base_slug}-{random_string}"
kwargs['slug'] = slug
super().__init__(*args, **kwargs)
class Image(db.Model):
id = db.Column(db.Integer, primary_key=True)
filename = db.Column(db.String(200), nullable=False)
data = db.Column(db.LargeBinary, nullable=False) # Store actual image data
mime_type = db.Column(db.String(100), nullable=False) # Store MIME type
created_at = db.Column(db.DateTime, default=datetime.utcnow) |