From 6f48bf541b7b0eeb0c6968ca63125bf73e016643 Mon Sep 17 00:00:00 2001 From: Walter Purcaro Date: Tue, 12 May 2015 03:15:13 +0200 Subject: Fix some directory names --- pyload/Database/Storage.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 pyload/Database/Storage.py (limited to 'pyload/Database/Storage.py') diff --git a/pyload/Database/Storage.py b/pyload/Database/Storage.py new file mode 100644 index 000000000..a19f67606 --- /dev/null +++ b/pyload/Database/Storage.py @@ -0,0 +1,36 @@ +# -*- coding: utf-8 -*- +# @author: mkaay + +from pyload.Database import style +from pyload.Database import DatabaseBackend + + +class StorageMethods(object): + + @style.queue + def setStorage(db, identifier, key, value): + db.c.execute("SELECT id FROM storage WHERE identifier=? AND key=?", (identifier, key)) + if db.c.fetchone() is not None: + db.c.execute("UPDATE storage SET value=? WHERE identifier=? AND key=?", (value, identifier, key)) + else: + db.c.execute("INSERT INTO storage (identifier, key, value) VALUES (?, ?, ?)", (identifier, key, value)) + + + @style.queue + def getStorage(db, identifier, key=None): + if key is not None: + db.c.execute("SELECT value FROM storage WHERE identifier=? AND key=?", (identifier, key)) + row = db.c.fetchone() + if row is not None: + return row[0] + else: + db.c.execute("SELECT key, value FROM storage WHERE identifier=?", (identifier,)) + return {row[0]: row[1] for row in db.c} + + + @style.queue + def delStorage(db, identifier, key): + db.c.execute("DELETE FROM storage WHERE identifier=? AND key=?", (identifier, key)) + + +DatabaseBackend.registerSub(StorageMethods) -- cgit v1.2.3