X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Fbindings%2Fpython%2Fbt2%2Fbt2%2Fclock_snapshot.py;h=1ec41299e524eabf365637ec2539b639dee465b5;hb=57081273d1191fc79edc101af619fab96b72460d;hp=4bf37d622b232917fedd69420c7d65ccfffdfa1f;hpb=578e048b5debf169e286e5b5cc747b5d6c16886d;p=babeltrace.git

diff --git a/src/bindings/python/bt2/bt2/clock_snapshot.py b/src/bindings/python/bt2/bt2/clock_snapshot.py
index 4bf37d62..1ec41299 100644
--- a/src/bindings/python/bt2/bt2/clock_snapshot.py
+++ b/src/bindings/python/bt2/bt2/clock_snapshot.py
@@ -22,31 +22,28 @@
 
 from bt2 import native_bt, object, utils
 import numbers
-import bt2
+from bt2 import clock_class as bt2_clock_class
 import functools
 
 
-class _BaseClockSnapshot(object._UniqueObject):
+@functools.total_ordering
+class _ClockSnapshot(object._UniqueObject):
     @property
     def clock_class(self):
         cc_ptr = native_bt.clock_snapshot_borrow_clock_class_const(self._ptr)
         assert cc_ptr is not None
-        return bt2.clock_class._ClockClass._create_from_ptr_and_get_ref(cc_ptr)
-
+        return bt2_clock_class._ClockClass._create_from_ptr_and_get_ref(cc_ptr)
 
-@functools.total_ordering
-class _ClockSnapshot(_BaseClockSnapshot):
     @property
     def value(self):
         return native_bt.clock_snapshot_get_value(self._ptr)
 
     @property
     def ns_from_origin(self):
-        ret, ns = native_bt.clock_snapshot_get_ns_from_origin(self._ptr)
-
-        if ret == native_bt.CLOCK_SNAPSHOT_STATUS_OVERFLOW:
-            raise OverflowError("cannot get clock snapshot's nanoseconds from origin")
-
+        status, ns = native_bt.clock_snapshot_get_ns_from_origin(self._ptr)
+        utils._handle_func_status(
+            status, "cannot get clock snapshot's nanoseconds from origin"
+        )
         return ns
 
     def __eq__(self, other):
@@ -62,9 +59,5 @@ class _ClockSnapshot(_BaseClockSnapshot):
         return self.value < int(other)
 
 
-class _UnknownClockSnapshot(_BaseClockSnapshot):
-    pass
-
-
-class _InfiniteClockSnapshot(_BaseClockSnapshot):
+class _UnknownClockSnapshot:
     pass