[Código] Calcular área de un prísma

##############################
# Calcular area de un Prisma #
# Autor: Sthefano02          #
# http://comunidadraw.com/   #
##############################

def Area_Lateral(Pbase,altura):
    return Pbase * altura

def Area_Total(Alateral,base):
    return (Alateral + 2) * base


print '''
    = Area de un Prisma y Piramide =
    1 - Area lateral del prisma
    2 - Area total del prisma
    ================================
    '''
opcion = input('Elija una opcion > ')
if opcion == 1:
    Pbase = input('Ingrese el perimetro de la base > ')
    altura = input('Ingrese la altura > ')
    print 'El area lateral del prisma es %s' % Area_Lateral(Pbase,altura)
if opcion == 2:
    Alateral = input('Ingrese el area lateral del prisma > ')
    base = input('Ingrese el area de la base > ')
    print 'El area total del prisma es %s' % Area_Total(Alateral,base)