ruff ruff

Author: ffaristocrat

docs/conf.py

@@ -11,7 +11,7 @@
 import os
 import sys
 
-sys.path.insert(0, os.path.abspath('../src'))
+sys.path.insert(0, os.path.abspath("../src"))
 
 # -- Project information -----------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
@@ -27,16 +27,16 @@
     "sphinx.ext.napoleon",
     "sphinx.ext.autodoc",
     "sphinx.ext.autosummary",
-    "sphinx_rtd_theme"
+    "sphinx_rtd_theme",
 ]
 
 autosummary_generate = True
 
 templates_path = ["_templates"]
-exclude_patterns = ['_build', '_templates']
+exclude_patterns = ["_build", "_templates"]
 
 
 # -- Options for HTML output -------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
 
-html_theme = "sphinx_rtd_theme"
+html_theme = "sphinx_rtd_theme"

src/stac_utils/benchmark.py

@@ -46,11 +46,9 @@ def make_msg(msg: str, t: float) -> str:
     h, m = divmod(m, 60)
     d, h = divmod(h, 24)
     if d:
-        msg = (
-            f"{msg}: {d:.0f} days, " f"{h:.0f} hours, {m:.0f} minutes, {s:.0f} seconds"
-        )
+        msg = f"{msg}: {d:.0f} days, {h:.0f} hours, {m:.0f} minutes, {s:.0f} seconds"
     elif h:
-        msg = f"{msg}: " f"{h:.0f} hours, {m:.0f} minutes, {s:.0f} seconds"
+        msg = f"{msg}: {h:.0f} hours, {m:.0f} minutes, {s:.0f} seconds"
     elif m:
         msg = f"{msg}: {m:.0f} minutes, {s:.0f} seconds"
     else:

src/stac_utils/browser/driver.py

@@ -60,15 +60,15 @@ def __enter__(self):
         options.add_argument("--window-size=1280x1696")
         options.add_argument("--disable-dev-shm-usage")
         options.add_argument("--disable-dev-tools")
-        options.add_argument("--disable-blink-features=AutomationControlled") 
+        options.add_argument("--disable-blink-features=AutomationControlled")
         options.add_argument("--no-zygote")
         options.add_argument(f"--user-data-dir={tempfile.mkdtemp()}")
         options.add_argument(f"--data-path={tempfile.mkdtemp()}")
         options.add_argument(f"--disk-cache-dir={tempfile.mkdtemp()}")
         options.add_experimental_option(
             "prefs", {"download.default_directory": self.download_directory}
         )
-        options.add_experimental_option("useAutomationExtension", False) 
+        options.add_experimental_option("useAutomationExtension", False)
 
         self.driver = webdriver.Chrome(service=service, options=options)
 
@@ -77,7 +77,9 @@ def __enter__(self):
         """
         self.driver.download_directory = self.download_directory
 
-        self.driver.execute_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})") 
+        self.driver.execute_script(
+            "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
+        )
 
         return self.driver
 

src/stac_utils/google.py

@@ -528,6 +528,7 @@ def send_data_to_sheets(
     :param input_option: Specified input option, `RAW` by default
     :param client: Google Sheets client
     :param is_overwrite: Overwrite option, defaults to `True`
+    :param is_fill_in_nulls: replace nulls with blanks, default to `False`
     :return: Google Sheets API response
     """
 
@@ -602,21 +603,25 @@ def copy_file(file_id: str, new_file_name: str = None, client: Resource = None)
 
     return new_file_id
 
-def upload_file_to_drive(local_path: str,
-                         gdrive_file_name: str,
-                         existing_mime_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
-                         gdrive_mime_type = 'application/vnd.google-apps.spreadsheet',
-                         permissions = [{
-                            "type": "domain",
-                            "role": "reader",
-                            "domain": "staclabs.io",
-                            }],
-                        client: Resource = None
-                        ) -> str:
-    """
-    Uploads a local file to Google Drive. 
-    
-    By default, it uploads an Excel file and converts it to a Google Sheet. Specify the 
+
+def upload_file_to_drive(
+    local_path: str,
+    gdrive_file_name: str,
+    existing_mime_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+    gdrive_mime_type="application/vnd.google-apps.spreadsheet",
+    permissions=[
+        {
+            "type": "domain",
+            "role": "reader",
+            "domain": "staclabs.io",
+        }
+    ],
+    client: Resource = None,
+) -> str:
+    """
+    Uploads a local file to Google Drive.
+
+    By default, it uploads an Excel file and converts it to a Google Sheet. Specify the
     Mime Type parameters if you want to upload a CSV to a Google Sheet, a Word doc to a
     Google Doc, etc.
 
@@ -628,25 +633,20 @@ def upload_file_to_drive(local_path: str,
     :param gdrive_file_name: Name we should give the uploaded file
     :param existing_mime_type: Mime type of the local file (defaults to Excel)
     :param gdrive_mime_type: Mime type of the resulting file (defaults to Google Sheets)
-    :param permissions: Array of Google permissions objects specifying who should access the 
+    :param permissions: Array of Google permissions objects specifying who should access the
     new file. Defaults to giving everyone at stac labs read-only permissions.
     :param client: Google Drive client
     :return: id of the file on Google drive
     """
     client = client or auth_drive()
-    file_metadata = {
-            "name": gdrive_file_name,
-            "mimeType": gdrive_mime_type
-        }
-    media = MediaFileUpload(local_path,
-                            mimetype=existing_mime_type,
-                            resumable=True)
+    file_metadata = {"name": gdrive_file_name, "mimeType": gdrive_mime_type}
+    media = MediaFileUpload(local_path, mimetype=existing_mime_type, resumable=True)
     file = (
-            client.files()
-            .create(body=file_metadata, media_body=media, fields="id")
-            .execute()
-        )
-    fileId = file['id']
+        client.files()
+        .create(body=file_metadata, media_body=media, fields="id")
+        .execute()
+    )
+    fileId = file["id"]
 
     batch = client.new_batch_http_request()
     for perm in permissions:

src/stac_utils/reach.py

@@ -18,7 +18,7 @@ class ReachClient(HTTPClient):
     api_user: username, will pick up "REACH_API_USER" if it's in the environment
     api_password: password, will pick up "REACH_API_PASSWORD" if it's in the environment
     """
-    
+
     base_url = "https://api.reach.vote/api/v1"
 
     def __init__(self, api_user: str = None, api_password: str = None, *args, **kwargs):

src/tests/test_google.py

@@ -530,9 +530,9 @@ def test_text_stream_from_drive_with_client(
         mock_download_complete = MagicMock(progress=lambda: 1)
 
         # downloader start case (at 1)
-        mock_downloader.next_chunk.return_value[0].progress.return_value = (
-            mock_download_progress
-        )
+        mock_downloader.next_chunk.return_value[
+            0
+        ].progress.return_value = mock_download_progress
 
         # side effect to handle downloader range
         # ("if side_effect is an iterable then each call to the mock will return the next value from the iterable")