159 lines
5.0 KiB
Python
159 lines
5.0 KiB
Python
"""
|
|
POS Integration Service
|
|
Handles integration with external POS systems (Square, Toast, Lightspeed)
|
|
"""
|
|
|
|
import time
|
|
from fastapi import FastAPI, Request
|
|
from app.core.config import settings
|
|
from app.api import pos_config, webhooks, sync
|
|
from app.core.database import database_manager
|
|
from shared.service_base import StandardFastAPIService
|
|
|
|
|
|
class POSService(StandardFastAPIService):
|
|
"""POS Integration Service with standardized setup"""
|
|
|
|
def __init__(self):
|
|
# Define expected database tables for health checks
|
|
pos_expected_tables = [
|
|
'pos_configurations', 'pos_transactions', 'pos_transaction_items',
|
|
'pos_webhook_logs', 'pos_sync_logs'
|
|
]
|
|
|
|
# Define custom metrics for POS service
|
|
pos_custom_metrics = {
|
|
"pos_webhooks_received_total": {
|
|
"type": "counter",
|
|
"description": "Total POS webhooks received",
|
|
"labels": ["provider", "event_type"]
|
|
},
|
|
"pos_sync_jobs_total": {
|
|
"type": "counter",
|
|
"description": "Total POS sync jobs",
|
|
"labels": ["provider", "status"]
|
|
},
|
|
"pos_transactions_synced_total": {
|
|
"type": "counter",
|
|
"description": "Total transactions synced",
|
|
"labels": ["provider"]
|
|
},
|
|
"pos_webhook_processing_duration_seconds": {
|
|
"type": "histogram",
|
|
"description": "Time spent processing webhooks"
|
|
},
|
|
"pos_sync_duration_seconds": {
|
|
"type": "histogram",
|
|
"description": "Time spent syncing data"
|
|
}
|
|
}
|
|
|
|
super().__init__(
|
|
service_name="pos-service",
|
|
app_name="POS Integration Service",
|
|
description="Handles integration with external POS systems",
|
|
version="1.0.0",
|
|
cors_origins=settings.CORS_ORIGINS,
|
|
api_prefix="/api/v1",
|
|
database_manager=database_manager,
|
|
expected_tables=pos_expected_tables,
|
|
custom_metrics=pos_custom_metrics
|
|
)
|
|
|
|
async def on_startup(self, app: FastAPI):
|
|
"""Custom startup logic for POS service"""
|
|
# Custom startup completed
|
|
self.logger.info("POS Integration Service started successfully")
|
|
|
|
async def on_shutdown(self, app: FastAPI):
|
|
"""Custom shutdown logic for POS service"""
|
|
# Database cleanup is handled by the base class
|
|
pass
|
|
|
|
def get_service_features(self):
|
|
"""Return POS-specific features"""
|
|
return [
|
|
"pos_integration",
|
|
"square_support",
|
|
"toast_support",
|
|
"lightspeed_support",
|
|
"webhook_handling",
|
|
"transaction_sync",
|
|
"real_time_updates"
|
|
]
|
|
|
|
def setup_custom_middleware(self):
|
|
"""Setup custom middleware for POS service"""
|
|
# Middleware for request logging and timing
|
|
@self.app.middleware("http")
|
|
async def log_requests(request: Request, call_next):
|
|
start_time = time.time()
|
|
|
|
# Log request
|
|
self.logger.info(
|
|
"Incoming request",
|
|
method=request.method,
|
|
url=str(request.url),
|
|
client_ip=request.client.host if request.client else None
|
|
)
|
|
|
|
response = await call_next(request)
|
|
|
|
# Log response
|
|
process_time = time.time() - start_time
|
|
self.logger.info(
|
|
"Request completed",
|
|
method=request.method,
|
|
url=str(request.url),
|
|
status_code=response.status_code,
|
|
process_time=f"{process_time:.4f}s"
|
|
)
|
|
|
|
response.headers["X-Process-Time"] = str(process_time)
|
|
return response
|
|
|
|
def setup_custom_endpoints(self):
|
|
"""Setup custom endpoints for POS service"""
|
|
@self.app.get("/")
|
|
async def root():
|
|
"""Root endpoint"""
|
|
return {
|
|
"service": "POS Integration Service",
|
|
"version": "1.0.0",
|
|
"status": "running",
|
|
"supported_pos_systems": ["square", "toast", "lightspeed"]
|
|
}
|
|
|
|
|
|
# Create service instance
|
|
service = POSService()
|
|
|
|
# Create FastAPI app with standardized setup
|
|
app = service.create_app(
|
|
docs_url="/docs" if settings.ENVIRONMENT != "production" else None,
|
|
redoc_url="/redoc" if settings.ENVIRONMENT != "production" else None
|
|
)
|
|
|
|
# Setup standard endpoints
|
|
service.setup_standard_endpoints()
|
|
|
|
# Setup custom middleware
|
|
service.setup_custom_middleware()
|
|
|
|
# Setup custom endpoints
|
|
service.setup_custom_endpoints()
|
|
|
|
# Include routers
|
|
service.add_router(pos_config.router, tags=["pos-config"])
|
|
service.add_router(webhooks.router, tags=["webhooks"])
|
|
service.add_router(sync.router, tags=["sync"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"main:app",
|
|
host="0.0.0.0",
|
|
port=8000,
|
|
reload=True
|
|
) |