Made Enten permanENTEn

FaustBot/Model/DuckProvider.py

@@ -0,0 +1,36 @@
+import sqlite3
+
+
+class DucksProvider(object):
+    _CREATE_TABLE = 'CREATE TABLE IF NOT EXISTS ducks (id INTEGER PRIMARY KEY, \
+                    user TEXT, friends INTEGER, dead INTEGER)'
+    _GET_DUCKS = 'SELECT id, friends, dead FROM ducks WHERE user = ?'
+    _SAVE_OR_OVERWRITE = 'REPLACE INTO ducks (id, user, friends, dead) VALUES (?, ?, ?,?)'
+    _DELETE_DUCKS = 'DELETE FROM ducks WHERE user = ?'
+
+    def __init__(self):
+        self._database_connection = sqlite3.connect('faust_bot.db')
+        cursor = self._database_connection.cursor()
+        cursor.execute(DucksProvider._CREATE_TABLE)
+        self._database_connection.commit()
+
+    def get_ducks(self, user: str):
+        cursor = self._database_connection.cursor()
+        cursor.execute(DucksProvider._GET_DUCKS, (user.lower(),))
+        return cursor.fetchone()
+
+    def save_or_replace(self, user: str, friends: int, dead: int):
+        existing = self.get_ducks(user)
+        _id = existing[0] if existing is not None else None
+        data = (_id, user.lower(), friends, dead)
+        cursor = self._database_connection.cursor()
+        cursor.execute(DucksProvider._SAVE_OR_OVERWRITE, data)
+        self._database_connection.commit()
+
+    def delete_score(self, user: str):
+        cursor = self._database_connection.cursor()
+        cursor.execute(DucksProvider._DELETE_DUCKS, (user.lower(),))
+        self._database_connection.commit()
+
+    def __exit__(self):
+        self._database_connection.close()