def connect(db_name=settings['db_name']):
if db_type == 'sqlite':
try:
return sqlite3.connect(db_name, check_same_thread = False)
except OperationalError:
from utilsservlet import create_db
create_db(settings['db_name'])
return sqlite3.connect(db_name)
elif db_type == 'mysql':
try:
return pymysql.connect(host=settings['db_host'],
port=settings['db_port'],
user=settings['db_username'],
passwd=settings['db_password'],
db=settings['db_name'])
except pymysql.err.InternalError, e:
print "MySQL internal error - ", e[0], ": ", e[1]
pass
def connect(db_name=settings['db_name']):
if db_type == 'sqlite':
try:
return sqlite3.connect(db_name, check_same_thread=False)
except OperationalError:
from utilsservlet import create_db
create_db(settings['db_name'])
return sqlite3.connect(db_name)
elif db_type == 'mysql':
try:
return pymysql.connect(host=settings['db_host'],
port=settings['db_port'],
user=settings['db_username'],
passwd=settings['db_password'],
db=settings['db_name'])
except pymysql.err.InternalError, e:
print "MySQL internal error - ", e[0], ": ", e[1]
pass
#!/usr/bin/env python
# -*- coding: utf-8
import hashlib, time, sqlite3
from utilsservlet import rand_alpha_numeric, create_db
try:
create_db("sess")
except:
pass
class UserSession(object):
def __init__(self, key, uphash, ip, data='', currtime=time.asctime(time.gmtime())):
self.creation_time = currtime
self.key = key
self.uphash = uphash
self.ip = ip
self.data = data
def __str__(self):
return "%s\n%s\n%s\n%s\n%s" % (self.key, self.uphash, self.ip, self.data, self.creation_time)
class SessionDB:
@staticmethod
def add(session):
print session
conn = sqlite3.connect("sess")
cur = conn.cursor()
cur.execute("insert into sessions values(?,?,?,?,?)",
(session.key, session.uphash, session.ip, session.data, session.creation_time))
conn.commit()
del session
