Partially migrated to SQLAlchemy

requirements.txt

@@ -1,3 +1,4 @@
 flask
 mock
 psycopg2
+flask-sqlalchemy

smash/__init__.py

@@ -1,12 +1,16 @@
 import os
 from flask import Flask
-from . import config, database_postgresql, log, models
+from flask.ext.sqlalchemy import SQLAlchemy
+from . import config, database_sqlalchemy, log, models
 
 
 log.configure_logging()
 app = Flask(__name__)
 conf = config.Config('config.json')
 
+# Load database URL for SQLAlchemy from environment
+app.config['SQLALCHEMY_DATABASE_URI'] = os.environ['DATABASE_URL']
+
 # This flag tells the program it's deployed on heroku
 if 'HEROKU' in os.environ:
     conf.add(('HEROKU', 1))
@@ -36,8 +40,13 @@ if 'SECRETKEY' in conf.config:
 else:
     exit("Secret key not set.")
 
-db = database_postgresql.DatabasePostgreSQL()
-models.init_models(db)
+db = SQLAlchemy(app)
+
+from smash.models_sqlalchemy import *
+
+db.create_all()
+
+#models.init_models(db)
 
 
 from . import views

smash/models_sqlalchemy.py

@@ -0,0 +1,38 @@
+from smash import db
+
+
+class Quote(db.Model):
+    __tablename__ = 'quotes'
+
+    id = db.Column(db.Integer, primary_key=True)
+    rating = db.Column(db.Integer)
+    content = db.Column(db.String())
+    approved = db.Column(db.Boolean)
+    author_ip = db.Column(db.String())
+    time = db.Column(db.String())
+
+
+    def __init__(self, content, author_ip, time):
+        self.rating = 0
+        self.content = content
+        self.approved = False
+        self.author_ip = author_ip
+        self.time = time
+
+
+class Tag(db.Model):
+    __tablename__ = 'tags'
+
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String())
+
+
+    def __init(self, name):
+        self.name = name
+
+
+tags_to_quotes = db.Table(
+    'tagsToQuotes',
+    db.Column('tagid', db.Integer, db.ForeignKey('tags.id')),
+    db.Column('quoteid', db.Integer, db.ForeignKey('quotes.id'))
+)