Laurie/runtime import (#27)

* Auto importing modules (#24)

Co-authored-by: Devin Miller <devin.miller@unity3d.com>
Co-authored-by: flonou <florian.nouviale@inria.fr>

Author: 3 people

src/ros_tcp_endpoint/default_server_endpoint.py

@@ -1,15 +1,6 @@
 #!/usr/bin/env python
 
 import rospy
-import actionlib_msgs.msg
-import diagnostic_msgs.msg
-import geometry_msgs.msg
-import nav_msgs.msg
-import sensor_msgs.msg
-import shape_msgs.msg
-import stereo_msgs.msg
-import trajectory_msgs.msg
-import visualization_msgs.msg
 
 from ros_tcp_endpoint import TcpServer
 

src/ros_tcp_endpoint/server.py

@@ -17,6 +17,7 @@
 import json
 import sys
 import threading
+import importlib
 
 from .tcp_sender import UnityTcpSender
 from .client import ClientThread
@@ -139,6 +140,7 @@ def resolve_message_name(name):
         names = name.split('/')
         module_name = names[0]
         class_name = names[1]
+        importlib.import_module(module_name+ ".msg")
         module = sys.modules[module_name]
         if module is None:
             rospy.loginfo("Failed to resolve module {}".format(module_name))
@@ -150,4 +152,5 @@ def resolve_message_name(name):
             rospy.loginfo("Failed to resolve module {}.msg.{}".format(module_name, class_name))
         return module
     except (IndexError, KeyError, AttributeError) as e:
+        rospy.loginfo("Exception Raised: {}".format(e))
         return None