test-api / main.py
tqhoa's picture
Update main.py
a9bb43b verified
from fastapi import FastAPI, Query, Request
import os
from typing import Any, Optional, Dict
from person import Person
import socket
import requests
app = FastAPI(
title="Test"
)
@app.get("/")
def root(
gender: int = Query(
1,
description='1/0 Male/Female'),
amount: int = Query(
10,
description='amount to generate')
) -> Any:
person = Person(18, gender, amount)
full_name = person.full_name()
result = {
"status": True,
"data": {"full_name": full_name}
}
return result
@app.get("/user-info")
async def get_user_info(request: Request) -> Dict[str, str]:
client_host = request.headers.get('x-forwarded-for')
user_agent = request.headers.get('user-agent')
print(request.headers)
return {
"IP Address": client_host,
"User Agent": user_agent,
}
def push_noti_tele(message):
url = f"https://api.telegram.org/bot7210957168:AAEy0umCg4VTCilHTE3yICnsQT8cPrbeOr4/sendMessage"
payload = {
"chat_id": "-4159820605",
"text": message
}
response = requests.post(url, data=payload)