summaryrefslogtreecommitdiff
path: root/lib/db.py
diff options
context:
space:
mode:
authorpepperpepperpepper <pepper@scannerjammer.com>2015-10-09 15:52:02 -0700
committerpepperpepperpepper <pepper@scannerjammer.com>2015-10-09 15:52:02 -0700
commit42da3b72a8ee7deac3abe238fbdb1591d8d945b9 (patch)
tree004eff4289ff5e01d8d66fbb29942cb2f94f47a9 /lib/db.py
parent0c965a54861e779ca51b235e7a3c2475d8a84dfb (diff)
alright got rid of most relative paths
Diffstat (limited to 'lib/db.py')
-rw-r--r--lib/db.py66
1 files changed, 0 insertions, 66 deletions
diff --git a/lib/db.py b/lib/db.py
deleted file mode 100644
index 6a7e424..0000000
--- a/lib/db.py
+++ /dev/null
@@ -1,66 +0,0 @@
-# coding: utf-8
-"""all database connections and logic goes here"""
-from config import DB_HOST, DB_USER, DB_PASSWORD, DB_NAME
-import time, sys
-import simplejson as json
-
-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):
- """defines the table class"""
- __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):
- """wrapper for all db methods"""
- def __init__(self):
- engine = create_engine('mysql://{}:{}@{}/{}'.format(
- DB_USER,
- DB_PASSWORD,
- DB_HOST,
- DB_NAME
- ))
- self.Session = sessionmaker(bind=engine)
-
- def search(self, **kwargs):
- session = self.Session()
- results = session.query(ImCmd).filter_by(**kwargs).all()
- return map(lambda n: { "tag": n.tag, "data": json.loads(n.dataobj) }, results)
-
- def insert_cmd(self, **kwargs):
- try:
- session = self.Session()
- _entry_data = {
- 'date' : kwargs.get("date", int(time.time())),
- 'remote_addr' : kwargs['remote_addr'] or _NULL,
- 'name' : kwargs['username'] or _NULL,
- 'url' : kwargs['username'] or _NULL,
- 'dir' : kwargs['directory'] or _NULL,
- 'oldfile' : kwargs['oldfile'] or _NULL,
- 'newfile' : kwargs['newfile'] or _NULL,
- 'cmd' : kwargs['cmd'] or _NULL,
- 'dataobj' : kwargs['dataobj'] or _NULL,
- 'tag' : kwargs['tag'] or _NULL
- }
- session.add(ImCmd(**_entry_data))
- session.commit()
- #FIXME session.close()....
- except Exception as e:
- sys.stderr.write("Unable to commit database entry\n")
- sys.stderr.write(str(e))