1 files changed, 81 insertions, 0 deletions
diff --git a/Db/__init__.py b/Db/__init__.py
new file mode 100644
index 0000000..c42a7c5
--- /dev/null
+++ b/Db/__init__.py
@@ -0,0 +1,81 @@
+# coding: utf-8
+import time, sys
+HOST = "lalalizard.com"
+USER = "asdfus"
+PASSWORD = "gTYgT&M6q"
+DATABASE = "asdfus"
+
+from sqlalchemy import Column, Integer, LargeBinary, String, create_engine, sql 
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+_NULL = sql.null()
+
+Base = declarative_base()
+metadata = Base.metadata
+
+class ImCmd(Base):
+    __tablename__ = 'im_cmd'
+    id = Column(Integer, primary_key=True)
+    date = Column(Integer)
+    remote_addr = Column(String(16))
+    name = Column(String(16))
+    url = Column(String(256))
+    dir = Column(String(2))
+    oldfile = Column(String(256))
+    newfile = Column(String(256))
+    cmd = Column(LargeBinary)
+    dataobj = Column(LargeBinary)
+    tag = Column(String(50))
+
+class Db(object):
+  def __init__(self):
+    engine = create_engine('mysql://{}:{}@{}/{}'.format(
+      USER, 
+      PASSWORD, 
+      HOST,
+      DATABASE
+    ))
+    Session = sessionmaker(bind=engine)
+    self.session = Session()
+
+  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):
+
+#    sys.stderr.write("date={}".format(str(date)) + "\n")
+#    sys.stderr.write("remote_addr={}".format(str(remote_addr)) + "\n")
+#    sys.stderr.write("name={}".format(str(username)) + "\n")
+#    sys.stderr.write("url={}".format(str(url)) + "\n")
+#    sys.stderr.write("dir={}".format(str(directory)) + "\n")
+#    sys.stderr.write("oldfile={}".format(str(oldfile)) + "\n")
+#    sys.stderr.write("newfile={}".format(str(newfile)) + "\n")
+#    sys.stderr.write("cmd={}".format(str(cmd)) + "\n")
+#    sys.stderr.write("dataobj={}".format(str(dataobj)) + "\n")
+#    sys.stderr.write("tag={}".format(str(tag)) + "\n")
+    try:
+      entry = ImCmd(
+        date=date,
+        remote_addr=remote_addr,
+        name=username,
+        url=url,
+        dir=directory,
+        oldfile=oldfile,
+        newfile=newfile,
+        cmd=cmd,
+        dataobj=dataobj,
+        tag=tag
+      )
+      self.session.add(entry)
+      self.session.commit()
+    except Exception as e:
+      sys.stderr.write("Unable to commit database entry");
+      sys.stderr.write(str(e))