Compare commits

..

2 commits

4 changed files with 36 additions and 8 deletions

View file

@ -24,7 +24,7 @@ async def call(
) -> Coroutine[Any, Any, Any | None]: ) -> Coroutine[Any, Any, Any | None]:
try: try:
cp = __active_connections[chargepoint_id] cp = __active_connections[chargepoint_id]
return cp.call(payload, suppress, unique_id) return await cp.call(payload, suppress, unique_id)
except KeyError as e: except KeyError as e:
raise e raise e

View file

@ -5,9 +5,19 @@ from fastapi import APIRouter, HTTPException, Security
from fastapi.params import Depends from fastapi.params import Depends
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from ocpp.v201.call import ResetPayload
from app.database import get_db from app.database import get_db
from app.ocpp_proto import chargepoint_manager from app.ocpp_proto import chargepoint_manager
from app.schemas.chargepoint import ChargePoint, ChargePointCreate, ChargePointUpdate, ChargePointPassword, ChargePointConnectionInfo from app.schemas.chargepoint import (
ChargePoint,
ChargePointCreate,
ChargePointUpdate,
ChargePointPassword,
ChargePointConnectionInfo,
ChargePointResetRequest,
ChargePointResetResponse
)
from app.models.chargepoint import ChargePoint as DbChargePoint from app.models.chargepoint import ChargePoint as DbChargePoint
from app.security import get_api_key from app.security import get_api_key
@ -112,3 +122,17 @@ async def get_chargepoint_status(
return ChargePointConnectionInfo( return ChargePointConnectionInfo(
connected=chargepoint_manager.is_connected(chargepoint_id) connected=chargepoint_manager.is_connected(chargepoint_id)
) )
@router.post(path="/{chargepoint_id}/reset", response_model=ChargePointResetResponse)
async def reset_chargepoint(
chargepoint_id: UUID,
reset_request: ChargePointResetRequest,
api_key: str = Security(get_api_key)
):
if chargepoint_manager.is_connected(chargepoint_id) == False:
raise HTTPException(status_code=423, detail="Chargepoint not connected")
response = await chargepoint_manager.call(
chargepoint_id,
payload=ResetPayload(type=reset_request.type, evse_id=reset_request.evse_id)
)
return ChargePointResetResponse(status=response.status)

View file

@ -5,6 +5,8 @@ from pydantic import BaseModel
from app.schemas.connector import Connector from app.schemas.connector import Connector
from ocpp.v201.enums import ResetType, ResetStatusType
class ChargePointBase(BaseModel): class ChargePointBase(BaseModel):
friendly_name: str friendly_name: str
is_active: bool is_active: bool
@ -33,3 +35,10 @@ class ChargePointPassword(BaseModel):
class ChargePointConnectionInfo(BaseModel): class ChargePointConnectionInfo(BaseModel):
connected: bool connected: bool
class ChargePointResetRequest(BaseModel):
type: ResetType
evse_id: Optional[int] = None
class ChargePointResetResponse(BaseModel):
status: ResetStatusType

View file

@ -1,6 +1,5 @@
import base64 import base64
import binascii import binascii
from uuid import UUID
from starlette.authentication import ( from starlette.authentication import (
AuthCredentials, AuthenticationBackend, AuthenticationError, SimpleUser AuthCredentials, AuthenticationBackend, AuthenticationError, SimpleUser
) )
@ -23,13 +22,9 @@ class BasicAuthBackend(AuthenticationBackend):
raise AuthenticationError('Invalid basic auth credentials') raise AuthenticationError('Invalid basic auth credentials')
username, _, password = decoded.partition(":") username, _, password = decoded.partition(":")
try:
id = UUID(username)
except (ValueError) as exc:
raise AuthenticationError('Invalid basic auth credentials')
with SessionLocal() as db: with SessionLocal() as db:
chargepoint = db.query(ChargePoint).filter(ChargePoint.id == id).first() chargepoint = db.query(ChargePoint).filter(ChargePoint.friendly_name == username).first()
if chargepoint is None: if chargepoint is None:
raise AuthenticationError('Invalid basic auth credentials') raise AuthenticationError('Invalid basic auth credentials')
if chargepoint.password != password: if chargepoint.password != password: