Implement transaction handling

app/routers/chargepoint_v1.py

@@ -1,5 +1,6 @@
 import random
 import string
+from datetime import datetime, timedelta, UTC
 from uuid import UUID
 from fastapi import APIRouter, HTTPException, Security
 from fastapi.params import Depends
@@ -18,7 +19,9 @@ from app.schemas.chargepoint import (
     ChargePointResetRequest,
     ChargePointResetResponse
 )
+from app.schemas.id_token import IdTokenLearnRequest, IdTokenLearnResponse
 from app.models.chargepoint import ChargePoint as DbChargePoint
+from app.models.user import User as DbUser
 from app.security import get_api_key
 
 router = APIRouter(
@@ -77,9 +80,10 @@ async def create_chargepoint(
     db: Session = Depends(get_db)
 ):
     chargepoint_db = DbChargePoint(
-        friendly_name=chargepoint.friendly_name,
+        identity=chargepoint.identity,
         is_active=chargepoint.is_active,
-        password=''.join(random.choice(string.ascii_letters + string.digits) for i in range(24))
+        password=''.join(random.choice(string.ascii_letters + string.digits) for i in range(24)),
+        price=chargepoint.price
     )
     db.add(chargepoint_db)
     db.commit()
@@ -139,3 +143,73 @@ async def reset_chargepoint(
         return ChargePointResetResponse(status=response.status)
     except TimeoutError:
         raise HTTPException(status_code=503, detail="Chargepoint didn't respond in time.")
+    
+@router.post(path="/{chargepoint_id}/token-learning", status_code=201, response_model=IdTokenLearnResponse)
+async def create_id_token_learn_request(
+    chargepoint_id: UUID,
+    learn_request: IdTokenLearnRequest,
+    api_key: str = Security(get_api_key),
+    db: Session = Depends(get_db)
+):
+    chargepoint = db.get(DbChargePoint, chargepoint_id)
+    if chargepoint is None:
+        raise HTTPException(status_code=404, detail="Chargepoint not found")
+    
+    owner = db.get(DbUser, learn_request.user_id)
+    if owner == None:
+        raise HTTPException(status_code=422, detail=[{
+            "loc": ["body", "user_id"],
+            "msg": "Target user not found",
+            "type": "invalid_relation" 
+        }])
+    
+    chargepoint.learn_user_id = learn_request.user_id
+
+    if learn_request.until == None:
+        chargepoint.learn_until = datetime.now(UTC) + timedelta(minutes=5)
+    else:
+        chargepoint.learn_until = learn_request.until
+
+    db.commit()
+
+    return IdTokenLearnResponse(
+        user_id=chargepoint.learn_user_id,
+        until=chargepoint.learn_until
+    )
+
+@router.get(path="/{chargepoint_id}/token-learning", response_model=IdTokenLearnResponse)
+async def get_id_token_learn_request(
+    chargepoint_id: UUID,
+    api_key: str = Security(get_api_key),
+    db: Session = Depends(get_db)
+):
+    chargepoint = db.get(DbChargePoint, chargepoint_id)
+    if chargepoint is None:
+        raise HTTPException(status_code=404, detail="Chargepoint not found")
+    
+    if chargepoint.learn_user_id == None:
+        raise HTTPException(status_code=404, detail="No active learning request")
+
+    return IdTokenLearnResponse(
+        user_id=chargepoint.learn_user_id,
+        until=chargepoint.learn_until
+    )
+
+@router.delete(path="/{chargepoint_id}/token-learning", response_model=[])
+async def get_id_token_learn_request(
+    chargepoint_id: UUID,
+    api_key: str = Security(get_api_key),
+    db: Session = Depends(get_db)
+):
+    chargepoint = db.get(DbChargePoint, chargepoint_id)
+    if chargepoint is None:
+        raise HTTPException(status_code=404, detail="Chargepoint not found")
+    
+    if chargepoint.learn_user_id == None:
+        raise HTTPException(status_code=404, detail="No active learning request")
+    
+    chargepoint.learn_user_id = None
+    chargepoint.learn_until = None
+    db.commit()
+
+    return []