Linked List

linkedList.py

# Simple Node definition with value and next pointer.
class Node:
    def __init__(self, value = 0):
        self.value = value
        self.next  = None
        
        for v in values:
            self.add(v)
            
    def __iter__(self):
        """LinkedList iterator operator
        
        Returns:
            Iterable
        """
        self.node = self
        return self

    def __next__(self):
        """Returns next Node in iteration.
        
        Returns:
            Node
        """
        if self.node == None:
            raise StopIteration

        node      = self.node
        self.node = self.node.next

        return node
    
    def __str__(self):
        """String representation of Node.
        
        Returns:
            String
        """
        return self.value

# Simple LinkedList definition.
class LinkedList:
    def __init__(self, values = []):
        self.head   = None
        self.length = 0
        
    def add(self, value):
        """Add Node to end of LinkedList.
        
        Args:
            value(int): new node value
        Returns:
            Bool
        """
        if self.isEmpty():
            self.head    = ListNode(value)
            self.length += 1
        else:
            curr = self.head

            while curr.next != None:
                curr = curr.next

            curr.next    = ListNode(value)
            self.length += 1
        
    def length(self):
        """Returns LinkedList length.
        
        Returns:
            Bool
        """
        return self.length
    
    def isEmpty(self):
        """Is LinkedList empty?.
        
        Returns:
            Bool
        """
        return self.head == None
    
    def toList(self):
        """Returns List of all Node values in list.
        
        Returns:
            List
        """
        return [n.val for v in iter(self)]
    
    def __iter__(self):
        """LinkedList iterator operator
        
        Returns:
            Iterable
        """
        return self.head.__iter__()

    def __str__(self):
        """String representation of LinkedList.
        
        Returns:
            String
        """
        return "->".join([str(n.val) for n in iter(self)])