1223. validate code with simple regex

Basic regex tasks. Write a function that takes in a numeric code of any length. The function should check if the code begins with 1, 2, or 3 and return true if so. Return false otherwise.

You can assume the input will always be a number.

Solution:

def validate_code(code: int) -> bool:
    return int(str(code)[0]) <= 3

import re

def validate_code(code):
    return bool(re.match("^[1-3]",str(code)))