aboutsummaryrefslogtreecommitdiffstats
path: root/todb.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--todb.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/todb.py b/todb.py
new file mode 100644
index 0000000..d911a29
--- /dev/null
+++ b/todb.py
@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+
+import os
+import psycopg
+
+pg_db = os.environ['el_pg_db']
+pg_host = os.environ['el_pg_host']
+pg_user = os.environ['el_pg_user']
+pg_pass = os.environ['el_pg_pass']
+
+def todb(sql, values):
+ with psycopg.connect(dbname=pg_db, host=pg_host, user=pg_user, password=pg_pass) as conn:
+ if type(values) == list:
+ conn.executemany(sql, values)
+ elif type(values) == tuple:
+ conn.execute(sql, values)
+ else:
+ print("OH SHIT")
+ exit()