adding some common stuff, less code replica

1 files changed, 2 insertions, 0 deletions

diff --git a/todb.py b/todb.py
index e5988cd..7426ace 100644
--- a/todb.py
+++ b/todb.py
@@ -13,8 +13,10 @@ def todb(sql, values):
         if type(values) == list:
             cur = conn.cursor()
             cur.executemany(sql, values)
+            return cur.rowcount
         elif type(values) == tuple: 
             conn.execute(sql, values)
+            return 1
         else:
             print("OH SHIT")
             exit()