lint, format: Run linter & formatter

redjax · redjax · commit a672c24d3049 · 2023-08-16T21:22:28.000-04:00
diff --git a/red_utils/__init__.py b/red_utils/__init__.py
@@ -3,8 +3,13 @@
 import pkgutil
 
 ## Import modules with only stdlib dependencies directly
-from .context_managers import DictProtect, ListProtect, SQLiteConnManager
-from .context_managers import async_benchmark, benchmark
+from .context_managers import (
+    DictProtect,
+    ListProtect,
+    SQLiteConnManager,
+    async_benchmark,
+    benchmark,
+)
 from .dict_utils import debug_dict, merge_dicts, update_dict, validate_dict
 from .file_utils import crawl_dir, default_json_dir, export_json, ts
 from .hash_utils import get_hash_from_str
@@ -24,7 +29,6 @@
     trim_uuid,
 )
 
-
 ## Use pkgutil to only load modules
 #  if dependencies are met
 if pkgutil.find_loader("diskcache"):
diff --git a/red_utils/context_managers/__init__.py b/red_utils/context_managers/__init__.py
@@ -1,5 +1,5 @@
 from __future__ import annotations
 
 from .benchmarks import async_benchmark, benchmark
+from .database_managers.sqlite_managers import SQLiteConnManager
 from .object_managers.protect import DictProtect, ListProtect
-from .database_managers.sqlite_managers import SQLiteConnManager
diff --git a/red_utils/context_managers/benchmarks/fn_benchmarks.py b/red_utils/context_managers/benchmarks/fn_benchmarks.py
@@ -3,7 +3,6 @@
 from contextlib import asynccontextmanager, contextmanager
 import time
 
-
 @contextmanager
 def benchmark(description: str = "Unnamed function timer") -> None:
     """Time a function call.
diff --git a/red_utils/context_managers/database_managers/__init__.py b/red_utils/context_managers/database_managers/__init__.py
@@ -1 +1,3 @@
-from .sqlite_managers import SQLiteConnManager
+from __future__ import annotations
+
+from .sqlite_managers import SQLiteConnManager
diff --git a/red_utils/context_managers/database_managers/sqlite_managers.py b/red_utils/context_managers/database_managers/sqlite_managers.py
@@ -1,5 +1,4 @@
-"""
-Context manager utilities for interacting with SQLite databases, using
+"""Context manager utilities for interacting with SQLite databases, using
 the stdlib sqlite3 library.
 """
 from __future__ import annotations
@@ -11,14 +10,14 @@
 
 class SQLiteConnManager:
     """Handle interactions with a SQLite database.
-    
+
     Uses built-in functions to query a database, execute SQL statements,
     and gracefully open/close the DB using context managers.
-    
+
     Usage:
         Provide a path string to the SQLite database:
             sqlite_connection = SQLiteConnManager(path="/path/to/db.sqlite")
-        
+
         Call sqlite3 functions, i.e. "get_tables()":
             tables = sqlite_conn.get_tables()
     """
diff --git a/red_utils/context_managers/object_managers/protect.py b/red_utils/context_managers/object_managers/protect.py
@@ -4,7 +4,6 @@
 import inspect
 import json
 
-
 class ListProtect:
     """Protect a list during modification by modifying a copy instead of the original.
 
diff --git a/red_utils/dict_utils/operations.py b/red_utils/dict_utils/operations.py
@@ -4,7 +4,6 @@
 
 from .validators import validate_dict
 
-
 def debug_dict(in_dict: dict = None) -> None:
     """Debug print a dict by looping overkeys and printing.
 
diff --git a/red_utils/dict_utils/validators.py b/red_utils/dict_utils/validators.py
@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-
 def validate_dict(_dict: dict[str, str] = None) -> dict[str, str]:
     if not _dict:
         raise ValueError("Missing dict to evaluate")
diff --git a/red_utils/diskcache_utils/operations.py b/red_utils/diskcache_utils/operations.py
@@ -30,7 +30,6 @@
 
 from diskcache import Cache
 
-
 def convert_to_seconds(unit: str = None, amount: int = None) -> int:
     ## Allowed strings for conversion
     valid_time_units: list[int] = ["seconds", "hours", "minutes", "days", "weeks"]
diff --git a/red_utils/diskcache_utils/validators.py b/red_utils/diskcache_utils/validators.py
@@ -9,7 +9,6 @@
 
 from diskcache import Cache
 
-
 def validate_key(key: valid_key_types = None, none_ok: bool = False) -> Union[str, int]:
     """Validate input diskcache key.
 
diff --git a/red_utils/fastapi_utils/dependencies.py b/red_utils/fastapi_utils/dependencies.py
@@ -3,7 +3,6 @@
 from fastapi import Request
 from loguru import logger as log
 
-
 async def logging_dependency(request: Request) -> None:
     """https://stackoverflow.com/a/63413392."""
     # log.debug(f"{request.method} {request.url}")
