Add Comment

1 files changed, 3 insertions, 2 deletions

diff --git a/src/database/models/blog.py b/src/database/models/blog.py
index 6a0fb42..f18c7b6 100644
--- a/src/database/models/blog.py
+++ b/src/database/models/blog.py
@@ -1,6 +1,6 @@
-import datetime
+from typing import List
 from sqlalchemy import BigInteger, String, DateTime
-from sqlalchemy.orm import mapped_column, Mapped
+from sqlalchemy.orm import mapped_column, Mapped, relationship
 from src.database import db
 
 
@@ -9,5 +9,6 @@ class Blog(db.Model):
 
     id = mapped_column(BigInteger, primary_key=True)
     domain: Mapped[str] = mapped_column(String(255), nullable=False, unique=True)
+    comments = relationship('Comment', back_populates='blog')
     created_at = mapped_column(DateTime)
     updated_at = mapped_column(DateTime)