diff --git a/direct/src/distributed/ClientDistClass.py b/direct/src/distributed/ClientDistClass.py index 7f00a58ad7..d23cd54382 100644 --- a/direct/src/distributed/ClientDistClass.py +++ b/direct/src/distributed/ClientDistClass.py @@ -5,6 +5,7 @@ import DirectNotifyGlobal import ClientDistUpdate class ClientDistClass: + notify = DirectNotifyGlobal.directNotify.newCategory("ClientDistClass") def __init__(self, dcClass): self.number = dcClass.getNumber() @@ -16,8 +17,12 @@ class ClientDistClass: self.broadcastRequiredCDU = self.listBroadcastRequiredCDU(self.allCDU) self.allRequiredCDU = self.listRequiredCDU(self.allCDU) # Import the class, and store the constructor - exec("import " + self.name) - self.constructor = eval(self.name + "." + self.name) + try: + exec("import " + self.name) + self.constructor = eval(self.name + "." + self.name) + except ImportError, e: + self.notify.warning("%s.py does not exist." % (self.name)) + self.constructor = None return None def parseFields(self, dcClass): diff --git a/direct/src/distributed/ClientDistUpdate.py b/direct/src/distributed/ClientDistUpdate.py index 9a82ffd4d3..a254ff4e96 100644 --- a/direct/src/distributed/ClientDistUpdate.py +++ b/direct/src/distributed/ClientDistUpdate.py @@ -22,13 +22,16 @@ class ClientDistUpdate: self.divisors = [] self.deriveTypesFromParticle(dcField) # Figure out our function pointer - exec("import " + cdc.name, moduleGlobals, moduleLocals) try: + exec("import " + cdc.name, moduleGlobals, moduleLocals) self.func = eval(cdc.name + "." + cdc.name + "." + self.name) # Only catch name and attribute errors # as all other errors are legit errors + except ImportError, e: + self.notify.warning("%s.py does not exist." % (cdc.name)) + self.func = None except (NameError, AttributeError), e: - #ClientDistUpdate.notify.warning(cdc.name + "." + self.name + + #self.notify.warning(cdc.name + "." + self.name + # " does not exist") self.func = None return None @@ -47,7 +50,7 @@ class ClientDistUpdate: self.types.append(componentField.getElementType(j)) self.divisors.append(componentField.getElementDivisor(j)) else: - ClientDistUpdate.notify.error("field is neither atom nor molecule") + self.notify.error("field is neither atom nor molecule") return None def updateField(self, cdc, do, di):