Remove unneeded files
This commit is contained in:
parent
ac09c9c8d8
commit
d59733533b
3 changed files with 0 additions and 176 deletions
|
@ -1,57 +0,0 @@
|
||||||
import sqlite3
|
|
||||||
|
|
||||||
class Database(object):
|
|
||||||
"""Interface to a local database that stores data for the website.
|
|
||||||
"""
|
|
||||||
def __init__(self, dbname):
|
|
||||||
self.dbname = dbname
|
|
||||||
|
|
||||||
|
|
||||||
@property
|
|
||||||
def conn(self):
|
|
||||||
return sqlite3.connect(self.dbname)
|
|
||||||
|
|
||||||
|
|
||||||
@property
|
|
||||||
def cursor(self):
|
|
||||||
return self.conn.cursor()
|
|
||||||
|
|
||||||
|
|
||||||
def select(self, table, fields, condition=None):
|
|
||||||
cur = self.cursor
|
|
||||||
|
|
||||||
if condition is not None:
|
|
||||||
cur.execute("SELECT {} FROM {} WHERE {}".format(fields, table,
|
|
||||||
condition))
|
|
||||||
else:
|
|
||||||
cur.execute("SELECT {} FROM {}".format(fields, table))
|
|
||||||
|
|
||||||
return cur.fetchall()
|
|
||||||
|
|
||||||
|
|
||||||
def create_table(self, table, cols):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "CREATE TABLE {}({})".format(table, cols)
|
|
||||||
cur.execute(query)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def insert(self, table, columns, values, params):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "INSERT INTO {}({}) VALUES ({})".format(table, columns, values)
|
|
||||||
cur.execute(query, params)
|
|
||||||
conn.commit()
|
|
||||||
return cur
|
|
||||||
|
|
||||||
|
|
||||||
def delete(self, table, condition):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "DELETE FROM {} WHERE {}".format(table, condition)
|
|
||||||
cur.execute(query)
|
|
||||||
conn.commit()
|
|
|
@ -1,66 +0,0 @@
|
||||||
import os
|
|
||||||
import psycopg2
|
|
||||||
import urlparse
|
|
||||||
|
|
||||||
class DatabasePostgreSQL(object):
|
|
||||||
"""Interface to the heroku PostgreSQL database plugin.
|
|
||||||
"""
|
|
||||||
def __init__(self):
|
|
||||||
urlparse.uses_netloc.append("postgres")
|
|
||||||
self.url = urlparse.urlparse(os.environ["DATABASE_URL"])
|
|
||||||
|
|
||||||
|
|
||||||
@property
|
|
||||||
def conn(self):
|
|
||||||
return psycopg2.connect(
|
|
||||||
database=self.url.path[1:],
|
|
||||||
user=self.url.username,
|
|
||||||
password=self.url.password,
|
|
||||||
host=self.url.hostname,
|
|
||||||
port=self.url.port
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@property
|
|
||||||
def cursor(self):
|
|
||||||
return self.conn.cursor()
|
|
||||||
|
|
||||||
|
|
||||||
def select(self, table, fields, condition=None):
|
|
||||||
cur = self.cursor
|
|
||||||
|
|
||||||
if condition is not None:
|
|
||||||
cur.execute("SELECT {} FROM {} WHERE {}".format(fields, table,
|
|
||||||
condition))
|
|
||||||
else:
|
|
||||||
cur.execute("SELECT {} FROM {}".format(fields, table))
|
|
||||||
|
|
||||||
return cur.fetchall()
|
|
||||||
|
|
||||||
|
|
||||||
def create_table(self, table, cols):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "CREATE TABLE {}({})".format(table, cols)
|
|
||||||
cur.execute(query)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def insert(self, table, columns, values, params):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "INSERT INTO {}({}) VALUES ({})".format(table, columns, values)
|
|
||||||
cur.execute(query, params)
|
|
||||||
conn.commit()
|
|
||||||
return cur
|
|
||||||
|
|
||||||
|
|
||||||
def delete(self, table, condition):
|
|
||||||
conn = self.conn
|
|
||||||
cur = conn.cursor()
|
|
||||||
|
|
||||||
query = "DELETE FROM {} WHERE {}".format(table, condition)
|
|
||||||
cur.execute(query)
|
|
||||||
conn.commit()
|
|
|
@ -1,53 +0,0 @@
|
||||||
class Model(object):
|
|
||||||
#Name of the table to be created
|
|
||||||
tablename = "abstractmodel"
|
|
||||||
|
|
||||||
#Attributes are tuples - attribute name, data type, other requirements
|
|
||||||
id = ("id", "INTEGER", "PRIMARY KEY AUTOINCREMENT")
|
|
||||||
|
|
||||||
|
|
||||||
class Quote(Model):
|
|
||||||
tablename = "quotes"
|
|
||||||
|
|
||||||
id = ("id", "INTEGER", "PRIMARY KEY AUTOINCREMENT")
|
|
||||||
rating = ("rating", "INTEGER", "NOT NULL")
|
|
||||||
content = ("content", "TEXT", "NOT NULL")
|
|
||||||
approved = ("approved", "BOOLEAN", "NOT NULL")
|
|
||||||
author_ip = ("author_ip", "TEXT", "NOT NULL")
|
|
||||||
time = ("time", "TEXT", "NOT NULL")
|
|
||||||
|
|
||||||
|
|
||||||
class Tag(Model):
|
|
||||||
tablename = "tags"
|
|
||||||
|
|
||||||
id = ("id", "INTEGER", "PRIMARY KEY AUTOINCREMENT")
|
|
||||||
name = ("name", "TEXT", "UNIQUE NOT NULL")
|
|
||||||
|
|
||||||
|
|
||||||
class TagsToQuotes(Model):
|
|
||||||
tablename = "tagsToQuotes"
|
|
||||||
|
|
||||||
id = ("id", "INTEGER", "PRIMARY KEY AUTOINCREMENT")
|
|
||||||
tag_id = ("tagid", "INTEGER", "NOT NULL")
|
|
||||||
quote_id = ("quoteid", "INTEGER", "NOT NULL")
|
|
||||||
|
|
||||||
|
|
||||||
def init_models(db):
|
|
||||||
import logging
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
for model in Model.__subclasses__():
|
|
||||||
columns = [x for x in model.__dict__ if '__' not in x and x !=
|
|
||||||
'tablename']
|
|
||||||
|
|
||||||
try:
|
|
||||||
db.create_table(
|
|
||||||
model.tablename,
|
|
||||||
','.join([' '.join(model.__dict__[x]) for x in columns])
|
|
||||||
)
|
|
||||||
logger.info("Created table in the database:"
|
|
||||||
" {}".format(model.tablename))
|
|
||||||
except:
|
|
||||||
#If we can't create a table here, it means it's already in the
|
|
||||||
#database, so we can skip it
|
|
||||||
pass
|
|
Loading…
Add table
Reference in a new issue