mirror of
https://github.com/bec-project/bec_atlas.git
synced 2025-07-14 07:01:48 +02:00
refactor: formatter
This commit is contained in:
@ -3,12 +3,13 @@ from datetime import datetime, timedelta
|
|||||||
from typing import Annotated
|
from typing import Annotated
|
||||||
|
|
||||||
import jwt
|
import jwt
|
||||||
from bec_atlas.datasources.scylladb import scylladb_schema as schema
|
|
||||||
from fastapi import Depends, HTTPException, status
|
from fastapi import Depends, HTTPException, status
|
||||||
from fastapi.security import OAuth2PasswordBearer
|
from fastapi.security import OAuth2PasswordBearer
|
||||||
from jwt.exceptions import InvalidTokenError
|
from jwt.exceptions import InvalidTokenError
|
||||||
from pwdlib import PasswordHash
|
from pwdlib import PasswordHash
|
||||||
|
|
||||||
|
from bec_atlas.datasources.scylladb import scylladb_schema as schema
|
||||||
|
|
||||||
ALGORITHM = "HS256"
|
ALGORITHM = "HS256"
|
||||||
ACCESS_TOKEN_EXPIRE_MINUTES = 30
|
ACCESS_TOKEN_EXPIRE_MINUTES = 30
|
||||||
|
|
||||||
|
@ -1,13 +1,14 @@
|
|||||||
import json
|
import json
|
||||||
import os
|
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.cluster import Cluster
|
||||||
from cassandra.cqlengine import connection
|
from cassandra.cqlengine import connection
|
||||||
from cassandra.cqlengine.management import create_keyspace_simple, sync_table
|
from cassandra.cqlengine.management import create_keyspace_simple, sync_table
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from bec_atlas.authentication import get_password_hash
|
||||||
|
from bec_atlas.datasources.scylladb import scylladb_schema as schema
|
||||||
|
|
||||||
|
|
||||||
class ScylladbDatasource:
|
class ScylladbDatasource:
|
||||||
KEYSPACE = "bec_atlas"
|
KEYSPACE = "bec_atlas"
|
||||||
|
@ -1,10 +1,11 @@
|
|||||||
import socketio
|
import socketio
|
||||||
import uvicorn
|
import uvicorn
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
from bec_atlas.datasources.datasource_manager import DatasourceManager
|
from bec_atlas.datasources.datasource_manager import DatasourceManager
|
||||||
from bec_atlas.router.redis_router import RedisRouter, RedisWebsocket
|
from bec_atlas.router.redis_router import RedisRouter, RedisWebsocket
|
||||||
from bec_atlas.router.scan_router import ScanRouter
|
from bec_atlas.router.scan_router import ScanRouter
|
||||||
from bec_atlas.router.user import UserRouter
|
from bec_atlas.router.user import UserRouter
|
||||||
from fastapi import FastAPI
|
|
||||||
|
|
||||||
CONFIG = {"redis": {"host": "localhost", "port": 6380}, "scylla": {"hosts": ["localhost"]}}
|
CONFIG = {"redis": {"host": "localhost", "port": 6380}, "scylla": {"hosts": ["localhost"]}}
|
||||||
|
|
||||||
|
@ -4,10 +4,11 @@ import json
|
|||||||
from typing import TYPE_CHECKING
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
import socketio
|
import socketio
|
||||||
from bec_atlas.router.base_router import BaseRouter
|
|
||||||
from bec_lib.endpoints import MessageEndpoints
|
from bec_lib.endpoints import MessageEndpoints
|
||||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||||||
|
|
||||||
|
from bec_atlas.router.base_router import BaseRouter
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from bec_lib.redis_connector import RedisConnector
|
from bec_lib.redis_connector import RedisConnector
|
||||||
|
|
||||||
|
@ -1,7 +1,8 @@
|
|||||||
|
from fastapi import APIRouter, Depends
|
||||||
|
|
||||||
from bec_atlas.authentication import get_current_user
|
from bec_atlas.authentication import get_current_user
|
||||||
from bec_atlas.datasources.scylladb import scylladb_schema as schema
|
from bec_atlas.datasources.scylladb import scylladb_schema as schema
|
||||||
from bec_atlas.router.base_router import BaseRouter
|
from bec_atlas.router.base_router import BaseRouter
|
||||||
from fastapi import APIRouter, Depends
|
|
||||||
|
|
||||||
|
|
||||||
class ScanRouter(BaseRouter):
|
class ScanRouter(BaseRouter):
|
||||||
|
@ -1,13 +1,14 @@
|
|||||||
from typing import Annotated
|
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 import APIRouter, Depends
|
||||||
from fastapi.exceptions import HTTPException
|
from fastapi.exceptions import HTTPException
|
||||||
from fastapi.security import OAuth2PasswordRequestForm
|
from fastapi.security import OAuth2PasswordRequestForm
|
||||||
from pydantic import BaseModel
|
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):
|
class UserLoginRequest(BaseModel):
|
||||||
username: str
|
username: str
|
||||||
|
@ -2,6 +2,7 @@ import argparse
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
import libtmux
|
import libtmux
|
||||||
|
|
||||||
from bec_atlas.utils.service_handler import ServiceHandler
|
from bec_atlas.utils.service_handler import ServiceHandler
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user