mirror of
https://github.com/Xevion/linkpulse.git
synced 2025-12-06 09:15:32 -06:00
reformat, organize imports
This commit is contained in:
@@ -59,7 +59,7 @@ def main(*args: str) -> None:
|
||||
# import most useful objects, models, and functions
|
||||
lp = linkpulse # alias
|
||||
from linkpulse.app import app
|
||||
from linkpulse.models import BaseModel, User, Session
|
||||
from linkpulse.models import BaseModel, Session, User
|
||||
from linkpulse.utilities import get_db
|
||||
|
||||
db = get_db()
|
||||
|
||||
@@ -30,9 +30,7 @@ async def lifespan(_: FastAPI) -> AsyncIterator[None]:
|
||||
db.connect()
|
||||
db.create_tables([models.User, models.Session])
|
||||
|
||||
FastAPICache.init(
|
||||
backend=InMemoryBackend(), prefix="fastapi-cache", cache_status_header="X-Cache"
|
||||
)
|
||||
FastAPICache.init(backend=InMemoryBackend(), prefix="fastapi-cache", cache_status_header="X-Cache")
|
||||
|
||||
scheduler.start()
|
||||
|
||||
|
||||
@@ -1,11 +1,12 @@
|
||||
import os
|
||||
from dataclasses import dataclass
|
||||
|
||||
import structlog
|
||||
from fastapi import HTTPException, Request, Response, status
|
||||
from limits.aio.strategies import MovingWindowRateLimiter
|
||||
from limits.aio.storage import MemoryStorage
|
||||
from limits import parse
|
||||
from limits.aio.storage import MemoryStorage
|
||||
from limits.aio.strategies import MovingWindowRateLimiter
|
||||
from linkpulse.models import Session
|
||||
from dataclasses import dataclass
|
||||
|
||||
storage = MemoryStorage()
|
||||
strategy = MovingWindowRateLimiter(storage)
|
||||
|
||||
@@ -31,9 +31,7 @@ def drop_color_message_key(_: Any, __: Any, event_dict: EventDict) -> EventDict:
|
||||
return event_dict
|
||||
|
||||
|
||||
def setup_logging(
|
||||
json_logs: Optional[bool] = None, log_level: Optional[str] = None
|
||||
) -> None:
|
||||
def setup_logging(json_logs: Optional[bool] = None, log_level: Optional[str] = None) -> None:
|
||||
# Pull from environment variables, apply defaults if not set
|
||||
json_logs = json_logs or os.getenv("LOG_JSON_FORMAT", "true").lower() == "true"
|
||||
log_level = log_level or os.getenv("LOG_LEVEL", "INFO")
|
||||
@@ -158,8 +156,6 @@ def setup_logging(
|
||||
sys.__excepthook__(exc_type, exc_value, exc_traceback)
|
||||
return
|
||||
|
||||
root_logger.error(
|
||||
"Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback)
|
||||
)
|
||||
root_logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
|
||||
|
||||
sys.excepthook = handle_exception
|
||||
|
||||
@@ -3,7 +3,7 @@ from typing import Annotated, Optional, Tuple
|
||||
|
||||
import structlog
|
||||
from fastapi import APIRouter, Depends, Response, status
|
||||
from linkpulse.dependencies import SessionDependency, RateLimiter, SessionModel
|
||||
from linkpulse.dependencies import RateLimiter, SessionDependency, SessionModel
|
||||
from linkpulse.models import Session, User
|
||||
from linkpulse.utilities import utc_now
|
||||
from pwdlib import PasswordHash
|
||||
|
||||
@@ -28,8 +28,6 @@ async def get_migration() -> dict[str, Any]:
|
||||
:rtype: dict[str, Any]
|
||||
"""
|
||||
# Kind of insecure, but this is just a demo thing to show that migratehistory is available.
|
||||
cursor = db.execute_sql(
|
||||
"SELECT name, migrated_at FROM migratehistory ORDER BY migrated_at DESC LIMIT 1"
|
||||
)
|
||||
cursor = db.execute_sql("SELECT name, migrated_at FROM migratehistory ORDER BY migrated_at DESC LIMIT 1")
|
||||
name, migrated_at = cursor.fetchone()
|
||||
return {"name": name, "migrated_at": migrated_at}
|
||||
|
||||
@@ -1,14 +1,14 @@
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
import pytest
|
||||
import structlog
|
||||
from fastapi import status
|
||||
from fastapi.testclient import TestClient
|
||||
from linkpulse.app import app
|
||||
from linkpulse.tests.test_session import expired_session, session
|
||||
from linkpulse.tests.test_user import user
|
||||
from linkpulse.tests.test_session import session, expired_session
|
||||
from linkpulse.utilities import utc_now
|
||||
|
||||
import pytest
|
||||
|
||||
logger = structlog.get_logger()
|
||||
|
||||
|
||||
|
||||
@@ -3,11 +3,10 @@ from datetime import timedelta
|
||||
import pytest
|
||||
import structlog
|
||||
from linkpulse.models import Session
|
||||
from linkpulse.routers.auth import validate_session
|
||||
from linkpulse.tests.random import random_string
|
||||
from linkpulse.tests.test_user import user
|
||||
from linkpulse.utilities import utc_now
|
||||
from linkpulse.routers.auth import validate_session
|
||||
|
||||
from peewee import IntegrityError
|
||||
|
||||
logger = structlog.get_logger()
|
||||
|
||||
Reference in New Issue
Block a user