knight
a2a485dd8e
- Integrated express-basic-auth middleware in index.js to secure non-GET routes with basic authentication. - Updated libraryManager.py to use HTTPBasicAuth for API requests, enhancing security for book management operations. - Modified public/index.html to improve the user interface with a new search feature and dynamic book table. - Removed obsolete public/library.html file to streamline the project structure. - Updated package.json and package-lock.json to include express-basic-auth as a new dependency.
128 lines
4.4 KiB
Python
128 lines
4.4 KiB
Python
import requests
|
|
from rich.console import Console
|
|
from rich.prompt import Prompt
|
|
from rich.table import Table
|
|
from fuzzywuzzy import process
|
|
from requests.auth import HTTPBasicAuth
|
|
|
|
API_BASE_URL = "https://localhost:3000" # Replace with your actual API base URL
|
|
console = Console()
|
|
|
|
# Use environment variables or a secure method to store credentials
|
|
USERNAME = 'admin'
|
|
PASSWORD = 'library@123' # Replace with your actual password
|
|
|
|
def list_books():
|
|
response = requests.get(f"{API_BASE_URL}/api/books-with-images", verify=False)
|
|
if response.status_code == 200:
|
|
books = response.json()
|
|
table = Table(title="Books")
|
|
table.add_column("ISBN", justify="right", style="cyan", no_wrap=True)
|
|
table.add_column("Title", style="magenta")
|
|
table.add_column("Authors", style="green")
|
|
for book in books:
|
|
table.add_row(book['isbn'], book['title'], book['authors'])
|
|
console.print(table)
|
|
else:
|
|
console.print("Failed to fetch books.", style="bold red")
|
|
|
|
def add_book():
|
|
isbn = Prompt.ask("Enter ISBN")
|
|
title = Prompt.ask("Enter Title")
|
|
authors = Prompt.ask("Enter Authors")
|
|
data = {
|
|
"isbn": isbn,
|
|
"title": title,
|
|
"authors": authors
|
|
}
|
|
response = requests.post(
|
|
f"{API_BASE_URL}/store-book",
|
|
json=data,
|
|
verify=False,
|
|
auth=HTTPBasicAuth(USERNAME, PASSWORD)
|
|
)
|
|
if response.status_code == 200:
|
|
console.print("Book added successfully.", style="bold green")
|
|
else:
|
|
console.print("Failed to add book.", style="bold red")
|
|
|
|
def remove_book():
|
|
isbn = Prompt.ask("Enter ISBN of the book to remove")
|
|
response = requests.delete(
|
|
f"{API_BASE_URL}/book/{isbn}",
|
|
verify=False,
|
|
auth=HTTPBasicAuth(USERNAME, PASSWORD)
|
|
)
|
|
if response.status_code == 200:
|
|
console.print("Book removed successfully.", style="bold green")
|
|
else:
|
|
console.print("Failed to remove book.", style="bold red")
|
|
|
|
def change_book_status():
|
|
isbn = Prompt.ask("Enter ISBN of the book to change status")
|
|
status = Prompt.ask("Enter new status (e.g., Available, Checked Out)")
|
|
data = {"status": status}
|
|
response = requests.put(
|
|
f"{API_BASE_URL}/book/{isbn}",
|
|
json=data,
|
|
verify=False,
|
|
auth=HTTPBasicAuth(USERNAME, PASSWORD)
|
|
)
|
|
if response.status_code == 200:
|
|
console.print("Book status updated successfully.", style="bold green")
|
|
else:
|
|
console.print("Failed to update book status.", style="bold red")
|
|
|
|
def search_books():
|
|
query = Prompt.ask("Enter search query")
|
|
response = requests.get(f"{API_BASE_URL}/api/books-with-images", verify=False)
|
|
if response.status_code == 200:
|
|
books = response.json()
|
|
book_titles = [book['title'] for book in books]
|
|
matches = process.extract(query, book_titles, limit=5)
|
|
|
|
if matches:
|
|
table = Table(title="Search Results")
|
|
table.add_column("ISBN", justify="right", style="cyan", no_wrap=True)
|
|
table.add_column("Title", style="magenta")
|
|
table.add_column("Authors", style="green")
|
|
table.add_column("Score", justify="right", style="yellow")
|
|
|
|
for match in matches:
|
|
title, score = match
|
|
book = next(book for book in books if book['title'] == title)
|
|
table.add_row(book['isbn'], book['title'], book['authors'], str(score))
|
|
|
|
console.print(table)
|
|
else:
|
|
console.print("No matches found.", style="bold red")
|
|
else:
|
|
console.print("Failed to fetch books.", style="bold red")
|
|
|
|
def main():
|
|
while True:
|
|
console.print("\n[bold]Admin Console[/bold]")
|
|
console.print("1. List Books")
|
|
console.print("2. Add Book")
|
|
console.print("3. Remove Book")
|
|
console.print("4. Change Book Status")
|
|
console.print("5. Search Books")
|
|
console.print("6. Exit")
|
|
choice = Prompt.ask("Choose an option", choices=["1", "2", "3", "4", "5", "6"], default="6")
|
|
|
|
if choice == "1":
|
|
list_books()
|
|
elif choice == "2":
|
|
add_book()
|
|
elif choice == "3":
|
|
remove_book()
|
|
elif choice == "4":
|
|
change_book_status()
|
|
elif choice == "5":
|
|
search_books()
|
|
elif choice == "6":
|
|
console.print("Exiting...", style="bold yellow")
|
|
break
|
|
|
|
if __name__ == "__main__":
|
|
main() |