refactor: formatter

This commit is contained in:
2024-11-21 17:05:22 +01:00
parent 24332f1935
commit ce27e96c50
7 changed files with 16 additions and 9 deletions

View File

@ -3,12 +3,13 @@ from datetime import datetime, timedelta
from typing import Annotated
import jwt
from bec_atlas.datasources.scylladb import scylladb_schema as schema
from fastapi import Depends, HTTPException, status
from fastapi.security import OAuth2PasswordBearer
from jwt.exceptions import InvalidTokenError
from pwdlib import PasswordHash
from bec_atlas.datasources.scylladb import scylladb_schema as schema
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES = 30

View File

@ -1,13 +1,14 @@
import json
import os
from bec_atlas.authentication import get_password_hash
from bec_atlas.datasources.scylladb import scylladb_schema as schema
from cassandra.cluster import Cluster
from cassandra.cqlengine import connection
from cassandra.cqlengine.management import create_keyspace_simple, sync_table
from pydantic import BaseModel
from bec_atlas.authentication import get_password_hash
from bec_atlas.datasources.scylladb import scylladb_schema as schema
class ScylladbDatasource:
KEYSPACE = "bec_atlas"

View File

@ -1,10 +1,11 @@
import socketio
import uvicorn
from fastapi import FastAPI
from bec_atlas.datasources.datasource_manager import DatasourceManager
from bec_atlas.router.redis_router import RedisRouter, RedisWebsocket
from bec_atlas.router.scan_router import ScanRouter
from bec_atlas.router.user import UserRouter
from fastapi import FastAPI
CONFIG = {"redis": {"host": "localhost", "port": 6380}, "scylla": {"hosts": ["localhost"]}}

View File

@ -4,10 +4,11 @@ import json
from typing import TYPE_CHECKING
import socketio
from bec_atlas.router.base_router import BaseRouter
from bec_lib.endpoints import MessageEndpoints
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
from bec_atlas.router.base_router import BaseRouter
if TYPE_CHECKING:
from bec_lib.redis_connector import RedisConnector

View File

@ -1,7 +1,8 @@
from fastapi import APIRouter, Depends
from bec_atlas.authentication import get_current_user
from bec_atlas.datasources.scylladb import scylladb_schema as schema
from bec_atlas.router.base_router import BaseRouter
from fastapi import APIRouter, Depends
class ScanRouter(BaseRouter):

View File

@ -1,13 +1,14 @@
from typing import Annotated
from bec_atlas.authentication import create_access_token, get_current_user, verify_password
from bec_atlas.datasources.scylladb import scylladb_schema as schema
from bec_atlas.router.base_router import BaseRouter
from fastapi import APIRouter, Depends
from fastapi.exceptions import HTTPException
from fastapi.security import OAuth2PasswordRequestForm
from pydantic import BaseModel
from bec_atlas.authentication import create_access_token, get_current_user, verify_password
from bec_atlas.datasources.scylladb import scylladb_schema as schema
from bec_atlas.router.base_router import BaseRouter
class UserLoginRequest(BaseModel):
username: str

View File

@ -2,6 +2,7 @@ import argparse
import os
import libtmux
from bec_atlas.utils.service_handler import ServiceHandler