In Python, is it possible to assign an if-then statement to a variable?

Question

I've defined a few variables below and I want to use them to define a variable.

today = datetime.today()
datem = str(datetime(today.year, today.month, 1))
curr_month = datem[5:7]
curr_year = datem[2:4]
list_early_fy = ['04', '05', '06', '07', '08', '09', '10', '11', '12']

I then want to use these to define the fiscal year I'm using. I tried both methods below (the first one was what I was really looking for), but both just said "invalid syntax".

test_year = if curr_month in list_early_fy:
            print(int(curr_year)+1, curr_year)
    
def test_year:
        if curr_month in list_early_fy:
            print(int(curr_year)+1, curr_year)

Finally, I want to use that "test_year" variable in other places in my code. Anyway, any suggestions on how to make this work?

question from:https://stackoverflow.com/questions/65850929/in-python-is-it-possible-to-assign-an-if-then-statement-to-a-variable

Answer 1

A function needs to at least have parentheses, even if it takes no parameters, so def test_year: won't work. An if statement can't be assigned to a variable. Instead, you could define a function that accepts curr_month and list_early_fy as parameters,

def test_year(curr_month, curr_year, list_early_fy):
        if curr_month in list_early_fy:
            print(int(curr_year)+1, curr_year)

And you'd call it via test_year(curr_month, curr_year, list_early_fy).

If you had a class that stored curr_month, curr_year, and list_early_fy, you could reference those class variables:

class MyClass(object):
    def __init__(self, curr_month, curr_year, list_early_fy):
        self.curr_month = curr_month
        self.curr_year = curr_year
        self.list_early_fy = list_early_fy
    
    def test_year():
        if self.curr_month in self.list_early_fy:
            print(int(self.curr_year)+1, self.curr_year)

c = MyClass(curr_month, curr_year, list_early_fy)
c.test_year()