from bson import SON
from pymongo import MongoClient
from pymongo.errors import WriteConcernError
from pymongo.monitoring import CommandListener
import logging


class CommandLogger(CommandListener):
    def started(self, event):
        logging.info("Command {0.command_name} with request id "
                     "{0.request_id} started on server "
                     "{0.connection_id} - {0.command}".format(event))

    def succeeded(self, event):
        logging.info("Command {0.command_name} with request id "
                     "{0.request_id} on server {0.connection_id} "
                     "succeeded in {0.duration_micros} "
                     "microseconds - {0.reply}".format(event))

    def failed(self, event):
        logging.info("Command {0.command_name} with request id "
                     "{0.request_id} on server {0.connection_id} "
                     "failed in {0.duration_micros} "
                     "microseconds".format(event))


logging.basicConfig(level=logging.INFO)


client = MongoClient(directConnection=False, event_listeners=[CommandLogger()])
client.admin.command(SON([('configureFailPoint', 'failCommand'),
                          ('mode', {'times': 1}),
                          ('data', {
                              'failCommands': ['insert'],
                              'writeConcernError': {
                                  'code': 91,
                                  'errmsg': 'Replication is being shut down'}})]))

try:
    client.foo.bar.insert_one({})
except WriteConcernError as exc:
    print(exc._error_labels)
    raise

client.admin.command(SON([('configureFailPoint', 'failCommand'),
                          ('mode', 'off')]))