017 Stack Implementation in Python.py

# Stack Implementation in Python [version: 0.1] 

print('Stack Operations in Python')
stack = []

def push():
    item = input('\nEnter Item: ')
    stack.append(int(item)) if item.isnumeric() == True else stack.append(item)
    print('Pushed Successfully!')
    more = input('Push more? (y/n): ')
    push() if more in 'yY' else None
def pop():
    print('\nStack is Empty!') if len(stack) == 0 else print(f'\n{stack.pop()} - Popped Successfully!')
    more = input('Pop more? (y/n): ')
    pop() if more in 'yY' else None
def peek():
    print('\nStack is Empty!') if len(stack) == 0 else print(f'Top Item: {stack[-1]}')
def size():
    print(f'\nSize of the Stack: {len(stack)}')
while True:
    user = input('\n____________________\nPress [1]: Push\nPress [2]: Pop\nPress [3]: Peek\nPress [4]: Display\nPress [5]: Size\nPress [6]: Exit\n~> ')
    push() if user == '1' else pop() if user == '2' else peek() if user == '3' else print(f'\nStack: {stack}') if user == '4' else size() if user == '5' else exit() if user == '6' else print('\nERROR 101: Wrong input!')

# online study matrial(theory/concept) of Stack: https://www.geeksforgeeks.org/stack-in-python