Bug 17298: AttributeError during import of SMESH_BelongToGeom.py.

diff --git a/src/SMESH_SWIG/smeshDC.py b/src/SMESH_SWIG/smeshDC.py
index 69000315970841ce0998640b42b9f0aa0032d532..f980b9e93e9b80325b44bf516efef6b5cb0bb1a1 100644 (file)
--- a/src/SMESH_SWIG/smeshDC.py
+++ b/src/SMESH_SWIG/smeshDC.py
@@ -306,7 +306,7 @@ class smeshDC(SMESH._objref_SMESH_Gen):
         if CritType in [FT_BelongToGeom,     FT_BelongToPlane, FT_BelongToGenSurface, 
                         FT_BelongToCylinder, FT_LyingOnGeom]:
             # Check treshold
-            if isinstance(aTreshold, self.geompyD.GEOM._objref_GEOM_Object):
+            if isinstance(aTreshold, geompyDC.GEOM._objref_GEOM_Object):
                 aCriterion.ThresholdStr = GetName(aTreshold)
                 aCriterion.ThresholdID = salome.ObjectToID(aTreshold)
             else:
@@ -611,7 +611,7 @@ class Mesh_Segment(Mesh_Algorithm):
         store_geom = self.geom
         if vertex:
             if type(vertex) is types.IntType:
-                vertex = self.mesh.geompyD.SubShapeAllSorted(self.geom,self.mesh.geompyD.ShapeType["VERTEX"])[vertex]
+                vertex = self.mesh.geompyD.SubShapeAllSorted(self.geom, geompyDC.ShapeType["VERTEX"])[vertex]
                 pass
             self.geom = vertex
             pass