added editor_on_asset_post_import, 20tab#71

Author: Roberto De Ioris

Source/UnrealEnginePython/Private/UEPyEditor.cpp

@@ -141,7 +141,7 @@ PyObject *py_unreal_engine_import_asset(PyObject * self, PyObject * args) {
 		}
 		else {
 			return PyErr_Format(PyExc_Exception, "uobject is not a Class");
-		}	
+		}
 	}
 	else if (PyUnicodeOrString_Check(obj)) {
 		char *class_name = PyUnicode_AsUTF8(obj);
@@ -156,8 +156,8 @@ PyObject *py_unreal_engine_import_asset(PyObject * self, PyObject * args) {
 	else {
 		return PyErr_Format(PyExc_Exception, "invalid uobject");
 	}
-		
-	
+
+
 	if (factory_class) {
 		factory = NewObject<UFactory>(GetTransientPackage(), factory_class);
 		if (!factory) {
@@ -179,7 +179,7 @@ PyObject *py_unreal_engine_import_asset(PyObject * self, PyObject * args) {
 		Py_INCREF(ret);
 		return (PyObject *)ret;
 	}
-	 
+
 
 	Py_INCREF(Py_None);
 	return Py_None;
@@ -593,7 +593,7 @@ PyObject *py_unreal_engine_add_component_to_blueprint(PyObject * self, PyObject
 		return PyErr_Format(PyExc_Exception, "uobject is in invalid state");
 	Py_INCREF(ret);
 	return (PyObject *)ret;
-	
+
 }
 
 PyObject *py_unreal_engine_blueprint_add_member_variable(PyObject * self, PyObject * args) {
@@ -659,5 +659,22 @@ PyObject *py_unreal_engine_blueprint_add_new_timeline(PyObject * self, PyObject
 	return (PyObject *)ret;
 }
 
+PyObject *py_unreal_engine_editor_on_asset_post_import(PyObject * self, PyObject * args) {
+	PyObject *py_callable;
+	if (!PyArg_ParseTuple(args, "O:editor_on_asset_post_import", &py_callable)) {
+		return NULL;
+	}
+
+	if (!PyCallable_Check(py_callable))
+		return PyErr_Format(PyExc_Exception, "object is not a callable");
+
+	UPythonDelegate *py_delegate = NewObject<UPythonDelegate>();
+	py_delegate->SetPyCallable(py_callable);
+	py_delegate->AddToRoot();
+	FEditorDelegates::OnAssetPostImport.AddUObject(py_delegate, &UPythonDelegate::PyFOnAssetPostImport);
+	Py_INCREF(Py_None);
+	return Py_None;
+}
+
 #endif
 

Source/UnrealEnginePython/Private/UEPyEditor.h

@@ -30,5 +30,6 @@ PyObject *py_unreal_engine_blueprint_add_member_variable(PyObject *, PyObject *)
 PyObject *py_unreal_engine_blueprint_add_new_timeline(PyObject *, PyObject *);
 
 PyObject *py_unreal_engine_editor_play(PyObject *, PyObject *);
+PyObject *py_unreal_engine_editor_on_asset_post_import(PyObject *, PyObject *);
 
 #endif

Source/UnrealEnginePython/Private/UEPyModule.cpp

@@ -158,6 +158,8 @@ static PyMethodDef unreal_engine_methods[] = {
 	{ "slate_window", py_unreal_engine_slate_window, METH_VARARGS, "" },
 	{ "slate_button", py_unreal_engine_slate_button, METH_VARARGS, "" },
 	{ "get_editor_window", py_unreal_engine_get_editor_window, METH_VARARGS, "" },
+
+	{ "editor_on_asset_post_import", py_unreal_engine_editor_on_asset_post_import, METH_VARARGS, "" },
 #endif
 
 	{ "new_object", py_unreal_engine_new_object, METH_VARARGS, "" },

Source/UnrealEnginePython/Public/PythonDelegate.cpp

@@ -109,6 +109,20 @@ bool UPythonDelegate::Tick(float DeltaTime)
 	return false;
 }
 
+#if WITH_EDITOR
+void UPythonDelegate::PyFOnAssetPostImport(UFactory *factory, UObject *u_object)
+{
+	FScopePythonGIL gil;
+	PyObject *ret = PyObject_CallFunction(py_callable, (char *)"OO", ue_get_python_wrapper((UObject *)factory), ue_get_python_wrapper(u_object));
+	if (!ret) {
+		unreal_engine_py_log_error();
+		return;
+	}
+	Py_DECREF(ret);
+}
+#endif
+
+
 UPythonDelegate::~UPythonDelegate()
 {
 	FScopePythonGIL gil;

Source/UnrealEnginePython/Public/PythonDelegate.h

@@ -20,6 +20,10 @@ class UPythonDelegate : public UObject
 
 	bool Tick(float DeltaTime);
 
+#if WITH_EDITOR
+	void PyFOnAssetPostImport(UFactory *factory, UObject *u_object);
+#endif
+
 private:
 	UFunction *signature;
 	bool signature_set;