diff --git a/red_utils/fastapi_utils/operations.py b/red_utils/fastapi_utils/operations.py
@@ -23,7 +23,6 @@
 from fastapi import APIRouter, FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-
 def fix_api_docs(app: FastAPI = None):
     """Fix error loading /docs when a root_path is set.
 
diff --git a/red_utils/fastapi_utils/validators.py b/red_utils/fastapi_utils/validators.py
@@ -4,7 +4,6 @@
 
 from fastapi import APIRouter
 
-
 def is_str(input: str = None) -> str:
     if not input:
         raise ValueError("Missing input to evaluate")
diff --git a/red_utils/file_utils/operations.py b/red_utils/file_utils/operations.py
@@ -9,7 +9,6 @@
 
 from .constants import default_json_dir, ts
 
-
 def export_json(
     input: Union[str, list[list, dict], dict[str, Any]] = None,
     output_dir: str = default_json_dir,
diff --git a/red_utils/hash_utils/operations.py b/red_utils/hash_utils/operations.py
@@ -2,7 +2,6 @@
 
 import hashlib
 
-
 def get_hash_from_str(input_str: str = None, encoding: str = "utf-8") -> str:
     if not input_str:
         raise ValueError("Missing input string")
diff --git a/red_utils/httpx_utils/operations.py b/red_utils/httpx_utils/operations.py
@@ -10,7 +10,6 @@
 
 from httpx import Client
 
-
 def merge_headers(
     original_headers: dict[str, str] = default_headers,
     update_vals: dict[str, str] = None,
diff --git a/red_utils/httpx_utils/validators.py b/red_utils/httpx_utils/validators.py
@@ -6,7 +6,6 @@
 
 from httpx import AsyncClient, Client
 
-
 def validate_client(
     client: Union[Client, AsyncClient] = None
 ) -> Union[Client, AsyncClient]:
diff --git a/red_utils/loguru_utils/operations.py b/red_utils/loguru_utils/operations.py
@@ -22,7 +22,6 @@
 
 from loguru import logger
 
-
 def add_sink(
     _logger: logger = None,
     sink: Union[str, Path, io.TextIOWrapper, Handler, Callable, Coroutine] = None,
diff --git a/red_utils/loguru_utils/validators.py b/red_utils/loguru_utils/validators.py
@@ -17,7 +17,6 @@
 
 from loguru import logger
 
-
 def validate_logger(_logger: logger = None, none_ok: bool = False) -> logger:
     """Validate a loguru.Logger object."""
     if none_ok:
diff --git a/red_utils/msgpack_utils/operations.py b/red_utils/msgpack_utils/operations.py
@@ -8,7 +8,6 @@
 
 import msgpack
 
-
 def ensure_path(dir: Union[str, Path] = None) -> bool:
     """Ensure a directory path exists.
 
diff --git a/red_utils/sqlalchemy_utils/__init__.py b/red_utils/sqlalchemy_utils/__init__.py
@@ -22,6 +22,8 @@
 """
 from __future__ import annotations
 
+from . import custom_types
+
 ## Import SQLAlchemy dependencies
 ## Import SQLAlchemy DeclarativeBase object
 from .base import Base
@@ -36,6 +38,7 @@
 
 ## Import constants
 from .constants import valid_db_types
+from .custom_types import CompatibleUUID
 
 ## Import custom SQLAlchemy utils
 from .utils import (
@@ -54,6 +57,3 @@
     orm as sa_orm,
 )
 from sqlalchemy.orm import Session, scoped_session, sessionmaker
-
-from . import custom_types
-from .custom_types import CompatibleUUID
diff --git a/red_utils/sqlalchemy_utils/connection_models.py b/red_utils/sqlalchemy_utils/connection_models.py
@@ -5,7 +5,6 @@
 
 import sqlalchemy as sa
 
-
 @dataclass
 class saConnectionBase:
     """Base class for SQLAlchemy connection models.
diff --git a/red_utils/sqlalchemy_utils/custom_types.py b/red_utils/sqlalchemy_utils/custom_types.py
@@ -1,5 +1,4 @@
-"""
-Define custom type classes for SQLAlchemy.
+"""Define custom type classes for SQLAlchemy.
 
 - [SQLAlchemy docs: Custom Types](https://docs.sqlalchemy.org/en/20/core/custom_types.html#custom-types)
 
@@ -16,13 +15,15 @@ class SomeModel(Base):
     ## Tell this model to convert uuid.UUID Python types to custom CompatibleUUID class
     type_annotation_map = {uuid.UUID: CompatibleUUID}
 """
+from __future__ import annotations
+
 from typing import Any
 import uuid
+
 from sqlalchemy import BINARY
-from sqlalchemy.types import TypeDecorator, CHAR
-from sqlalchemy.engine.interfaces import Dialect
 from sqlalchemy.dialects.postgresql import UUID
-
+from sqlalchemy.engine.interfaces import Dialect
+from sqlalchemy.types import CHAR, TypeDecorator
 
 class CompatibleUUID(TypeDecorator):
     """Define a custom UUID, overriding SQLAlchemy's UUId type.
diff --git a/red_utils/sqlalchemy_utils/utils.py b/red_utils/sqlalchemy_utils/utils.py
@@ -22,7 +22,6 @@
 from sqlalchemy.orm import Session, sessionmaker
 from sqlalchemy.schema import CreateTable
 
-
 def debug_metadata_obj(metadata_obj: sa.MetaData = None) -> None:
     """Debug-print a SQLAlchemy MetaData object.
 
diff --git a/red_utils/time_utils/operations.py b/red_utils/time_utils/operations.py
@@ -10,7 +10,6 @@
 
 from .constants import default_format, twelve_hour_format
 
-
 def datetime_as_str(ts: dt = None, format: str = default_format) -> str:
     """Convert a datetime.datetime object to a string.
 
diff --git a/red_utils/uuid_utils/classes.py b/red_utils/uuid_utils/classes.py
@@ -2,7 +2,6 @@
 
 from dataclasses import dataclass
 
-
 @dataclass
 class UUIDLength:
     """Simple dataclass to store UUID string lengths."""
