initial commit

pgbotlib/dbstuff.py

@@ -0,0 +1,22 @@
+import random
+import psycopg
+
+
+class DBConn:
+    def __init__(self, *args, **kwargs) -> None:
+        self.connection = psycopg.connect(*args, **kwargs)
+        self.cursor = self.connection.cursor()
+
+    def update(self, query: str, values: tuple) -> list:
+        self.cursor.execute(query, values)
+        return self.connection.commit()
+
+    def query_raw(self, query: str, values: tuple) -> list:
+        self.cursor.execute(query, values)
+        return self.cursor.fetchall()
+
+    def query_random(self, query: str, values: tuple) -> str:
+        result = self.query_raw(query, values)
+        if not result:
+            return None
+        return random.choice(result)[0]