client.py

import requests
import os
from dotenv import load_dotenv

load_dotenv()

BASE_URL = "http://127.0.0.1:5000"

API_KEY = os.getenv("API_KEY")

HEADERS = {
    "X-API-Key": API_KEY
}


def api_request(method, endpoint, **kwargs):
    try:
        response = requests.request(method, f"{BASE_URL}{endpoint}", headers=HEADERS, timeout=5, **kwargs)
        return response

    except requests.exceptions.Timeout:
        print("Request timed out.")
        return None

    except requests.exceptions.ConnectionError:
        print("Could not connect to the API server.")
        return None
    
    
def handle_response(response):
    if response is None:
        return

    print("Status:", response.status_code)

    try:
        data = response.json()
        print("Data:", data)

    except ValueError:
        print("Response is not valid JSON.")

def get_users():
    response = api_request("GET","/users") 
    handle_response(response)

def get_user(user_id):
    response = api_request("GET",f"/users/{user_id}")
    handle_response(response)
        
    
def create_user(name, email):
    data = {
        "name": name,
        "email": email
    }
    
    response = api_request("POST","/users",json= data)
    handle_response(response)
        

def update_user(user_id, name, email):
    
    data = {
        "name": name,
        "email": email
    }
    
    response = api_request("PUT",f"/users/{user_id}",json=data)
    handle_response(response)

    
def patch_user(user_id, **data):
    response = api_request("PATCH",f"/users/{user_id}",json=data)

    handle_response(response)
    
  
def delete_user(user_id):
    response = api_request("DELETE",f"/users/{user_id}")
    handle_response(response)
    
def search_users(query):
    response = api_request("GET","/users/search",params={"q": query})
    handle_response(response)
    
   
    
while True:

    print("\n===== USER API CLIENT =====")
    print("1. Get All Users")
    print("2. Get User")
    print("3. Create User")
    print("4. Update User")
    print("5. Patch User")
    print("6. Delete User")
    print("7. Search Users")
    print("8. Exit")

    choice = input("Enter choice: ")

    if choice == "1":

        get_users()

    elif choice == "2":

        user_id = int(input("Enter User ID: "))
        get_user(user_id)

    elif choice == "3":

        name = input("Enter name: ")
        email = input("Enter email: ")

        create_user(name, email)

    elif choice == "4":

        user_id = int(input("Enter User ID: "))
        name = input("Enter name: ")
        email = input("Enter email: ")

        update_user(
            user_id,
            name,
            email
        )

    elif choice == "5":

        user_id = int(input("Enter User ID: "))

        name = input("Enter new name (press Enter to skip): ")
        email = input("Enter new email (press Enter to skip): ")

        data = {}

        if name:
            data["name"] = name

        if email:
            data["email"] = email

        patch_user(
            user_id,
            **data
        )

    elif choice == "6":

        user_id = int(input("Enter User ID: "))
        delete_user(user_id)

    elif choice == "7":

        query = input("Enter search text: ")
        search_users(query)
    
    elif choice == "8":

        print("Exiting...")
        break

    else:

        print("Invalid choice.")