MongoDB自动化测试后清理脚本无法运行问题求助

Hey Rob, let's figure out why your MongoDB cleanup script isn't running and isn't producing any output—there are a few clear issues in your code snippet that are causing this, plus some easy fixes to get it working:

Common Issues in Your Current Code #

• Incorrect method parameter: Your deleteSecond method starts with sel instead of self—this is a critical mistake in Python class methods, as self is required to reference the class instance. Without it, calling the method will throw an error (but you won't see it if you don't handle exceptions).
• No class instantiation or method calls: Defining a class and its methods doesn't automatically run them. You need to create an instance of deleteMongo and explicitly call your delete methods for any code to execute.
• No error handling or logging: If something goes wrong (like a failed database connection, wrong collection name, or invalid query), you won't get any feedback because there's no code to catch or report errors.

Fixed & Improved Code #

Here's a revised version of your script that addresses all these issues, plus adds clarity and error visibility:

from pymongo import MongoClient
from pymongo.errors import PyMongoError

class DeleteMongo():  # PEP8-compliant class naming (CamelCase)
    def delete_first(self):  # PEP8-compliant method naming (snake_case)
        try:
            # Establish database connection
            client = MongoClient('databaseaddress')  # Replace with your actual connection string
            db = client.TableData
            
            # Execute delete and print result
            delete_result = db.Employe.delete_one({"name": "EmployeOne"})
            print(f"Deleted {delete_result.deleted_count} document from 'Employe' collection")
            
            client.close()
        except PyMongoError as e:
            print(f"Error during 'Employe' cleanup: {str(e)}")

    def delete_second(self):
        try:
            client = MongoClient('databaseaddress')
            db = client.TableData
            # Add your actual delete logic here (example below)
            delete_result = db.AnotherCollection.delete_many({})  # Deletes all documents in the collection
            print(f"Deleted {delete_result.deleted_count} documents from 'AnotherCollection' collection")
            
            client.close()
        except PyMongoError as e:
            print(f"Error during 'AnotherCollection' cleanup: {str(e)}")

# Critical: Execute the cleanup logic when the script runs
if __name__ == "__main__":
    cleanup_handler = DeleteMongo()
    cleanup_handler.delete_first()
    cleanup_handler.delete_second()

Key Fixes & Enhancements #

1. Corrected method parameter: Changed sel to self in delete_second so the method can properly interact with the class instance.
2. Added execution entry point: The if __name__ == "__main__": block ensures that when you run the script, it creates an instance of DeleteMongo and calls your delete methods.
3. Error handling: Added try-except blocks to catch MongoDB-specific errors and print them, so you'll know exactly if there's a connection issue, invalid query, or missing collection.
4. Visibility: Added print statements to show how many documents were deleted, so you can confirm the cleanup worked as expected.
5. PEP8 compliance: Renamed classes/methods to follow Python's style guidelines, making the code easier to read and maintain.

Additional Tips #

• Double-check your databaseaddress—it should be a valid MongoDB connection string (e.g., mongodb://localhost:27017/ for a local instance, or a string with credentials for a remote database).
• Verify collection names are spelled correctly (e.g., is Employe supposed to be Employee? If the collection doesn't exist, delete_one won't throw an error, but deleted_count will be 0).
• For cleaner connection management, use a context manager:

  with MongoClient('databaseaddress') as client:
      db = client.TableData
      # Your delete logic here
  

  This automatically closes the connection when done, no need for client.close().

内容的提问来源于stack exchange，提问作者Rob1425