hivemind-sqlite-database 0.4.0a4__tar.gz → 0.4.0a5__tar.gz

{hivemind_sqlite_database-0.4.0a4 → hivemind_sqlite_database-0.4.0a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hivemind-sqlite-database
-Version: 0.4.0a4
+Version: 0.4.0a5
 Summary: sqlite database plugin for hivemind-core
 Author-email: jarbasAi <jarbasai@mailfence.com>
 License: Apache-2.0

{hivemind_sqlite_database-0.4.0a4 → hivemind_sqlite_database-0.4.0a5}/hivemind_sqlite_database/__init__.py

@@ -215,6 +215,26 @@ class SQLiteDB(AbstractDB):
             LOG.error(f"Failed to add client to SQLite: {e}")
             return False
 
+    def update_last_seen(self, api_key: str, seen_at: float) -> bool:
+        """Atomically advance ``last_seen`` for the matching API key."""
+        try:
+            with self._write_lock, self.conn:
+                cursor = self.conn.execute(
+                    """
+                    UPDATE clients
+                    SET last_seen = CASE
+                        WHEN last_seen IS NULL OR last_seen < ? THEN ?
+                        ELSE last_seen
+                    END
+                    WHERE api_key = ?
+                    """,
+                    (seen_at, seen_at, api_key),
+                )
+            return cursor.rowcount > 0
+        except sqlite3.Error as e:
+            LOG.error(f"Failed to update last_seen in SQLite: {e}")
+            return False
+
     def search_by_value(self, key: str, val: Union[str, bool, int, float]) -> List[Client]:
         """
         Search for clients by a specific key-value pair in the SQLite database.

{hivemind_sqlite_database-0.4.0a4 → hivemind_sqlite_database-0.4.0a5}/hivemind_sqlite_database/version.py

@@ -2,7 +2,7 @@
 VERSION_MAJOR = 0
 VERSION_MINOR = 4
 VERSION_BUILD = 0
-VERSION_ALPHA = 4
+VERSION_ALPHA = 5
 # END_VERSION_BLOCK
 
 __version__ = f"{VERSION_MAJOR}.{VERSION_MINOR}.{VERSION_BUILD}" + (f"a{VERSION_ALPHA}" if VERSION_ALPHA else "")

{hivemind_sqlite_database-0.4.0a4 → hivemind_sqlite_database-0.4.0a5}/hivemind_sqlite_database.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hivemind-sqlite-database
-Version: 0.4.0a4
+Version: 0.4.0a5
 Summary: sqlite database plugin for hivemind-core
 Author-email: jarbasAi <jarbasai@mailfence.com>
 License: Apache-2.0

{hivemind_sqlite_database-0.4.0a4 → hivemind_sqlite_database-0.4.0a5}/tests/test_sqlitedb.py

@@ -189,6 +189,22 @@ class TestSQLiteDBSearchByValue(unittest.TestCase):
         self.assertEqual(results, [])
 
 
+class TestSQLiteDBLastSeen(unittest.TestCase):
+    def test_update_last_seen_never_moves_timestamp_backward(self):
+        db = make_db()
+        db.add_item(make_client(1, "key", last_seen=200.0))
+
+        self.assertTrue(db.update_last_seen("key", 100.0))
+        self.assertEqual(db.search_by_value("api_key", "key")[0].last_seen, 200.0)
+
+        self.assertTrue(db.update_last_seen("key", 300.0))
+        self.assertEqual(db.search_by_value("api_key", "key")[0].last_seen, 300.0)
+
+    def test_update_last_seen_returns_false_for_missing_key(self):
+        db = make_db()
+        self.assertFalse(db.update_last_seen("missing", 100.0))
+
+
 class TestSQLiteDBIter(unittest.TestCase):
     def test_iter_yields_all_rows(self):
         db = make_db()