aboutsummaryrefslogtreecommitdiff
path: root/cgi/database.py
diff options
context:
space:
mode:
Diffstat (limited to 'cgi/database.py')
-rw-r--r--cgi/database.py102
1 files changed, 54 insertions, 48 deletions
diff --git a/cgi/database.py b/cgi/database.py
index c8611c5..9b2c1e7 100644
--- a/cgi/database.py
+++ b/cgi/database.py
@@ -7,63 +7,69 @@ from settings import Settings
database_lock = threading.Lock()
try:
- # Although SQLAlchemy is optional, it is highly recommended
- import sqlalchemy.pool as pool
- _mysql = pool.manage( module = _mysql,
- pool_size = Settings.DATABASE_POOL_SIZE,
- max_overflow = Settings.DATABASE_POOL_OVERFLOW)
- Settings._.USING_SQLALCHEMY = True
+ # Although SQLAlchemy is optional, it is highly recommended
+ import sqlalchemy.pool as pool
+ _mysql = pool.manage(module=_mysql,
+ pool_size=Settings.DATABASE_POOL_SIZE,
+ max_overflow=Settings.DATABASE_POOL_OVERFLOW)
+ Settings._.USING_SQLALCHEMY = True
except ImportError:
- pass
+ pass
+
def OpenDb():
- if Settings._.CONN is None:
- Settings._.CONN = _mysql.connect(host = Settings.DATABASE_HOST,
- user = Settings.DATABASE_USERNAME,
- passwd = Settings.DATABASE_PASSWORD,
- db = Settings.DATABASE_DB)
+ if Settings._.CONN is None:
+ Settings._.CONN = _mysql.connect(host=Settings.DATABASE_HOST,
+ user=Settings.DATABASE_USERNAME,
+ passwd=Settings.DATABASE_PASSWORD,
+ db=Settings.DATABASE_DB)
+
def FetchAll(query, method=1):
- """
- Query and fetch all results as a list
- """
- db = Settings._.CONN
-
- db.query(query)
- r = db.use_result()
- return r.fetch_row(0, method)
+ """
+ Query and fetch all results as a list
+ """
+ db = Settings._.CONN
+
+ db.query(query)
+ r = db.use_result()
+ return r.fetch_row(0, method)
+
def FetchOne(query, method=1):
- """
- Query and fetch only the first result
- """
- db = Settings._.CONN
-
- db.query(query)
- r = db.use_result()
- try:
- return r.fetch_row(1, method)[0]
- except:
- return None
+ """
+ Query and fetch only the first result
+ """
+ db = Settings._.CONN
+
+ db.query(query)
+ r = db.use_result()
+ try:
+ return r.fetch_row(1, method)[0]
+ except:
+ return None
+
def UpdateDb(query):
- """
- Update the DB (UPDATE/DELETE) and return # of affected rows
- """
- db = Settings._.CONN
-
- db.query(query)
- return db.affected_rows()
-
+ """
+ Update the DB (UPDATE/DELETE) and return # of affected rows
+ """
+ db = Settings._.CONN
+
+ db.query(query)
+ return db.affected_rows()
+
+
def InsertDb(query):
- """
- Insert into the DB and return the primary key of new row
- """
- db = Settings._.CONN
-
- db.query(query)
- return db.insert_id()
+ """
+ Insert into the DB and return the primary key of new row
+ """
+ db = Settings._.CONN
+
+ db.query(query)
+ return db.insert_id()
+
def CloseDb():
- if Settings._.CONN is not None:
- Settings._.CONN.close()
+ if Settings._.CONN is not None:
+ Settings._.CONN.close()