有没有办法重写 if 语句以避免重复这两行代码？

Question

I wrote insertion_sort function and added an if statement to choose if I want to order de array in ascendent or descendent mode. As you can see in the following code inside the if statements there are 2 lines of code repeated. I would like to know if there is another way to write those ifs so there´s no line repeated. I´ve been thinking about it but I can't think of any alternative.

Thanks in advance.

def swap_elements(array, pos1, pos2):
    array[pos1], array[pos2] = array[pos2], array[pos1]
    return array

def insertion_sort(array, method):
    i = 1
    while i < len(array):
        key = array[i]
        j = 0
        while j < i:
            if method:
                if array[j] > key:
                    key = array[j]
                    array = swap_elements(array, i, j)
            else:
                if array[j] < key:
                    key = array[j]
                    array = swap_elements(array, i, j)
            j += 1
        i += 1

    return array

Answer 1

Latest version of the code:

def swap_elements(array, pos1, pos2):
    array[pos1], array[pos2] = array[pos2], array[pos1]

def insertion_sort(array, method):
    i = 1
    while i < len(array):
        key = array[i]
        j = 0
        while j < i:
            if (method and array[j] > key) or (not method and array[j] < key):
                key = array[j]
                swap_elements(array, i, j)
            j += 1
        i += 1