파이썬 - class의 정적 함수를 동적으로 교체
지난 글에서,
파이썬 3.x에서의 동적 함수 추가
; https://www.sysnet.pe.kr/2/0/13380
동적으로 외부의 함수를 내부에 추가하는 방법을 알아봤는데요, 이번에는 이미 정의된 함수를 대체하는 방법을 알아보겠습니다. 우선, 예제를 볼까요? ^^
class MyObject:
def __init__(self):
pass
@staticmethod
def set_name(text):
MyObject.name = text
current_module = __import__(__name__)
class_type = current_module.__dict__.get('MyObject')
print(class_type, type(class_type))
""" 출력 결과
<class '__main__.MyObject'> <class 'type'>
"""
이렇게 구한 class_type은 "MyObject" 이름으로 접근한 것과 완전히 동일한 타입 인스턴스입니다. 그렇기 때문에 다음과 같이 멤버를 접근하는 것도 가능합니다.
class_type.set_name('hello world') # MyObject.set_name 호출과 동일
print(MyObject.name) # 출력 결과: hello world
그런데, 저 set_name 정적 메서드를 동적으로 접근하려면 어떻게 해야 할까요?
set_name_method = class_type.__dict__.get('set_name')
print(set_name_method, type(set_name_method))
""" 출력 결과
<staticmethod object at 0x000001B55CE77DC0> <class 'staticmethod'>
"""
__dict__ 통해 얻은 경우 예상했던 대로 (decorator로 지정한) staticmethod 타입의 인스턴스가 나오는데요, 따라서 이렇게 얻은 set_name_method로는 호출이 안 됩니다.
set_name_method('hello world') # 오류 발생 TypeError: 'staticmethod' object is not callable
지난 글에서 다룬 staticmethod의 사용법과 비교해 보면 재미있는 결과를 얻게 되는데요,
def f_static(arg):
print('f_static', arg)
MyObject.fs = staticmethod(f_static)
MyObject.fs('hello world') # 정상적으로 호출
print(MyObject.fs, type(MyObject.fs)) # <function f_static at 0x000001DA86207280> <class 'function'>
위의 코드에서도 f_static을 staticmethod 타입으로 감쌌지만 MyObject.fs의 타입이 "staticmethod"가 아닌 function으로 나온다는 차이점이 있습니다.
어쨌든, (__dict__를 통해) 동적으로 구한 staticmethod 인스턴스는 decorator가 지정된 함수를 반환하는 __func__을 통해야만 합니다.
set_name_method.__func__('hello world')
print(MyObject.name) # 출력 결과: hello world
그렇다면, 위와 같이 정의된 정적 함수를 아래의 (클래스가 아닌 모듈에 속한) 함수로,
name = None
def global_set_name(text):
global name
name = text
교체하려면 어떻게 해야 할까요? 간단하게는 class 이름 또는 그것의 type 인스턴스를 접근해 바꿀 수 있습니다.
# 클래스 이름을 사용해도 되고,
MyObject.set_name = global_set_name
MyObject.set_name('hello world-1')
print(name) # 출력 결과: hello world-1
# 동적으로 구한 class 'type'의 인스턴스를 이용해도 됨
class_type.set_name = global_set_name
class_type.set_name('hello world-2')
print(name) # 출력 결과: hello world-2
그런데, 'set_name' 함수를 (__dict__를 통해) 동적으로 접근하고 싶은 경우라면 사정이 달라집니다. 해당 함수를 구하는 경우에는 __dict__를 이용할 수 있었지만, 이걸로는 교체까진 되지 않습니다.
class_type.__dict__['set_name'] = global_set_name
# 오류 발생: TypeError: 'mappingproxy' object does not support item assignment
class_type.__dict__는 mappingproxy 타입으로 read-only라서 저런 오류가 발생합니다. 이에 대해 검색해 보면,
How to modify class __dict__ (a mappingproxy)? [duplicate]
; https://stackoverflow.com/questions/20019333/how-to-modify-class-dict-a-mappingproxy
파이썬의 setattr 내장 함수를 이용하라고 하는군요. ^^
setattr(class_type, 'set_name', global_set_name)
class_type.set_name('hello world')
print(name) # global_set_name 함수로 교체됐으므로 전역 name 변수의 값이 바뀜
# 출력 결과: hello world
그 외 class의 instance 함수는 decorator가 지정되지 않은 유형이므로 __func__ 등을 고려하지 않아도 된다는 점만 빼고는 위의 원칙을 그대로 적용할 수 있습니다.
또한, setattr의 두 번째 인자가 멤버를 구분하는 데 있어 키 타입으로 str 문자열을 사용한다는 점에서
파이썬에서는 같은 이름의 다른 인자를 갖는 메서드 오버로딩이 지원되지 않음을 짐작할 수 있을 것입니다.
[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]