Spaces:
Sleeping
Sleeping
File size: 1,861 Bytes
af30a30 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
from database.setup_db import get_database
from datetime import datetime
import uuid
from pymongo import DESCENDING
def get_current_week_identifier():
"""
Generate a week identifier based on the current year and week number.
Example: '2023_45' for the 45th week of 2023.
"""
now = datetime.now()
year = now.year
week_number = now.isocalendar()[1] # ISO week number
return f"{year}_Week_{week_number}"
def insert_weekly_task_data(json_data,):
"""
Insert JSON data into the weekly_tasks collection.
Args:
json_data (dict): JSON object containing task data.
"""
db = get_database()
collection = db["weekly_tasks"]
# Generate the current week identifier
week_identifier = get_current_week_identifier()
unique_id = str(uuid.uuid4())
# Check if a document for the given week already exists
existing_document = collection.find_one({"unique_id": unique_id})
if existing_document:
print(f"Document with id: {unique_id} already exists. Skipping insert.")
return
# Insert the document if it doesn't already exist
document = {
"week": week_identifier,
"unique_id": unique_id,
"tasks": json_data,
"created_at": datetime.now()
}
result = collection.insert_one(document)
print(f"Inserted document with ID: {result.inserted_id}")
def fetch_recent_two_entries():
"""
Fetch the two most recent entries from the weekly_tasks collection
based on the created_at timestamp.
Returns:
list: A list of the two most recent documents from the collection.
"""
db = get_database()
collection = db["weekly_tasks"]
# Query to fetch the two most recent entries
recent_entries = list(
collection.find().sort("created_at", DESCENDING).limit(2)
)
return recent_entries
|