Python fuction to plot the y value of option at expiration

option_fx.py

def option_fx(x, longShort: str, callPut: str, strike, cost):
    s = strike
    u = cost
    
    if longShort == 'L':
        if callPut == 'C':
            return (0 if x < s else x - s) - u # Long call
        else:
            return (0 if x > s else s - x) - u # Long Put
    
    else:
        if callPut == 'C':
            return (0 if x < s else s - x) + u # Short call
        else:
            return (0 if x > s else x - s) + u # Short Put