-
-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathpayment_factory.py
More file actions
30 lines (24 loc) · 1.04 KB
/
payment_factory.py
File metadata and controls
30 lines (24 loc) · 1.04 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from fastapi import FastAPI, HTTPException
class PaymentProcessor:
def process_payment(self, amount: float):
pass
class CreditCardProcessor(PaymentProcessor):
def process_payment(self, amount: float):
return f"Processing credit card payment for {amount} dollars."
class PayPalProcessor(PaymentProcessor):
def process_payment(self, amount: float):
return f"Processing PayPal payment for {amount} dollars."
class PaymentProcessorFactory:
@staticmethod
def create_processor(processor_type: str) -> PaymentProcessor:
if processor_type == "credit_card":
return CreditCardProcessor()
elif processor_type == "paypal":
return PayPalProcessor()
else:
raise HTTPException(status_code=400, detail="Invalid payment processor type.")
app = FastAPI()
@app.post("/process_payment/{processor_type}")
def process_payment(processor_type: str, amount: float):
processor = PaymentProcessorFactory.create_processor(processor_type)
return processor.process_payment(amount)