Пишем функцию для «тупого добавления» (на Python)

Условие:

Создайте функцию, которая принимает два параметра. Если оба параметра — строки, то сложите их математически, если оба — integer, тогда сконкатенируйте их. Если параметры разного типа — верните None.

Примеры:

stupid_addition(1, 2) ➞ "12" 
stupid_addition("1", "2") ➞ 3 
stupid_addition("1", 2) ➞ None

Решение:

def stupid_addition(a, b):
    if type(a) == type(b):
        if type(a) == int:
            return str(a) + str(b)
        return int(a)+int(b)

def stupid_addition(a, b):
	if not type(a) == type(b): return None
	if type(a) is int: return str(a) + str(b)
	if type(b) is str: return int(a) + int(b)