update charge api endpoint

Author: surajit

crud/charge.py

@@ -3,13 +3,17 @@
 
 from models import payment
 
+
 class ChargeExistException(Exception):
     pass
 
 
-async def add_charge(db: Session, charge):
+async def add_charge(db: Session, charge, org_id: int):
     try:
-        db_item = payment.Charge(**charge)
+        charge_dict = charge.dict()
+        del charge_dict["company_name"]
+        charge_dict["org_id"] = org_id
+        db_item = payment.Charge(**charge_dict)
         db.add(db_item)
         db.commit()
         db.refresh(db_item)

routes/v1/charge.py

@@ -11,6 +11,7 @@
 from logger.log import save_log
 from dependencies.dependencies import get_db
 from crud import charge as _charge
+from schemas.schema import CreateCharge
 
 router = APIRouter(
     prefix="/api/v1",
@@ -21,19 +22,11 @@
 
 @router.post("/charges/")
 @save_log
-async def create_charge(request: Request, db: Session = Depends(get_db)):
+async def create_charge(request: Request, charge: CreateCharge, db: Session = Depends(get_db)):
     try:
-        data = await request.json()
-        company_name = data["company_name"]
+        company_name = charge.company_name
         company = filter_company_by_name(db, company_name).all()[0]
-        company_data = dict(
-            currency=data["currency"],
-            amount=data["amount"],
-            charge_type=data["charge_type"],
-            org_id=company.id,
-            created_at=datetime.datetime.now(),
-        )
-        response = await _charge.add_charge(db, company_data)
+        response = await _charge.add_charge(db, charge, company.id)
     except ChargeExistException as exc:
         raise HTTPException(status_code=200, detail=str(exc))
     return response

schemas/schema.py

@@ -20,3 +20,15 @@ class CreateContact(ContactBase):
 
 class CreateCompany(CompanyBase):
     pass
+
+
+class ChargeBase(BaseModel):
+    currency: str
+    amount: int
+    charge_type: str
+    company_name: str
+
+
+class CreateCharge(ChargeBase):
+    pass
+