database.py

database codebase

database.py

@@ -0,0 +1,27 @@
+# database.py
+import psycopg2
+from psycopg2.extras import RealDictCursor
+
+def connect_db():
+    conn = psycopg2.connect(
+        dbname="solana_bot_db",
+        user="your_user",
+        password="your_password",
+        host="localhost",
+        port="5432"
+    )
+    return conn
+
+def insert_token_data(token_data):
+    query = """
+    INSERT INTO tokens (id, name, launch_time, liquidity, volume)
+    VALUES (%s, %s, %s, %s, %s)
+    ON CONFLICT (id) DO NOTHING;
+    """
+    conn = connect_db()
+    with conn.cursor() as cur:
+        for token in token_data:
+            cur.execute(query, (token['id'], token['name'], token['launch_time'],
+                                token['liquidity'], token['volume']))
+    conn.commit()
+    conn.close()