diff options
| author | pepperpepperpepper <pepper@scannerjammer.com> | 2015-10-09 15:52:02 -0700 |
|---|---|---|
| committer | pepperpepperpepper <pepper@scannerjammer.com> | 2015-10-09 15:52:02 -0700 |
| commit | 42da3b72a8ee7deac3abe238fbdb1591d8d945b9 (patch) | |
| tree | 004eff4289ff5e01d8d66fbb29942cb2f94f47a9 /lib/db.py | |
| parent | 0c965a54861e779ca51b235e7a3c2475d8a84dfb (diff) | |
alright got rid of most relative paths
Diffstat (limited to 'lib/db.py')
| -rw-r--r-- | lib/db.py | 66 |
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)) |
