app.py

"""
Server for ethiclo
"""

import os
from flask import Flask, request, jsonify, render_template
import psycopg2 as pg
from flask_cors import CORS
# from dotenv import load_dotenv
from helpers import handle_url
from scraper import sustainability_search
from models.scoring.scoring_classes import predict_sustainability

# load_dotenv()

# Initialize flask app
app = Flask(__name__)
CORS(app)

# Base route for testing
@app.route('/')
def healthcheck():
    """Healthcheck entrypoint.
    """
    return jsonify({'status': 200})


@app.route('/add_shopper/<string:email>', methods=["POST"])
def add_shopper(email: str):
    """Add a shopper to the db.
    """
    # Connect to the db
    conn = connect()
    cur = conn.cursor()
    print("Connected to db")

    # Insert the user into the db
    try:
        insert = "INSERT INTO Shopper (email) VALUES (%s)"
        cur.execute(insert, [email])
        conn.commit()
        cur.close()
        conn.close()
        return jsonify({'status': 200})
    except pg.Error:
        cur.close()
        conn.close()
        return jsonify({'status': 400, 'error': pg.Error})
    
@app.route('/get_my_products/<string:email>', methods=["POST"])
def get_my_products(email: str):
    # Connect to the db
    conn = connect()
    cur = conn.cursor()
    print("Connected to db")

    # Insert the user into the db
    try:
        products = "SELECT * FROM Product WHERE shopper = %s AND alt_to = 0"
        cur.execute(products, [email])
        resp = cur.fetchall()
        cur.close()
        conn.close()
        # print(jsonify(resp))

        prod = []
        for product in resp:
            new = {}
            new['id'] = product[0]
            new['url'] = product[1]
            new['img_src'] = product[2]
            new['title'] = product[3]
            new['price'] = product[4]
            new['brand'] = product[5]
            new['description'] = product[6]
            new['score'] = product[7]
            prod.append(new)
        print(prod)
        cur.close()
        conn.close()
        return jsonify(prod)
    except pg.Error:
        cur.close()
        conn.close()
        return jsonify({'status': 400, 'error': pg.Error})



@app.route('/add_url', methods=["POST"])
def add_url():
    """Send a URL to the db.
    """
    # Get the data from the request
    data = request.get_json()
    url = data['url']
    email = data['email']
    # Connect to the db
    conn = connect()
    cur = conn.cursor()

    # Insert the url into the db
    try:
        insert = "INSERT INTO Website (url, shopper) VALUES (%s, %s);"
        cur.execute(insert, [url, email])
        conn.commit()

        # TODO: Call helper function
        data = handle_url(url)

        # insert the product into the db
        insert_product = """
            INSERT INTO Product (url, img_src, title, price, brand, description, score, alt_to, shopper)
            VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s);
        """
        cur.execute(insert_product, [data['url'], data['image'], data['title'], 
                                     float(data['price'][1:]), data['brand'], 
                                     'description', data['score'], 0, email])
        
        conn.commit()

        # Get the product id
        get_product_id = """
            SELECT product_id FROM Product
            WHERE url = %s AND shopper = %s;
        """
        cur.execute(get_product_id, [data['url'], email])
        product_id = cur.fetchone()[0]
    
        # make the query
        query = f"{data['classification']} {data['title'].lower().replace(data['brand'].lower(), '')}"
        print(f"This is the query: {query}\n\n\n")
        sustainable_items = sustainability_search(query)

        for item in sustainable_items:
            # TODO: score the item
            text = [f"{sustainable_items[item]['brand']} {sustainable_items[item]['title']}"]
            predictions = predict_sustainability(text)
            score = predictions.tolist()
            cur.execute(insert_product, [sustainable_items[item]['url'], sustainable_items[item]['img'], sustainable_items[item]['title'], 
                                         float(sustainable_items[item]['price'][1:]), sustainable_items[item]['brand'], 
                                         sustainable_items[item]['description'], score, product_id, email])
            conn.commit()

        cur.close()
        conn.close()
        return jsonify({'status': 200})
    except pg.Error:
        cur.close()
        conn.close()
        return jsonify({'status': 400, 'error': pg.Error})


@app.route('/get_sustainable_products', methods=["POST"])
def get_sustainable_products():
    """Fetch sustainable alternatives from the database
    """
    data = request.get_json()
    url = data['url']
    email = data['email']

    # Get the sustainable alternatives from the db
    get_sustainable_products = """
        SELECT * FROM Product
        WHERE alt_to IN (
            SELECT product_id FROM Product
            WHERE url = %s AND shopper = %s
        );
    """
    conn = connect()
    cur = conn.cursor()

    try:
        cur.execute(get_sustainable_products, [url, email])
        resp = cur.fetchall()
        prod = []
        for product in resp:
            new = {}
            new['id'] = product[0]
            new['url'] = product[1]
            new['img_src'] = product[2]
            new['title'] = product[3]
            new['price'] = product[4]
            new['brand'] = product[5]
            new['description'] = product[6]
            new['score'] = product[7]
            prod.append(new)
        print(prod)
        cur.close()
        conn.close()
        total = len(prod)
        return jsonify(prod[total - int(total /2):])
    except pg.Error:
        cur.close()
        conn.close()
        return jsonify({'status': 400, 'error': pg.Error})


def connect():
    """Connect to the db.
    """
    try:
        connection = pg.connect(
            dbname=os.environ["PGDATABASE"], user=os.environ["PGUSER"],
            password=os.environ["PASSWORD"],
            options="-c search_path=ethiclo"
        )
        return connection
    except pg.Error:
        return None


if __name__ == '__main__':
    app.run(debug=True, port=os.getenv("PORT", default=5000))