def _set_target(self, obj): for _, fname in model_map.values(): setattr(self, fname, None) if obj is None: return _, fname = model_map[obj.__class__] setattr(self, fname, obj)
def _get_target(self): for _, fname in model_map.values(): if hasattr(self, fname) and getattr(self, fname): return getattr(self, fname)