adding boilerplate flask code

- adding sqlalchemy
- adding postgres support
- adding marshmallow
- adding flask forms and wtf

Author: Athithyaa

.gitignore

@@ -94,3 +94,5 @@ ENV/
 .idea/
 T9WebApp.iml
 ubuntu-aws.pem
+apvirt/
+database/config.properties

application.py

@@ -1,12 +1,71 @@
-from flask import Flask
+from flask import Flask, render_template, request
+from database.db import db_session, init_db
+import logging
+from logging import Formatter, FileHandler
+from forms import *
+import os
 
 app = Flask(__name__)
+app.secret_key = 's3cr3t'
 
 
-@app.route("/")
-def hello_world():
-    return 'Hello World'
+@app.teardown_request
+def shutdown_session(exception=None):
+    db_session.remove()
+
+
+@app.route('/')
+def home():
+    return render_template('pages/placeholder.home.html')
+
+
+@app.route('/about')
+def about():
+    return render_template('pages/placeholder.about.html')
+
+
+@app.route('/login')
+def login():
+    form = LoginForm(request.form)
+    return render_template('forms/login.html', form=form)
+
+
+@app.route('/register')
+def register():
+    form = RegisterForm(request.form)
+    return render_template('forms/register.html', form=form)
+
+
+@app.route('/forgot')
+def forgot():
+    form = ForgotForm(request.form)
+    return render_template('forms/forgot.html', form=form)
+
+# Error handlers.
+
+
+@app.errorhandler(500)
+def internal_error(error):
+    #db_session.rollback()
+    return render_template('errors/500.html'), 500
+
+
+@app.errorhandler(404)
+def not_found_error(error):
+    return render_template('errors/404.html'), 404
+
+if not app.debug:
+    file_handler = FileHandler('error.log')
+    file_handler.setFormatter(
+        Formatter('%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]')
+    )
+    app.logger.setLevel(logging.INFO)
+    file_handler.setLevel(logging.INFO)
+    app.logger.addHandler(file_handler)
+    app.logger.info('errors')
+
 
 if __name__ == "__main__":
+    init_db()
     app.debug = True
     app.run()

database/__init__.py

@@ -0,0 +1,28 @@
+from sqlalchemy import create_engine
+
+
+# fill out the path to sqlite engine
+def get_username(file):
+    f = open(file)
+    uname = ""
+    pword = ""
+    host = ""
+    for line in f:
+        split_str = line.split("=")
+        if split_str[0] == "username":
+            uname = split_str[1]
+        if split_str[0] == "password":
+            pword = split_str[1]
+        if split_str[0] == "host":
+            host = split_str[1]
+    return [uname, pword, host]
+
+
+file = "../database/config.properties"
+config = get_username(file)
+engine = create_engine('postgresql+psycopg2://' + config[0].replace("\n", "") + ':' + config[1].replace("\n", "") +
+                       '@' + config[2].replace("\n", "") + '/postgres')
+conn = engine.connect()
+conn.execute("commit")
+conn.execute("create database t9_db")
+conn.close()

database/create_db.py

@@ -0,0 +1,56 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import scoped_session, sessionmaker
+from sqlalchemy.ext.declarative import declarative_base
+import os
+from urlparse import urlparse
+
+
+def get_username(file):
+    f = open(file)
+    uname = ""
+    pword = ""
+    host = ""
+    for line in f:
+        split_str = line.split("=")
+        if split_str[0] == "username":
+            uname = split_str[1]
+        if split_str[0] == "password":
+            pword = split_str[1]
+        if split_str[0] == "host":
+            host = split_str[1]
+    return [uname, pword, host]
+
+
+url = None
+
+try:
+    url = urlparse(os.environ["DATABASE_URL"])
+
+except KeyError:
+    print("cannot find environment variable")
+
+engine = None
+
+if url:
+    engine = create_engine('postgresql+psycopg2://' + url.username + ':' + url.password +
+                           '@' + url.hostname + ':' + str(url.port) + '/' + url.path[1:])
+else:
+    file = "./database/config.properties"
+    config = get_username(file)
+    engine = create_engine('postgresql+psycopg2://' + config[0].replace("\n", "") + ':' + config[1].replace("\n", "") +
+                           '@' + config[2].replace("\n", "") + '/t9_db')
+
+db_session = scoped_session(sessionmaker(autocommit=False,
+                                         autoflush=False,
+                                         bind=engine))
+
+Base = declarative_base()
+Base.query = db_session.query_property()
+
+
+def init_db():
+    # import all modules here that might define models so that
+    # they will be registered properly on the metadata.  Otherwise
+    # you will have to import them first before calling init_db()
+    import models
+    Base.metadata.create_all(bind=engine)

