Update utils/database.py

utils/database.py

@@ -35,19 +35,20 @@ def init_db():
     Base.metadata.create_all(bind=engine)
     print("Database initialized.")
 
-def post_exists(session: Session, image_url: str) -> bool:
+async def post_exists(session: AsyncSession, image_url: str) -> bool:
     """
     Check if a post with the given image_url already exists in the database.
     """
-    return session.query(Post.id).filter(Post.image_url == image_url).first() is not None
+    result = await session.execute(select(Post).filter(Post.image_url == image_url))
+    return result.scalars().first() is not None
 
-def save_to_db(data: List[Dict]):
+async def save_to_db(data: List[Dict]):
     """
     Save data to the PostgreSQL database, avoiding duplicates.
     """
-    with SessionLocal() as session:
+    async with AsyncSessionLocal() as session:
         for post in data:
-            if not post_exists(session, post.get("image_url")):
+            if not await post_exists(session, post.get("image_url")):
                 new_post = Post(
                     username=post.get("username", ""),
                     caption=post.get("caption", ""),
@@ -61,7 +62,7 @@ def save_to_db(data: List[Dict]):
                     promote=post.get("promote", False),
                 )
                 session.add(new_post)
-        session.commit()
+        await session.commit()
     print("Data saved to database.")
 
 def fetch_posts_from_db(username: str) -> List[Dict]: