Reminds me of the joke: A guy goes to his doctor and says, "Doc, it hurts when I do this." Doctor replies, "Don't do that."
Seriously, don't even think about gluing the driveshaft in. There's no way that could do any good. As long as the locomotive operates without the driveshaft falling out, I would just continue to be careful handling it.
DFF