Binary File Search & Display Records

import pickle

def display_all_records():
    try:
        with open('employee.dat', 'rb') as file:
            print("All Employee Records:")
            print("=" * 50)
            record_count = 0
            
            try:
                while True:
                    record = pickle.load(file)
                    record_count += 1
                    print(f"Record {record_count}:")
                    print(f"ID: {record['id']}")
                    print(f"Name: {record['name']}")
                    print(f"Department: {record['dept']}")
                    print(f"Salary: {record['salary']}")
                    print("-" * 30)
            except EOFError:
                print(f"Total records displayed: {record_count}")
                
    except FileNotFoundError:
        print("File not found!")
    except Exception as e:
        print(f"Error: {e}")

def search_by_department():
    try:
        with open('employee.dat', 'rb') as file:
            dept_name = input("Enter department to search: ")
            print(f"Employees in {dept_name} department:")
            print("=" * 40)
            found = False
            
            try:
                while True:
                    record = pickle.load(file)
                    if record['dept'].lower() == dept_name.lower():
                        print(f"Name: {record['name']}")
                        print(f"ID: {record['id']}")
                        print(f"Salary: {record['salary']}")
                        print("-" * 25)
                        found = True
            except EOFError:
                if not found:
                    print("No records found for this department!")
                    
    except FileNotFoundError:
        print("File not found!")
    except Exception as e:
        print(f"Error: {e}")

# Main program
if __name__ == "__main__":
    choice = input("1. Display all records\n2. Search by department\nEnter choice: ")
    if choice == '1':
        display_all_records()
    elif choice == '2':
        search_by_department()
    else:
        print("Invalid choice!")