postgres support added

Author: shravanc

app/__init__.py

@@ -1,7 +1,10 @@
 from flask import Flask
-
+import os
 from flask_pymongo import PyMongo
 from celery import Celery
+from flask_sqlalchemy import SQLAlchemy
+
+
 
 mongo = PyMongo()
 celery = Celery('flask_api')
@@ -22,9 +25,19 @@ def initialize_extensions(application):
     mongo.init_app(application)
     from app.models.user import User
 
+    from app.models.rating import Rating
+
+    db = SQLAlchemy(application)
+
+
 
 def register_blueprints(application):
     from app.controllers import user_blueprints
     application.register_blueprint(user_blueprints)
 
 
+
+config_filename = os.path.abspath(os.path.dirname(__file__)) + "/../instance/development.cfg"
+app = Flask(__name__)
+app.config.from_pyfile(config_filename)
+db = SQLAlchemy(app)

app/controllers/__init__.py

@@ -2,6 +2,7 @@
 from flask import Blueprint, current_app
 
 from app.controllers.users_controller import index as users_index
+from app.controllers.users_controller import ratings as users_ratings
 from app.controllers.users_controller import home as users_home
 
 
@@ -10,6 +11,7 @@
 
 user_blueprints = Blueprint('users', 'api', template_folder=template_dir)
 user_blueprints.add_url_rule('/users', view_func=users_index, methods=['GET'])
+user_blueprints.add_url_rule('/ratings', view_func=users_ratings, methods=['GET'])
 user_blueprints.add_url_rule('/home', view_func=users_home, methods=['GET'])
 
 

app/controllers/users_controller.py

@@ -2,6 +2,8 @@
 
 #from . import user_blueprints
 from app.models.user import User
+from app.models.rating import Rating
+
 from app import mongo
 from app.jobs.user_job import new_task
 import app.helpers.user_service as us
@@ -13,6 +15,19 @@ def index():
     us.say_hello('hello')
     return jsonify({'data': users})
 
+
+def ratings():
+  user_ratings = Rating.query.all()
+  ratings = []
+  for rating in user_ratings:
+    d1 = {}
+    d1["user_id"] = rating.user_id
+    d1["item_id"] = rating.item_id
+    d1["rating"]  = rating.rating
+    ratings.append(d1)
+  return jsonify({'data': ratings})
+
+
 #@user_blueprints.route('/home')
 def home():
     return render_template('index.html')

app/models/rating.py

@@ -0,0 +1,19 @@
+from app import db
+from sqlalchemy.dialects.postgresql import JSON
+
+
+class Rating(db.Model):
+  __tablename__ = 'ratings'
+
+
+  id = db.Column(db.Integer, primary_key=True)
+  user_id = db.Column(db.Integer)
+  item_id = db.Column(db.Integer)
+  rating  = db.Column(db.Integer)
+
+
+  def __repr__(self):
+    return '<id {}>'.format(self.id)
+
+
+

instance/development.cfg

@@ -4,4 +4,18 @@ import os
 
 basedir = os.path.abspath(os.path.dirname(__file__))
 
+# MongoDB
 MONGO_URI = "mongodb://127.0.0.1:27017/codeitup_team10"
+
+
+
+# PostgresSQL
+POSTGRES_USER = "postgres"
+POSTGRES_PW   = "welcome"
+POSTGRES_URL  = "127.0.0.1:5432"
+POSTGRES_DB   = "user_system_test"
+
+DB_URL = 'postgresql+psycopg2://{user}:{pw}@{url}/{db}'.format(user=POSTGRES_USER,pw=POSTGRES_PW,url=POSTGRES_URL,db=POSTGRES_DB)
+
+SQLALCHEMY_TRACK_MODIFICATIONS = False
+SQLALCHEMY_DATABASE_URI = DB_URL