database/db.py

@@ -0,0 +1,33 @@
+from flask_wtf import Form
+from wtforms import TextField, PasswordField
+from wtforms.validators import DataRequired, EqualTo, Length
+
+# Set your classes here.
+
+
+class RegisterForm(Form):
+    name = TextField(
+        'Username', validators=[DataRequired(), Length(min=6, max=25)]
+    )
+    email = TextField(
+        'Email', validators=[DataRequired(), Length(min=6, max=40)]
+    )
+    password = PasswordField(
+        'Password', validators=[DataRequired(), Length(min=6, max=40)]
+    )
+    confirm = PasswordField(
+        'Repeat Password',
+        [DataRequired(),
+        EqualTo('password', message='Passwords must match')]
+    )
+
+
+class LoginForm(Form):
+    name = TextField('Username', [DataRequired()])
+    password = PasswordField('Password', [DataRequired()])
+
+
+class ForgotForm(Form):
+    email = TextField(
+        'Email', validators=[DataRequired(), Length(min=6, max=40)]
+    )

forms.py

@@ -0,0 +1,8 @@
+from marshmallow_sqlalchemy import ModelSchema
+from models.Company import Company
+
+
+class CompanySchema(ModelSchema):
+    class Meta:
+        model = Company
+

ma_schema/CompanySchema.py

@@ -0,0 +1,8 @@
+from marshmallow_sqlalchemy import ModelSchema
+from models.Review import Review
+
+
+class ReviewSchema(ModelSchema):
+    class Meta:
+        model = Review
+

ma_schema/ReviewSchema.py

@@ -0,0 +1,8 @@
+from marshmallow_sqlalchemy import ModelSchema
+from models.User import User
+
+
+class UserSchema(ModelSchema):
+    class Meta:
+        model = User
+

ma_schema/UserSchema.py

@@ -0,0 +1,14 @@
+from database.db import Base
+from sqlalchemy import Column, String, Integer
+from sqlalchemy.orm import relationship
+
+
+class Company(Base):
+    __tablename__ = 'Companies'
+    id = Column(String, primary_key=True)
+    name = Column(String(100))
+    r_cnt = Column(Integer)
+    reviews = relationship("Company")
+
+    def __repr__(self):
+        return "<Company:" + self.id + " " + self.name + ">"

ma_schema/__init__.py

@@ -0,0 +1,17 @@
+from sqlalchemy import ForeignKey
+from database.db import Base
+from sqlalchemy import Column, String
+from sqlalchemy.orm import relationship
+
+
+class Review(Base):
+    __tablename__ = 'Reviews'
+    id = Column(String, primary_key=True)
+    user_id = Column(String(100), ForeignKey("User.id"), nullable=False)
+    company_id = Column(String(100), ForeignKey("Company.id"), nullable=False)
+    content = Column(String(100))
+    user = relationship("User", ForeignKey("User.id"))
+    company = relationship("Company", ForeignKey("Company.id"))
+
+    def __repr__(self):
+        return "<Review:" + self.id + " " + self.content + ">"

models/Company.py

@@ -0,0 +1,15 @@
+from database.db import Base
+from sqlalchemy import Column, String
+from sqlalchemy.orm import relationship
+
+
+class User(Base):
+    __tablename__ = 'Users'
+
+    id = Column(String, primary_key=True)
+    name = Column(String(120))
+    reviews = relationship("Review")
+
+    def __repr__(self):
+        return "<User:" + self.id + " " + self.name + ">"
+

models/Review.py

@@ -1,9 +1,19 @@
-appdirs==1.4.1
+Fabric==1.8.2
 Flask==0.10.1
-itsdangerous==0.24
-Jinja2==2.9.5
-MarkupSafe==0.23
-packaging==16.8
-pyparsing==2.1.10
-six==1.10.0
-Werkzeug==0.11.15
+Flask-SQLAlchemy==1.0
+Flask-WTF==0.9.4
+Jinja2==2.7.2
+MarkupSafe==0.18
+SQLAlchemy==0.9.3
+WTForms==1.0.5
+Werkzeug==0.9.4
+coverage==3.7.1
+ecdsa==0.10
+gunicorn==19.3.0
+itsdangerous==0.23
+paramiko==1.12.2
+pycrypto==2.6.1
+wsgiref==0.1.2
+psycopg2==2.6.2
+marshmallow-sqlalchemy
+flask-marshmallow