diff options
Diffstat (limited to 'pb/lib')
| -rw-r--r-- | pb/lib/Db/__init__.py | 71 | ||||
| -rw-r--r-- | pb/lib/Db/sqlalchemy_example_code.py | 105 | ||||
| -rw-r--r-- | pb/lib/Utils/__init__.py | 123 | ||||
| -rw-r--r-- | pb/lib/__init__.py | 0 |
4 files changed, 0 insertions, 299 deletions
diff --git a/pb/lib/Db/__init__.py b/pb/lib/Db/__init__.py deleted file mode 100644 index ac7ca17..0000000 --- a/pb/lib/Db/__init__.py +++ /dev/null @@ -1,71 +0,0 @@ -# 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): - 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)) diff --git a/pb/lib/Db/sqlalchemy_example_code.py b/pb/lib/Db/sqlalchemy_example_code.py deleted file mode 100644 index fc38b38..0000000 --- a/pb/lib/Db/sqlalchemy_example_code.py +++ /dev/null @@ -1,105 +0,0 @@ -# coding: utf-8 -import MySQLdb -import time, sys -USER = "asdfus" -PASSWORD = "gTYgT&M6q" -DATABASE = "asdfus" - - -from sqlalchemy import Column, Integer, LargeBinary, String, create_engine -from sqlalchemy.ext.declarative import declarative_base -from sqlalchemy.orm import sessionmaker -engine = create_engine('mysql://{}:{}@localhost/{}'.format(USER, PASSWORD, DATABASE)) -Session = sessionmaker(bind=engine) -session = Session() -connection = engine.connect() -#result = connection.execute("select username from users") -#for row in result: -# print "username:", row['username'] -#connection.close() - - -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)) -#def __repr__(self): -#... return "<User(name='%s', fullname='%s', password='%s')>" % ( -#... self.name, self.fullname, self.password) - - -for instance in session.query(ImCmd).order_by(ImCmd.id): - print instance.name, instance.date - - -ed_user = User(name='ed', fullname='Ed Jones', password='edspassword') -session.add(ed_user) -#session.add_all([ -#... User(name='wendy', fullname='Wendy Williams', password='foobar'), -#... User(name='mary', fullname='Mary Contrary', password='xxg527'), -#... User(name='fred', fullname='Fred Flinstone', password='blah')]) - -#session.commit() - -class Db(object): - def __init__ (self): - self.conn = None - self.cursor = None - self.connect() - - def connect (self): - self.conn = MySQLdb.connect (host = "lalalizard.com", - user = USER, - passwd = PASSWORD, - db = DATABASE - ) - self.cursor = self.conn.cursor () - - def execute (self,sql,args=()): - try: - self.cursor.execute(sql,args) - except MySQLdb.Error, e: - print "Error %d: %s" % (e.args[0], e.args[1]) - # sys.exit (1) - self.connect() - self.cursor.execute(sql,args) - - def lastinsertid (self): - 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 diff --git a/pb/lib/Utils/__init__.py b/pb/lib/Utils/__init__.py deleted file mode 100644 index 1d5a708..0000000 --- a/pb/lib/Utils/__init__.py +++ /dev/null @@ -1,123 +0,0 @@ -import re -from pb.Config import * -import time -import urllib -import urllib2 -import sys -import os -from subprocess import Popen,PIPE,call -Request = urllib2.Request -urlencode = urllib.urlencode -urlopen = urllib2.urlopen - -def call_cmd(cmd, error=""): - try: - call(cmd) - except Exception as e: - raise (str(e)) - -def is_color(s): - if s == "": - return "transparent" - if re.match('(rgba?\([0-9]+,[0-9]+,[0-9]+\))|([a-zA-Z]+)|(\#[A-Ha-h0-9]+)', s): - return s.replace(' ', ''); - else: - sys.stderr.write("Not a color: {}\n".format(s)) - raise ValueError - -def dimensions (filepath): - #works in lieu of a mimetype check (it reads the header as well) - ident = (Popen([BIN_IDENTIFY, filepath], stdout=PIPE).communicate()[0]).split(" ") - return ident[2].split("x") - -def is_number(s): - try: - return int(s) - except (ValueError, TypeError): - return False - -def bool_correct(s): - if re.match(r'^false$', s, re.IGNORECASE): - return False - elif re.match(r'^true$', s, re.IGNORECASE): - return True - else: - return s - -class dotdict(dict): - """dot.notation access to dictionary attributes""" - def __getattr__(self, attr): - return self.get(attr) - __setattr__= dict.__setitem__ - __delattr__= dict.__delitem__ - -def get_mimetype(f): - try: - mimetype = Popen( - [BIN_IDENTIFY, f], stdout=PIPE - ).communicate()[0].split(" ")[1].lower() - return mimetype - except Exception as e: - sys.stderr.write("couldn't determine mimetype") - sys.stderr.write(str(e)) - raise; - -def sanitize (str): - return re.sub(r'\W+', '', str) - -def now(): - return int(time.time()) - -def browser_request (url, data=None): - headers = { - 'User-Agent': 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)', - 'Accept': '*/*', - } - try: - req = Request(url, data, headers) - response = urlopen(req) - except IOError, e: - if hasattr(e, 'code'): - sys.stderr.write( '%s - ERROR %s' % (url, e.code) ) - raise; - return None - else: - return response - -def download(url, destination, max_size=MAX_SIZE): - response = browser_request(url, None) - rawimg = response.read() - if len(rawimg) == 0: - sys.stderr.write("got zero-length file") - raise; - if len(rawimg) > max_size: - sys.stderr.write("file too big: max size {} KB / {} is {} KB".format( - str(MAX_SIZE/1024), - destination, - str(len(rawimg)/1024) - )) - raise; - f = open(destination, "w") - f.write(rawimg) - f.close() - -def file_size (filepath): - try: - return os.stat(filepath)[6] - except Exception as e: - sys.stderr.write("Couldn't determine filesize\n") - sys.stderr.write(str(e)+"\n") - raise; - -def gif_frames(filepath): - try: - info = Popen([BIN_IDENTIFY,filepath], stdout=PIPE).communicate()[0] - frames = filter((lambda x: x), map( - (lambda x: x.split(" ")[0]), - (info).split('\n') - )) - return frames - except Exception as e: - sys.stderr.write("IMGRID couldn't get gif frames") - sys.stderr.write(str(e)) - raise; diff --git a/pb/lib/__init__.py b/pb/lib/__init__.py deleted file mode 100644 index e69de29..0000000 --- a/pb/lib/__init__.py +++ /dev/null |
