add price

Author: meghost77

config.py

@@ -36,6 +36,9 @@ class Settings(BaseSettings):
                 commentCount
                 tags
                 authorUsername
+                price {
+                    formattedPrice
+                }
             }
             pageInfo {
                 hasNextPage

db/models.py

@@ -1,11 +1,16 @@
 from datetime import datetime
 from typing import List, Optional, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 # Base model for a post
 
 
+class Price(BaseModel):
+    formattedPrice: Optional[str] = Field(
+        None, description="Formatted price of the item")
+
+
 class Post(BaseModel):
     id: Union[str, int]  # ID can be either string or integer
     bodyHTML: str  # HTML content of the post
@@ -18,6 +23,7 @@ class Post(BaseModel):
     commentCount: int  # Number of comments on the post
     tags: Optional[List[str]] = None
     authorUsername: Optional[str] = None
+    price: Optional[Price] = None
 # Detailed model for a post extending the base Post model
 
 

utils/services.py

@@ -115,7 +115,10 @@ async def parse_and_store_posts(search_query: str, city: Optional[str] = None):
                 'firstImage': item['imagesList'][0] if item['imagesList'] else None,
                 'commentCount': item.get('commentCount', 0),
                 'tags': item.get('tags', []),
-                'authorUsername': item.get('authorUsername')
+                'authorUsername': item.get('authorUsername'),
+                'price': {
+                    'formattedPrice': item.get('price', {}).get('formattedPrice')
+                } if item.get('price') else None
             }
             logger.debug(f"Storing post {post_data['id']}")
             await update_one(collection_name, {'id': post_data['id']}, {'$set': post_data}, upsert=True)