58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import logging
|
|
import uvicorn
|
|
|
|
from database import engine, Base
|
|
from dotenv import load_dotenv
|
|
from fastapi import FastAPI, Request, status
|
|
from fastapi.responses import JSONResponse
|
|
from loggerino import timestamp_log_config
|
|
from os import getenv
|
|
from schemas import SMSMessage
|
|
from service import inbound_sms_handler, retrieve_sms_messages_by_phone_number
|
|
from sqlalchemy.orm import Session, joinedload
|
|
from uvicorn.config import LOGGING_CONFIG
|
|
|
|
|
|
# Create instance of FastAPI
|
|
app = FastAPI()
|
|
|
|
# Load dotenv file (.env)
|
|
load_dotenv()
|
|
|
|
# Load custom logger format
|
|
logger = logging.getLogger("uvicorn.error")
|
|
|
|
# Create tables (if they don't exist)
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
# Load whitelisted IPs
|
|
WHITELISTED_IPS = getenv("WHITELISTED_IPS").split(',')
|
|
|
|
|
|
@app.middleware("http")
|
|
async def validate_ip(request: Request, call_next):
|
|
ip = str(request.headers.get("x-forwarded-for", str(request.client.host)))
|
|
|
|
if ip not in WHITELISTED_IPS:
|
|
data = {"message": f"IP {ip} is not allowed to access this resource."}
|
|
return JSONResponse(status_code=status.HTTP_400_BAD_REQUEST, content=data)
|
|
|
|
return await call_next(request)
|
|
|
|
|
|
@app.post("/sms-message", status_code=status.HTTP_200_OK)
|
|
async def receive_sms_message(message: SMSMessage):
|
|
inbound_sms_handler(message)
|
|
|
|
|
|
@app.get('/sms-message/{number}')
|
|
def get_sms_messages_by_phone_number(number: str, limit: int = 10, page: int = 1):
|
|
contact = retrieve_sms_messages_by_phone_number(number, limit, page)
|
|
return {'status': 'success', 'results': len(contact.messages), 'response': contact}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run(
|
|
app, host="0.0.0.0", port=8000, log_config=timestamp_log_config(LOGGING_CONFIG)
|
|
)
|