admin_interface.py

import sys
from models.doctors import add_doctor, get_all_doctors
from models.patient import get_all_patients
from models.appointment import get_all_appointments_admin
import time

ADMIN_USERNAME = "admin"
ADMIN_PASSWORD = "password"

def admin_login():
    print("---- Admin Login ----")
    username = input("Username: ").strip()
    password = input("Password: ").strip()
    if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
        print("Login successful! Welcome, Admin.")
        return True
    else:
        print("Invalid credentials. Access denied.")
        return False
def admin_menu():
    print("\n--- Admin Menu (GOD User) ---")
    print("1. Add New Doctor")
    print("2. View All Doctors")
    print("3. View All Patients")
    print("4. View All Appointments")
    print("5. Generate Summary Report")
    print("6. Logout")
    print("--------------------")
    return input("Select an option (1-6): ").strip()
def handle_admin_choice(choice):
    match(choice):
        case '1':
            handle_add_doctor()
        case '2':
            handle_view_doctors()
        case '3':
            handle_view_patients()
        case '4':
            handle_view_appointments()
        case '5':
            handle_generate_report()
        case '6':
            print("👋 Admin logged out.")
            return False 
        case _:
            print("❗ Invalid choice. Please select a number from 1 to 6.")
    return True 
def handle_add_doctor():
    print("\n--- Add New Doctor ---")
    doctor_id = input("Doctor ID: ").strip()
    first_name = input("First Name: ").strip()
    last_name = input("Last Name: ").strip()
    specialization = input("Specialization: ").strip()
    phone_number = input("Phone Number: ").strip()
    success = add_doctor(doctor_id, first_name, last_name, specialization, phone_number)
    if success:
        print(f"Dr. {first_name} {last_name} added successfully.")
    else:
        print("Failed to add doctor. Please try again.")
def handle_view_doctors():
    print("\n--- All Doctors ---")
    doctors = get_all_doctors()
    if not doctors:
        print("No doctors found.")
        return
    for doc in doctors:
        print(f"ID: {doc['doctor_id']} | Dr. {doc['first_name']} {doc['last_name']} | Specialty: {doc['specialization']}")
def handle_view_patients():
    print("\n--- All Patients ---")
    patients = get_all_patients()
    if not patients:
        print("No patients found.")
        return
    for i, p in enumerate(patients, start=1):
        print(f"{i}. ID: {p['patient_id']} | Name: {p['last_name']}, {p['first_name']} | Phone: {p['phone_number']} | Email: {p['email']}")
def handle_view_appointments():
    print("\n--- All Appointments ---")
    appointments = get_all_appointments_admin()
    if not appointments:
        print("No appointments found.")
        return
    for appt in appointments:
        print(f"Appointment ID: {appt['appointment_id']} | Patient: {appt['patient_name']} | Doctor: Dr. {appt['doctor_name']} | Date: {appt['appointment_date']} | Time: {appt['appointment_time']} | Status: {appt['status']}")
def handle_generate_report():
    print("\n--- Generating Summary Report ---")
    
    doctors=get_all_doctors()
    patients=get_all_patients()
    appointments=get_all_appointments_admin()

    if not appointments:
        print("No appointments found to generate report.")
        return
    print("\n--- 1. OVERALL TOTALS ---")
    print(f"Total Doctors Registered: {len(doctors)}")
    print(f"Total Patients Registered: {len(patients)}")
    print(f"Total Appointments Booked: {len(appointments)}")

    print("\n--- 2. APPOINTMENT STATUS HISTORY ---")
    status_counts = {}
    for appt in appointments:
        status = appt['status']
        status_counts[status] = status_counts.get(status, 0) + 1
    for status, count in status_counts.items():
        print(f"Total '{status}' Appointments: {count}")
    
   
    print("--- 3. DOCTOR ACTIVITY SUMMARY ---:")
    doctor_activity = {}
    for doc in doctors:
        doctor_activity[doc['doctor_id']] = { 
            'name': f"Dr. {doc['last_name']}",
            'count': 0
        }

    for appt in appointments:
        doc_id = appt['doctor_id']
        if doc_id in doctor_activity:
            doctor_activity[doc_id]['count'] += 1
    sorted_activity = sorted(doctor_activity.values(), key=lambda x: x['count'], reverse=True)
    for activity in sorted_activity:
        print(f"{activity['name']}: {activity['count']} appointments")

def run_admin_interface():
    print("Welcome to the Hospital Appointment Management System (HAMS) Admin Interface")
    if admin_login():
        running = True
        while running:
            running = handle_admin_choice(admin_menu())