Using ulid for _id

README.md

@@ -126,4 +126,3 @@ PantherDB is a <b>Simple</b>, <b>FileBase</b> and <b>Document Oriented</b> datab
 - [x] Add encryption
 - [ ] Complete tests TODO
 - [ ] Add B+ tree
-- [ ] Find better encryption solution

pantherdb/__init__.py

@@ -1,6 +1,6 @@
 from pantherdb.pantherdb import *  # noqa: F403
 
-__version__ = '1.3.6'
+__version__ = '1.4.0'
 
 
 __all__ = ('__version__', 'PantherDB', 'PantherCollection', 'PantherDocument', 'PantherDBException')

pantherdb/pantherdb.py

@@ -6,6 +6,8 @@
 
 import orjson as json
 
+from pantherdb.ulid import ULID
+
 
 class PantherDBException(Exception):
     pass
@@ -18,6 +20,7 @@ class PantherDB:
     __fernet: typing.Any  # type[cryptography.fernet.Fernet | None]
     __return_dict: bool
     __content: dict
+    __ulid: ULID
 
     def __new__(cls, *args, **kwargs):
         if cls.__name__ != 'PantherDB':
@@ -47,6 +50,7 @@ def __init__(
     ):
         self.__return_dict = return_dict
         self.__secret_key = secret_key
+        self.__ulid = ULID()
         self.__content = {}
         if self.__secret_key:
             from cryptography.fernet import Fernet
@@ -76,6 +80,10 @@ def content(self) -> dict:
     def return_dict(self) -> bool:
         return self.__return_dict
 
+    @property
+    def ulid(self) -> ULID:
+        return self.__ulid
+
     @property
     def secret_key(self) -> bytes | None:
         return self.__secret_key
@@ -243,7 +251,7 @@ def all(self) -> list[PantherDocument | dict]:
 
     def insert_one(self, **kwargs) -> PantherDocument | dict:
         documents = self._get_collection()
-        kwargs['_id'] = len(documents) + 1
+        kwargs['_id'] = self.ulid.new()
         documents.append(kwargs)
         self._write_collection(documents)
         return self.__create_result(kwargs)
@@ -376,7 +384,7 @@ def __init__(
         )
 
     def __str__(self) -> str:
-        items = ', '.join(f'id={v}' if k == '_id' else f'{k}={v}' for k, v in self.data.items())
+        items = ', '.join(f'{k}={v}' for k, v in self.data.items())
         return f'{self.collection_name}({items})'
 
     __repr__ = __str__
@@ -394,6 +402,7 @@ def __setattr__(self, key, value):
             '_PantherDB__secret_key',
             '_PantherDB__content',
             '_PantherDB__fernet',
+            '_PantherDB__ulid',
             '_PantherCollection__collection_name',
             '_PantherDocument__data',
         ]:

pantherdb/ulid.py

@@ -0,0 +1,22 @@
+import secrets
+from datetime import datetime, timezone
+
+
+class ULID:
+    """https://github.com/ulid/spec"""
+
+    def __init__(self):
+        self.crockford_base32_characters = '0123456789ABCDEFGHJKMNPQRSTVWXYZ'
+
+    def new(self):
+        current_timestamp = int(datetime.now(timezone.utc).timestamp() * 1000)
+        epoch_bits = '{0:050b}'.format(current_timestamp)
+        random_bits = '{0:080b}'.format(secrets.randbits(80))
+        bits = epoch_bits + random_bits
+        return self._generate(bits)
+
+    def _generate(self, bits: str) -> str:
+        return ''.join(
+            self.crockford_base32_characters[int(bits[i: i + 5], base=2)]
+            for i in range(0, 130, 5)
+        )