# Cody Cook
# CS-340 Client/Server Development
# SNHU
# 2024/04/02
# Project One

# Import necessary libraries
from pymongo import MongoClient
from pymongo.errors import ConnectionFailure, OperationFailure
from urllib.parse import quote_plus
from bson.objectid import ObjectId

# Define the AnimalShelter class
class AnimalShelter():
    """ CRUD operations for Animal collection in MongoDB """

    # Initialize the MongoClient and the database
    def __init__(self, username, password, hostname, port, database, collection):
        """ Initialize the MongoClient and the database """

        connection = "mongodb://%s:%s@%s:%d" % (quote_plus(username), quote_plus(password), hostname, port)

        # Try to connect to the database
        try:
            self.client = MongoClient(connection) # The client object is an instance of the MongoClient class
            self.database = self.client['%s' % (database)] # The database object is an instance of the database class
            self.collection = self.database['%s' % (collection)] # The collection object is an instance of the collection class

        # If the connection fails, print an error message
        except ConnectionFailure as e: 
            # If the connection fails, print an error message
            print("Could not connect to MongoDB: %s" % e)

# Create method
    def create(self, data):
        """ Create a new animal in the database """

        if data is not None:
            # Try to insert the data into the database
            try:
                self.collection.insert_one(data) # Insert the data into the collection
                print("Animal added successfully") # Print a success message
                return True # Return True if successful
            
            # If the operation fails, print an error message
            except OperationFailure as e:
                # If the operation fails, print an error message
                print("Could not add animal: %s" % e)
                # Return False if the operation fails
                return False
        else:
            # If the data is empty, print an error message
            print("Data is empty, nothing to add")
            # Return False if the data is empty
            return False

# Read method
    def read(self, data):
        """ Read an animal from the database """

        if data is not None:
            # Try to read the data from the database
            try: 
                # Read the data from the collection
                documents = list(self.collection.find(data, {"_id": False}))
                # Return the documents
                return documents
            
            # If the operation fails, print an error message
            except OperationFailure as e:
                # Print an error message
                print("Could not read animal: %s" % e)
                # Return an empty list
                return []
        else:
            # If the data is empty, print an error message
            print("Data is empty, nothing to read")
            # Return an empty list
            return []

# Update method
    def update(self, query, data):
        """ Update an animal in the database """
        
        # Check if the query and data are not empty
        if query is not None and data is not None:
            # Try to update the data in the database
            try:
                # Update the data in the collection
                result = self.collection.update_many(query, {'$set': data})
                # Print a success message
                print(f"Animals updated: {result.modified_count}")
                # Return the number of animals updated
                return result.modified_count
            except OperationFailure as e:
                # If the operation fails, print an error message
                print(f"Could not update animal: {e}")
                return 0
        else:
            # If the data or query is empty, print an error message and return 0
            print("Data and/or query is empty, nothing to update")
            return 0
        
# Delete method
    def delete(self, data):
        """ Delete an animal from the database """
        # Check if the data is not empty
        if data is not None:
            # Try to delete the data from the database
            try:
                # Delete the data from the collection
                result = self.collection.delete_many(data)
                # Print a success message
                print(f"Animals deleted: {result.deleted_count}")
                # Return the number of animals deleted
                return result.deleted_count
            except OperationFailure as e:
                # If the operation fails, print an error message and return 0
                print(f"Could not delete animal: {e}")
                return 0
        else:
            # If the data is empty, print an error message and return 0
            print("Data is empty, nothing to delete")
            return 0