main copy.py

from fastapi import FastAPI, HTTPException
from pymongo import MongoClient
from pymongo.collection import Collection
from typing import List
from models import Category, Product
from bson import ObjectId
app = FastAPI()

mongo_connection_string = "mongodb+srv://11335220:11335220@cluster0.ojobzbu.mongodb.net/Todo?retryWrites=true&w=majority"
client = MongoClient(mongo_connection_string)

# Specify the default database name here
db = client.get_database("Todo")

categories_collection: Collection = db["categories"]
products_collection: Collection = db["products"]


def find_category_by_name(name: str):
    return categories_collection.find_one({"name": name})


def find_product_by_name(name: str):
    return products_collection.find_one({"name": name})

# Get Category Posts.....................


@app.get("/categories/", response_model=List[Category])
def get_categories():
    categories = list(categories_collection.find({}))
    if not categories:
        raise HTTPException(status_code=404, detail="No categories found")

    return categories

# Category Post add.......................


@app.post("/categories/", response_model=Category)
def create_category(category: Category):
    existing_category = find_category_by_name(category.name)
    if existing_category:
        raise HTTPException(
            status_code=400, detail="Category with this name already exists")

    category_dict = category.dict()
    inserted_category = categories_collection.insert_one(category_dict)
    category.id = str(inserted_category.inserted_id)

    # Add a success message to the response
    return {"message": "Category created successfully", "category": category}

# Category Post delete....................


@app.delete("/categories/{category_id}")
def delete_category(category_id: str):
    category_id_obj = ObjectId(category_id)
    result = categories_collection.delete_one({"_id": category_id_obj})
    if result.deleted_count == 0:
        raise HTTPException(status_code=404, detail="Category not found")
    return {"message": "Category deleted successfully"}

# Get Single Categories  by ID


@app.get("/categories/{category_id}", response_model=Category)
def get_single_category(category_id: str):
    # Convert the category_id to an ObjectId (assuming you're using MongoDB ObjectIDs)
    category_id_obj = ObjectId(category_id)

    # Find the category in the database
    category = categories_collection.find_one({"_id": category_id_obj})

    if not category:
        raise HTTPException(status_code=404, detail="Category not found")
    return category


@app.put("/categories/{category_id}", response_model=Category)
def update_category(category_id: str, updated_category: Category):
    # Convert the category_id to an ObjectId (assuming you're using MongoDB ObjectIDs)
    category_id_obj = ObjectId(category_id)

    # Find the category in the database
    existing_category = categories_collection.find_one(
        {"_id": category_id_obj})

    if not existing_category:
        raise HTTPException(status_code=404, detail="Category not found")

    # Convert the updated category to a dictionary (using .dict()) and remove the "_id" field
    update_data = updated_category.dict(exclude_unset=True)

    # Update the existing category in the database
    categories_collection.update_one(
        {"_id": category_id_obj}, {"$set": update_data})

    # Fetch the updated category from the database and return it in the response
    updated_category = categories_collection.find_one({"_id": category_id_obj})
    return updated_category



@app.post("/products/", response_model=Product)
def create_product(product: Product):
    existing_product = find_product_by_name(product.name)
    if existing_product:
        raise HTTPException(
            status_code=400, detail="Product with this name already exists")
    product_dict = product.dict()
    inserted_product = products_collection.insert_one(product_dict)
    product.id = str(inserted_product.inserted_id)
    return product


@app.get("/products/", response_model=List[Product])
def get_products():
    return list(products_collection.find({}))


@app.get("/products/{category_id}", response_model=List[Product])
def get_products_by_category(category_id: str):
    return list(products_collection.find({"category_id": category_id}))