Как передать переменную как инстанс объекта в функции Django

У меня есть функция

def update_field():
    ...
    ...
    book.country = country_id
    ...

У меня есть один и тот же кусок кода в нескольких местах, и мне нужно сделать для него отдельную функцию. Например

def update_field(obj_field):
    ...
    ...
    obj_field = country_id
    ...

но когда я пытаюсь вызвать функцию, она не работает

country = book.country
update_field(country)

def update_field(obj, field_name):
    ...
    ...
    setattr(obj, field_name, country_id)
    ...