hash_table.py

# code for Hashtable from W-2_ChainingHashTable_zyBooks_Key-Value_CSV_Greedy.py
class ChainingHashTable:
    def __init__(self, initial_capacity=10):
        self.table = []
        for i in range(initial_capacity):
            self.table.append([])

    # insert or update package with package id as key
    def insert(self, key, item):
        bucket = hash(key) % len(self.table)
        bucket_list = self.table[bucket]

        for kv in bucket_list:
            if kv[0] == key:
                kv[1] = item  # Update the key if it exists
                return True

        bucket_list.append([key, item])
        return True

    # search for a package by key
    def search(self, key):
        bucket = hash(key) % len(self.table)
        bucket_list = self.table[bucket]

        for kv in bucket_list:
            if kv[0] == key:
                return kv[1]  # return the item if its found
        return None

    # remove a package by key
    def remove(self, key):
        bucket = hash(key) % len(self.table)
        bucket_list = self.table[bucket]

        for kv in bucket_list:
            if kv[0] == key:
                kv[1] = None
                return True
        return False