from datetime import datetime, UTC import os from uuid import UUID from ocpp.routing import on from ocpp.v201 import ChargePoint as cp from ocpp.v201 import call_result from ocpp.v201.datatypes import IdTokenInfoType, IdTokenType from ocpp.v201.enums import Action, RegistrationStatusType, AuthorizationStatusType, IdTokenType as IdTokenEnumType, TransactionEventType from app.database import SessionLocal from app.models.chargepoint import ChargePoint as DbChargePoint from app.models.connector import Connector as DbConnector from app.models.id_token import IdToken as DbIdToken from app.models.transaction import Transaction as DbTransaction from app.models.meter_value import MeterValue as DbMeterValue from app.schemas.connector import ConnectorStatus from app.schemas.transaction import TransactionStatus, TransactionEventTriggerReason from app.schemas.meter_value import Measurand, PhaseType class ChargePoint(cp): async def __update_last_seen(self): with SessionLocal() as db: db_chargepoint = db.query(DbChargePoint).filter(DbChargePoint.identity == self.id).first() db_chargepoint.last_seen = datetime.now(UTC) db.commit() async def __get_id_token_info(self, id_token) -> IdTokenInfoType: if id_token["type"] not in ["ISO14443", "ISO15693"]: return IdTokenInfoType( status=AuthorizationStatusType.invalid ) with SessionLocal() as db: db_id_token = db.query(DbIdToken).filter(DbIdToken.token == id_token["id_token"]).first() if db_id_token == None: return IdTokenInfoType( status=AuthorizationStatusType.unknown ) if db_id_token.is_active == False: id_token_info=IdTokenInfoType( status=AuthorizationStatusType.blocked ) else: id_token_info=IdTokenInfoType( status=AuthorizationStatusType.accepted, group_id_token=IdTokenType( type=IdTokenEnumType.central, id_token=str(db_id_token.owner_id) ) ) return id_token_info @on(Action.BootNotification) async def on_boot_notification(self, charging_station, **kwargs): with SessionLocal() as db: db_chargepoint = db.query(DbChargePoint).filter(DbChargePoint.identity == self.id).first() db_chargepoint.last_seen = datetime.now(UTC) for key in charging_station.keys(): if key in db_chargepoint.__dict__: setattr(db_chargepoint, key, charging_station[key]) db.commit() return call_result.BootNotificationPayload( current_time=datetime.now(UTC).isoformat(), interval=int(os.getenv("CS_HEARTBEAT_INTERVAL", "1800")), status=RegistrationStatusType.accepted ) @on(Action.Heartbeat) async def on_heartbeat_request(self): await self.__update_last_seen() return call_result.HeartbeatPayload( current_time=datetime.now(UTC).isoformat() ) @on(Action.StatusNotification) async def on_status_notification(self, evse_id: int, connector_id: int, connector_status: str, **kwargs): with SessionLocal() as db: db_chargepoint = db.query(DbChargePoint).filter(DbChargePoint.identity == self.id).first() db_chargepoint.last_seen = datetime.now(UTC) db_connector = db.query(DbConnector).filter( DbConnector.chargepoint_id == db_chargepoint.id, DbConnector.evse == evse_id, DbConnector.index == connector_id ).first() if db_connector == None: db_connector = DbConnector( chargepoint_id = db_chargepoint.id, evse = evse_id, index = connector_id, status = ConnectorStatus(connector_status) ) db.add(db_connector) else: db_connector.status = ConnectorStatus(connector_status) db.commit() return call_result.StatusNotificationPayload() @on(Action.Authorize) async def on_authorize(self, id_token, **kwargs): await self.__update_last_seen() id_token_info = await self.__get_id_token_info(id_token) return call_result.AuthorizePayload(id_token_info) @on(Action.TransactionEvent) async def on_transaction_event( self, event_type, timestamp, trigger_reason, transaction_info, id_token, meter_value, **kwargs ): if id_token != None: id_token_info = await self.__get_id_token_info(id_token) with SessionLocal() as db: chargepoint = db.query(DbChargePoint).filter(DbChargePoint.identity == self.id).first() chargepoint.last_seen = datetime.now(UTC) if event_type == str(TransactionEventType.started): meter_start=0 if meter_value != None: for meter_value_entry in meter_value: for sampled_value in meter_value_entry['sampled_value']: if "measurand" in sampled_value.keys(): if sampled_value['measurand'] == str(Measurand.ENERGY_ACTIVE_IMPORT_REGISTER): meter_start = sampled_value['value'] else: meter_start = sampled_value['value'] transaction = DbTransaction( id=transaction_info["transaction_id"], status=TransactionStatus.ONGOING, started_at=datetime.fromisoformat(timestamp), meter_start=meter_start, price=chargepoint.price, chargepoint_id=chargepoint.id ) if id_token != None: if id_token_info.status == AuthorizationStatusType.accepted: transaction.user_id = UUID(id_token_info.group_id_token.id_token) db.add(transaction) elif event_type == str(TransactionEventType.updated): transaction = db.get(DbTransaction, transaction_info["transaction_id"]) if meter_value != None: for meter_value_entry in meter_value: timestamp = datetime.fromisoformat(meter_value_entry['timestamp']) for sampled_value in meter_value_entry['sampled_value']: db_meter_value = DbMeterValue() db_meter_value.transaction_id = transaction.id db_meter_value.timestamp = timestamp if "measurand" in sampled_value.keys(): db_meter_value.measurand = Measurand(sampled_value['measurand']) else: db_meter_value.measurand = Measurand.ENERGY_ACTIVE_IMPORT_REGISTER if "phase" in sampled_value.keys(): db_meter_value.phase_type = PhaseType(sampled_value['phase']) if "unit_of_measure" in sampled_value.keys(): if "unit" in sampled_value['unit_of_measure']: db_meter_value.unit = sampled_value['unit_of_measure']['unit'] else: db_meter_value.unit = "Wh" db_meter_value.value = sampled_value['value'] db.add(db_meter_value) if id_token != None: if id_token_info.status == AuthorizationStatusType.accepted: transaction.user_id = UUID(id_token_info.group_id_token.id_token) elif event_type == str(TransactionEventType.ended): transaction = db.get(DbTransaction, transaction_info["transaction_id"]) transaction.status = TransactionStatus.ENDED transaction.ended_at = datetime.fromisoformat(timestamp) transaction.end_reason = TransactionEventTriggerReason(trigger_reason) meter_end=0 if meter_value != None: for meter_value_entry in meter_value: for sampled_value in meter_value_entry['sampled_value']: if "measurand" in sampled_value.keys(): if sampled_value['measurand'] == str(Measurand.ENERGY_ACTIVE_IMPORT_REGISTER): meter_end = sampled_value['value'] else: meter_end = sampled_value['value'] transaction.meter_end = meter_end if id_token != None: if id_token_info.status == AuthorizationStatusType.accepted: transaction.user_id = UUID(id_token_info.group_id_token.id_token) db.commit() if id_token != None: return call_result.TransactionEventPayload(id_token_info=id_token_info) return call_result.TransactionEventPayload()