def watch()

in cosmos-db-migration-utility/src/migrator-app/migrators/DatabaseMigrator.py [0:0]

• 27 lines of code
• 7 McCabe index (conditional complexity)

    def watch(self, tokens, notify_callback):
        try:
            db = self.__client.get_database(self.__database_name)
            logger.info("Fetching collections from Database: %s", self.__database_name)
            collection_names = db.collection_names(include_system_collections=False)
            logger.info("Found collections in database: %s; Collections: %s", self.__database_name,
                        json.dumps(collection_names))
            watch_threads = []
            for collection_name in collection_names:
                namespace = "{}.{}".format(self.__database_name, collection_name)
                token = {}
                if namespace in tokens:
                    token = tokens[namespace]
                collection_migrator = CollectionMigrator(self.__client, self.__cluster_name, self.__database_name, collection_name)
                t = threading.Thread(target=collection_migrator.watch, args=(token, notify_callback,))
                watch_threads.append(t)
                t.start()
                self.__collection_migrators.append(collection_migrator)
                logger.info("Found the collection with namespace %s.%s", self.__database_name, collection_name)
            # wait for threads to join
            for watch_thread in watch_threads:
                watch_thread.join()
        except Exception as e:
            logger.exception(e)
        finally:
            if self.__client is not None:
                self.__client.close()
                logger.info("Gracefully closing the connection")