Add time property (#6)

Author: oittaa

src/uuid6/__init__.py

@@ -10,7 +10,7 @@
 class DraftUUID(UUID):
     r"""UUID draft version objects"""
 
-    def __init__(self, int: int, version: int = None):
+    def __init__(self, int: int, version: int = None) -> None:
         r"""Create a UUID from a single 128-bit integer as the 'int' argument."""
 
         if not 0 <= int < 1 << 128:
@@ -26,6 +26,30 @@ def __init__(self, int: int, version: int = None):
             int |= version << 76
         super().__init__(int=int)
 
+    @property
+    def subsec(self) -> int:
+        return (
+            (self.time_mid & 0x0FFF) << 18
+            | (self.time_hi_version & 0x0FFF) << 6
+            | self.clock_seq_hi_variant & 0x3F
+        )
+
+    @property
+    def time(self) -> int:
+        if self.version == 6:
+            return (
+                (self.time_low << 28)
+                | (self.time_mid << 12)
+                | (self.time_hi_version & 0x0FFF)
+            )
+        if self.version == 7:
+            return self.unixts * 10 ** 9 + self.subsec
+        return super().time
+
+    @property
+    def unixts(self) -> int:
+        return self.time_low << 4 | self.time_mid >> 12
+
 
 def _getrandbits(k: int) -> int:
     import random

test/test_uuid6.py

@@ -1,9 +1,10 @@
 import unittest
+from time import time_ns
 from unittest.mock import patch
+from uuid import uuid1
 
 from uuid6 import DraftUUID, uuid6, uuid7
 
-
 YEAR_IN_NS = 3600 * 24 * 36525 * 10 ** 7
 
 
@@ -33,6 +34,7 @@ def test_invalid_int(self):
     def test_valid_int(self):
         test_uuid = DraftUUID(int=0)
         self.assertEqual(test_uuid.version, None)
+        self.assertEqual(test_uuid.time, 0)
         test_uuid = DraftUUID(int=(1 << 128) - 1)
         self.assertEqual(test_uuid.version, None)
 
@@ -96,6 +98,14 @@ def test_uuid7_far_in_future(self):
             uuid_2178_from_epoch = uuid7()
         self.assertLess(uuid_2178_from_epoch, uuid_prev)
 
+    def test_time(self):
+        uuid_1 = uuid1()
+        uuid_6 = uuid6()
+        self.assertAlmostEqual(uuid_6.time / 10 ** 7, uuid_1.time / 10 ** 7, 3)
+        cur_time = time_ns()
+        uuid_7 = uuid7()
+        self.assertAlmostEqual(uuid_7.time / 10 ** 9, cur_time / 10 ** 9, 3)
+
 
 if __name__ == "__main__":
     unittest.main()