collections-cache 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

collections_cache/collections_cache.py

@@ -3,6 +3,7 @@ from multiprocessing import Pool
 from os import cpu_count, path, makedirs, scandir
 from itertools import chain
 from random import choice
+#from threading import Thread as task
 import pickle
 
 class Collection_Cache:
@@ -27,7 +28,8 @@ class Collection_Cache:
 
     def initialize_databases(self, db_path):
         conn = sqlite3.connect(db_path)
-        conn.execute("PRAGMA journal_mode=WAL;")
+        self.configure_connection(conn)
+        #conn.execute("PRAGMA journal_mode=WAL;")
         conn.execute("""
             CREATE TABLE IF NOT EXISTS data(
                 key     TEXT,
@@ -47,8 +49,9 @@ class Collection_Cache:
 
     def get_all_keys(self, database):
         conn    = sqlite3.connect(database)
+        self.configure_connection(conn)
         cursor  = conn.cursor()
-        cursor.execute("PRAGMA journal_mode=WAL;")
+        #cursor.execute("PRAGMA journal_mode=WAL;")
         cursor.execute("SELECT key FROM data;")
         result  = cursor.fetchall()
         keys    = [(line[0], database) for line in result]
@@ -56,12 +59,14 @@ class Collection_Cache:
         return keys
 
     def set_key(self, key, value):
+        """Used to store values and associate a value with a key."""
         if key not in self.keys_databases:
             database_to_insert = choice(self.databases_list)
             #print(f"Inserting in {database_to_insert}")
             conn = sqlite3.connect(database_to_insert)
+            self.configure_connection(conn)
             cursor = conn.cursor()
-            cursor.execute("PRAGMA journal_mode=WAL;")
+            #cursor.execute("PRAGMA journal_mode=WAL;")
             cursor.execute("INSERT INTO data(key, value) VALUES (?, ?);", (key, pickle.dumps(value)))
             conn.commit()
             conn.close()
@@ -71,8 +76,9 @@ class Collection_Cache:
             #print(f"Updating key '{key}' in {self.keys_databases[key]}...")
             database_to_update = self.keys_databases[key]
             conn = sqlite3.connect(database_to_update)
+            self.configure_connection(conn)
             cursor = conn.cursor()
-            cursor.execute("PRAGMA journal_mode=WAL;")
+            #cursor.execute("PRAGMA journal_mode=WAL;")
             cursor.execute("UPDATE data SET value = ? WHERE key = ?;", (pickle.dumps(value), key))
             conn.commit()
             conn.close()
@@ -82,14 +88,21 @@ class Collection_Cache:
         self.keys_databases[key] = database
         #print(self.keys_databases)
 
+    def delete_to_keys_database(self, key):
+        """Removes the key from the dictionary of stored keys"""
+        if key in self.keys_databases:
+            del self.keys_databases[key]
+
     def get_key(self, key):
+        """Used to obtain the value stored by the key"""
         try:
             database_to_search = self.keys_databases[key]
             #print(database_to_search)
 
             conn = sqlite3.connect(database_to_search)
+            self.configure_connection(conn)
             cursor = conn.cursor()
-            cursor.execute("PRAGMA journal_mode=WAL;")
+            #cursor.execute("PRAGMA journal_mode=WAL;")
             cursor.execute("SELECT value FROM data WHERE key = ?", (key,))
             result = cursor.fetchall()
             conn.close()
@@ -97,3 +110,29 @@ class Collection_Cache:
             
         except Exception as error:
             return error
+
+    def delete_key(self, key):
+        """Used to delete the value stored by the key"""
+        try:
+            database_to_delete = self.keys_databases[key]
+            #print(database_to_search)
+
+            conn = sqlite3.connect(database_to_delete)
+            self.configure_connection(conn)
+            cursor = conn.cursor()
+            #cursor.execute("PRAGMA journal_mode=WAL;")
+            cursor.execute("DELETE FROM data WHERE key = ?", (key,))
+            conn.commit()
+            conn.close()
+            self.delete_to_keys_database(key)
+
+        except KeyError:
+            return f"Key '{key}' not found."
+            
+        except Exception as error:
+            return error
+
+    def configure_connection(self, conn):
+        conn.execute("PRAGMA journal_mode = WAL;")
+        conn.execute("PRAGMA synchronous = NORMAL;")
+        conn.execute("PRAGMA wal_autocheckpoint = 1000;")

{collections_cache-0.1.3.dist-info → collections_cache-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: collections-cache
-Version: 0.1.3
+Version: 0.1.5
 Summary: Collection Cache is a Python package for managing data collections across multiple SQLite databases. It allows efficient storage, retrieval, and updating of key-value pairs, supporting various data types serialized with pickle. The package uses parallel processing for fast access and manipulation of large collections.
 License: MIT
 Author: Luiz-Trindade

collections_cache-0.1.5.dist-info/RECORD

@@ -0,0 +1,6 @@
+collections_cache/__init__.py,sha256=uUp8lhp-HnZRumnU_8MT6qVq95t0pOzn7oLW7ARbnvc,48
+collections_cache/collections_cache.py,sha256=y85cIYXZW9j_CxKqnrV2OnwbBK6s9fwaBY5-woe32Ro,5288
+collections_cache-0.1.5.dist-info/LICENSE,sha256=RAIL-FmXSiNRgyiVlfhm2SvVI4XDVsN0jDt9207SJ8o,1168
+collections_cache-0.1.5.dist-info/METADATA,sha256=dtrLBINwp9heD8SomfCfx2hxoXhyUu0jfFg10kqZiYw,3199
+collections_cache-0.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+collections_cache-0.1.5.dist-info/RECORD,,

collections_cache-0.1.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-collections_cache/__init__.py,sha256=uUp8lhp-HnZRumnU_8MT6qVq95t0pOzn7oLW7ARbnvc,48
-collections_cache/collections_cache.py,sha256=Lqm9D-UAjPYHKc28yJtdpKpWORxrTGmg7t_1QCC8Pq4,3812
-collections_cache-0.1.3.dist-info/LICENSE,sha256=RAIL-FmXSiNRgyiVlfhm2SvVI4XDVsN0jDt9207SJ8o,1168
-collections_cache-0.1.3.dist-info/METADATA,sha256=96NFbdna6GVAYPOTbOi5kk13SOEaUwsHLUKEWMg-iAs,3199
-collections_cache-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-collections_cache-0.1.3.dist-info/RECORD,,