Cleanup.

Author: toddbirchard

config.py

@@ -12,4 +12,4 @@
 SQLALCHEMY_DATABASE_PEM = environ.get("SQLALCHEMY_DATABASE_PEM")
 
 # Reset data after each run
-CLEANUP_DATA = True
+CLEANUP_DATA = False

sqlalchemy_tutorial/part3_relationships/__init__.py

@@ -3,7 +3,7 @@
 
 from .joins import get_all_comments, get_all_posts
 from .objects import create_comment_objects, create_post_object, create_user_objects
-from .orm import create_comment, create_user, create_post
+from .orm import create_comment, create_post, create_user
 
 
 def create_relationships():

sqlalchemy_tutorial/part3_relationships/models.py

@@ -45,7 +45,7 @@ class Comment(Base):
     created_at = Column(DateTime, server_default=func.now())
 
     # Relationships
-    user = relationship("User", backref="comments")
+    user = relationship("User", backref="comment")
 
     def __repr__(self):
         return "<Comment %r>" % self.id

sqlalchemy_tutorial/part3_relationships/orm.py

@@ -20,7 +20,9 @@ def create_user(session: Session, user: User) -> User:
     :return: Optional[User]
     """
     try:
-        existing_user = session.query(User).filter(User.username == user.username).first()
+        existing_user = (
+            session.query(User).filter(User.username == user.username).first()
+        )
         if existing_user is None:
             session.add(user)  # Add the user
             session.commit()  # Commit the change
@@ -81,9 +83,7 @@ def create_comment(session: Session, comment: Comment) -> Comment:
     try:
         session.add(comment)  # Add the Comment
         session.commit()  # Commit the change
-        LOGGER.success(
-            f"Created comment {comment} posted by user {comment.user.username}"
-        )
+        LOGGER.success(f"Created comment {comment} from user {comment.user.username}.")
         return comment
     except IntegrityError as e:
         LOGGER.error(e.orig)