91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
'''
|
|
Folding@Home Client Control (FAHControl)
|
|
Copyright (C) 2010-2014 Stanford University
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
'''
|
|
|
|
from fah.db import Column
|
|
|
|
|
|
class Table:
|
|
def __init__(self, name, cols, constraints = ''):
|
|
self.name = name
|
|
self.cols = cols
|
|
self.constraints = constraints
|
|
|
|
|
|
def where(self, **kwargs):
|
|
if len(kwargs) == 0: return ''
|
|
sql = 'WHERE '
|
|
|
|
if len(kwargs) == 1 and 'where' in kwargs:
|
|
sql += kwargs['where']
|
|
|
|
else:
|
|
sql +=\
|
|
' AND '.join(map(lambda i: '"%s"=\'%s\'' % i, kwargs.items()))
|
|
|
|
return sql
|
|
|
|
|
|
def create(self, db):
|
|
sql = 'CREATE TABLE IF NOT EXISTS "%s" (%s' % (
|
|
self.name, ','.join(map(Column.get_sql, self.cols)))
|
|
|
|
if self.constraints: sql += ',%s' % self.constraints
|
|
sql += ')'
|
|
|
|
db.execute(sql).close()
|
|
|
|
|
|
def insert(self, db, **kwargs):
|
|
cols = filter(lambda col: col.name in kwargs, self.cols)
|
|
|
|
# Error checking
|
|
if len(cols) != len(kwargs):
|
|
col_names = set(map(Column.get_name, cols))
|
|
missing = filter(lambda kw: not kw in col_names, kwargs.keys())
|
|
raise Exception, 'Table %s does not have column(s) %s' % (
|
|
self.name, ', '.join(missing))
|
|
|
|
sql = 'REPLACE INTO "%s" ("%s") VALUES (%s)' % (
|
|
self.name, '","'.join(map(Column.get_name, cols)),
|
|
','.join(map(lambda col: col.get_db_value(kwargs[col.name]), cols)))
|
|
|
|
db.execute(sql).close()
|
|
|
|
|
|
def select(self, db, cols = None, **kwargs):
|
|
if cols is None:
|
|
cols = '"' + '","'.join(map(str, self.cols)) + '"'
|
|
|
|
sql = 'SELECT %s FROM %s' % (cols, self.name)
|
|
if 'orderby' in kwargs:
|
|
sql += ' ORDER BY ' + kwargs['orderby']
|
|
del kwargs['orderby']
|
|
|
|
sql += ' ' + self.where(**kwargs)
|
|
|
|
return db.execute(sql)
|
|
|
|
|
|
def delete(self, db, **kwargs):
|
|
sql = 'DELETE FROM %s %s' % (self.name, self.where(**kwargs))
|
|
db.execute(sql).close()
|
|
|
|
|
|
def drop(self, db):
|
|
db.execute('DROP TABLE IF EXISTS ' + self.name).close()
|