Format code with black

Author: Cloud1e

python/mspasspy/algorithms/calib.py

@@ -100,7 +100,7 @@ def __init__(
           correction.
         """
         db = fetch_dbhandle(dbname_or_handle)
-        
+
         if query:
             cursor = db[collection].find(query)
         else:

python/mspasspy/client.py

@@ -246,7 +246,9 @@ def __init__(
             elif isinstance(url, str) and not url.startswith("mongodb://"):
                 url = f"mongodb://{url}"
 
-            mongo_plugin = MongoDBWorker(dbname=dbname, url=url, dbclient_key="dbclient")
+            mongo_plugin = MongoDBWorker(
+                dbname=dbname, url=url, dbclient_key="dbclient"
+            )
             self._dask_client.register_plugin(mongo_plugin, name="mongodb_worker")
 
     def get_database_client(self):

python/mspasspy/db/database.py

@@ -2160,7 +2160,12 @@ def _delete_attributes(self, collection, keylist, query=None, verbose=False):
                         val = doc.pop(k)
                         if verbose:
                             print(
-                                "Deleted ", val, " with key=", k, " from doc with id=", id
+                                "Deleted ",
+                                val,
+                                " with key=",
+                                k,
+                                " from doc with id=",
+                                id,
                             )
                         counts[k] += 1
                         n += 1

python/mspasspy/io/distributed.py

@@ -69,7 +69,7 @@ def read_ensemble_parallel(
     read_distributed_data.  See that function for parameter descriptions.
     """
     db = fetch_dbhandle(dbname_or_handle)
-    
+
     if sort_clause:
         cursor = db[collection].find(query).sort(sort_clause)
     else:

python/mspasspy/util/db_utils.py

@@ -14,11 +14,11 @@
 def fetch_dbhandle(dbname_or_handle):
     """
     Fetch a Database handle from dbname_or_handle.
-    
+
     This function provides a unified way to get a Database instance.  In parallel mode
     (Dask worker), it fetches the DBClient from the worker plugin to avoid serializing
     Database/DBClient objects.  In serial mode, it returns the provided Database instance.
-    
+
     :param dbname_or_handle: Either a string (database name) or a Database instance.
         If a string is provided, the function assumes it's being called in a Dask worker
         context and will get the DBClient from worker.data["dbclient"].
@@ -27,7 +27,7 @@ def fetch_dbhandle(dbname_or_handle):
     :raises: ValueError if dbname_or_handle is neither a string nor a Database instance,
         or if called with a string outside of a Dask worker context.
     """
-    
+
     if isinstance(dbname_or_handle, str):
         try:
             worker = get_worker()
@@ -49,7 +49,9 @@ def fetch_dbhandle(dbname_or_handle):
     elif isinstance(dbname_or_handle, Database):
         db = dbname_or_handle
     else:
-        message = "fetch_dbhandle: illegal type for arg1={}\n".format(type(dbname_or_handle))
+        message = "fetch_dbhandle: illegal type for arg1={}\n".format(
+            type(dbname_or_handle)
+        )
         message += "Must be str defining a db name or a Database object"
         raise ValueError(message)
     return db
@@ -64,7 +66,9 @@ class MongoDBWorker(WorkerPlugin):
     connection leaks when Database objects are serialized to workers.
     """
 
-    def __init__(self, dbname, url="mongodb://localhost:27017/", dbclient_key="dbclient"):
+    def __init__(
+        self, dbname, url="mongodb://localhost:27017/", dbclient_key="dbclient"
+    ):
         self.dbname = dbname
         self.connection_url = url
         self.dbclient_key = dbclient_key
@@ -82,4 +86,3 @@ def teardown(self, worker):
         dbclient = worker.data.get(self.dbclient_key)
         if dbclient:
             dbclient.close()
-

python/mspasspy/util/seispp.py

@@ -34,7 +34,7 @@ def index_data(filebase, dbname_or_handle, ext="d3C", verbose=False):
     """
 
     db = fetch_dbhandle(dbname_or_handle)
-    
+
     # This loads default mspass schema
     mdef = MetadataDefinitions()
     yamlfile = filebase + ".yaml"

python/tests/io/test_distributed.py

@@ -1195,7 +1195,7 @@ def test_read_distributed_ensemble(
                 if wfid in wfl:
                     number_hits += 1
             assert number_hits == 1
-    
+
     # Clean up: close dask client
     if scheduler == "dask":
         dask_client.close()
@@ -1614,7 +1614,7 @@ def test_write_distributed_ensemble(
             assert d.is_defined("dir")
             assert d.is_defined("dfile")
             assert d["storage_mode"] == "file"
-    
+
     # Clean up: close dask client
     if scheduler == "dask":
         dask_client.close()