1 files changed, 25 insertions, 2 deletions
diff --git a/pb/lib/db.py b/pb/lib/db.py
index 2570b4f..66db35d 100755
--- a/pb/lib/db.py
+++ b/pb/lib/db.py
@@ -3,7 +3,7 @@ USER = "asdfus"
 PASSWORD = "gTYgT&M6q"
 DATABASE = "asdfus"
 
-class db:
+class Db(object):
         def __init__ (self):
                 self.conn = None
                 self.cursor = None
@@ -27,5 +27,28 @@ class db:
                         self.cursor.execute(sql,args)
 
         def lastinsertid (self):
-                return DB.conn.insert_id()
+                return self.conn.insert_id()
 
+        def insert_cmd (
+            self, 
+            date=time.time(), 
+            remote_addr="NULL", 
+            username="NULL", 
+            url="NULL", 
+            directory="NULL", 
+            oldfile="NULL", 
+            newfile="NULL", 
+            cmd="NULL", 
+            dataobj="NULL", 
+            tag="NULL"):
+            try:
+                sql = "INSERT INTO im_cmd "
+                sql += "(date, remote_addr, name, url, dir, oldfile, newfile, cmd, dataobj, tag) "
+                sql += "VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
+                #or "NULL"
+                args = (date, remote_addr, username, url, directory, oldfile, newfile, cmd, dataobj, tag)
+                #args = (now(), os.environ['REMOTE_ADDR'], name, url, dir, oldfile, newfile, " ".join(cmd),dataobj)
+                self.execute(sql, args)
+            except Exception as e:
+                sys.stderr.write(str(e))
+                return