main.py

from fastapi import FastAPI, Query
import random as r
import string

app = FastAPI()

# Define character sets
lower_charset = string.ascii_lowercase
upper_charset = string.ascii_uppercase
digitset = string.digits
symbolset = '!@#$%^&*()-_=+~:;{}[]/|<>?'

# Define character set options
options = {
    '1': ('Uppercase Letters Only', upper_charset),
    '2': ('Lowercase Letters Only', lower_charset),
    '3': ('Numbers Only', digitset),
    '4': ('Symbols Only', symbolset),
    '5': ('Uppercase and Lowercase Letters', upper_charset + lower_charset),
    '6': ('Uppercase Letters and Numbers', upper_charset + digitset),
    '7': ('Uppercase Letters and Symbols', upper_charset + symbolset),
    '8': ('Lowercase Letters and Numbers', lower_charset + digitset),
    '9': ('Lowercase Letters and Symbols', lower_charset + symbolset),
    '10': ('Numbers and Symbols', digitset + symbolset),
    '11': ('Symbols, Lowercase Letters, and Uppercase Letters', symbolset + lower_charset + upper_charset),
    '12': ('Symbols, Lowercase Letters, and Numbers', symbolset + lower_charset + digitset),
    '13': ('Symbols, Uppercase Letters, and Numbers', symbolset + upper_charset + digitset),
    '14': ('All Characters', upper_charset + lower_charset + digitset + symbolset)
}

@app.get("/generate-password")
def generate_password(char_type: str = Query(..., description="Character set option (1-14)"),
                     length: int = Query(..., description="Length of the password")):
    # Validate the character set choice
    if char_type not in options:
        return {"error": "Invalid choice. Please select a valid option."}

    # Generate the password
    possible_chars = options[char_type][1]
    generated_passwd = ''.join(r.choices(possible_chars, k=length))

    return {"generated_password": generated_passwd}