From 2bccf939863b729fa86fe31a09f7f61ba50fff6b Mon Sep 17 00:00:00 2001
From: Maxime Bonin <maximebonin@outlook.com>
Date: Tue, 13 Jun 2023 20:10:29 -0400
Subject: [PATCH] new module database

---
 apis/pastebin/.gitignore       |  3 ++
 apis/pastebin/db.py            | 83 ++++++++++++++++++++++++++++++++++
 apis/pastebin/main.py          |  4 ++
 apis/pastebin/requirements.txt |  3 +-
 4 files changed, 92 insertions(+), 1 deletion(-)
 create mode 100644 apis/pastebin/db.py

diff --git a/apis/pastebin/.gitignore b/apis/pastebin/.gitignore
index 68bc17f..4374cc2 100644
--- a/apis/pastebin/.gitignore
+++ b/apis/pastebin/.gitignore
@@ -158,3 +158,6 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+# pastebin project stuff
+*.db
\ No newline at end of file
diff --git a/apis/pastebin/db.py b/apis/pastebin/db.py
new file mode 100644
index 0000000..fd089ed
--- /dev/null
+++ b/apis/pastebin/db.py
@@ -0,0 +1,83 @@
+import sqlite3
+from sqlite3 import Connection
+import logging
+
+import base64, hashlib
+import datetime, time
+
+logging.basicConfig(format="%(process)d-%(levelname)s-%(message)s", level=logging.DEBUG)
+
+
+class Paste:
+    def __init__(self, paste, badge: str = ""):
+        self.id = self._hash(paste)
+        self.paste = paste
+        self.badge = badge
+        self.insert_timestamp = time.mktime(datetime.datetime.now().timetuple())
+
+    def _hash(self, val):
+        encoded_val = val.encode("utf-8")
+        h = hashlib.md5(encoded_val)
+        ret = base64.urlsafe_b64encode(h.digest()[:6])
+        return bytes.decode(ret, "utf-8")
+
+
+class DB:
+    def __init__(self, db_path: str = "default.db") -> None:
+        self._database = db_path
+        self._conn = self._create_connection(db_path)
+        self._create_pastebin_table()
+
+    def __del__(self):
+        if self._conn:
+            self._conn.close()
+
+    def insert_paste(self, paste: Paste) -> str:
+        sql = """ INSERT INTO pastebin(id,paste,insert_timestamp,badge)
+                    VALUES(?,?,?,?) """
+        try:
+            self._conn.cursor()
+            self._conn.cursor().execute(
+                sql, (paste.id, paste.paste, paste.insert_timestamp, paste.badge)
+            )
+            self._conn.commit()
+        except sqlite3.Error as e:
+            logging.error(
+                "unable to insert new paste in sqlite3 database", exc_info=True
+            )
+
+        return paste.id
+
+    def _create_pastebin_table(self) -> None:
+        sql_create_pastebin_table = """ CREATE TABLE IF NOT EXISTS pastebin (
+                                            id text PRIMARY KEY,
+                                            paste BLOB,
+                                            insert_timestamp REAL,
+                                            badge TEXT
+                                        ); """
+
+        try:
+            c = self._conn.cursor()
+            c.execute(sql_create_pastebin_table)
+        except sqlite3.Error as e:
+            logging.error("Error creating sqlite3 pastebin table", exc_info=True)
+
+    def _create_connection(self, db_file) -> Connection:
+        """returns a database connection to a SQLite database"""
+        conn = None
+        try:
+            conn = sqlite3.connect(db_file)
+            print(sqlite3.version)
+        except sqlite3.Error as e:
+            logging.error("Error creating sqlite3 database connection", exc_info=True)
+
+        return conn
+
+
+# very temporary: used to test the module locally
+#
+# if __name__ == "__main__":
+#     db = DB(r"pastebin.db")
+#     p = Paste(paste="zoom zoom", badge="mon ami")
+#     r = db.insert_paste(paste=p)
+#     print(r)
diff --git a/apis/pastebin/main.py b/apis/pastebin/main.py
index 30d4a45..44ff316 100644
--- a/apis/pastebin/main.py
+++ b/apis/pastebin/main.py
@@ -2,6 +2,10 @@
 
 from fastapi import FastAPI
 
+import logging
+
+logging.basicConfig(format="%(process)d-%(levelname)s-%(message)s", level=logging.debug)
+
 app = FastAPI()
 
 
diff --git a/apis/pastebin/requirements.txt b/apis/pastebin/requirements.txt
index 4a39716..bbd341e 100644
--- a/apis/pastebin/requirements.txt
+++ b/apis/pastebin/requirements.txt
@@ -1,3 +1,4 @@
 black
 fastapi
-uvicorn[standard]
\ No newline at end of file
+uvicorn[standard]
+db-sqlite3
\ No newline at end of file