app.py

from flask import Flask, jsonify, request
import database
from objects import User, Plant
from flask_jwt import JWT, jwt_required, current_identity
from werkzeug.security import safe_str_cmp
import os
import uuid
from flask_cors import CORS, cross_origin
import datetime
import bcrypt
from google.cloud import storage
from apscheduler.schedulers.background import BackgroundScheduler

app = Flask(__name__)

db = database.model()

salt = b'$2b$12$dQvFTcjXMlf6uz4INHgtXu'

IMG_PATH = 'docs/images/user_plants'
if not os.path.exists(IMG_PATH):
    os.makedirs(IMG_PATH)

# initialize, add the sync job, and start the scheduler
scheduler = BackgroundScheduler()
sync_job = scheduler.add_job(db.increment_day, trigger="cron", hour=0)
scheduler.start()
db.increment_day()


def authenticate(username, password):
    user = db.get_user_by_username(username=username)
    if user and bcrypt.checkpw(password.encode('utf-8'), user.password.encode('utf-8')): 
        return user

def identity(payload):
    user_id = payload['identity']
    user = db.get_user_by_id(user_id=user_id)
    return user

app.config['SECRET_KEY'] = 'This is a secret key!'

app.config['JWT_EXPIRATION_DELTA'] = datetime.timedelta(hours=48)
jwt = JWT(app, authenticate, identity)

#set the CORS headrer to allow all access
CORS(app, supports_credentials=True)

@app.route('/new_user', methods=["POST"])
def new_user():
    credentials = request.get_json()
    hashword = bcrypt.hashpw(credentials['password'].encode('utf-8'), salt)
    if db.create_user(credentials["username"],hashword.decode('utf-8')):
        return jsonify("Success"), 201
    return jsonify("Couldn't create user"), 500

@app.route('/plants/img', methods=["POST"])
@jwt_required()
def add_img():
    img = request.files['plant_img']
    filename = str(uuid.uuid4()) + '.jpg'
    with open(os.path.join(IMG_PATH, filename), "wb") as fp:
        fp.write(img.read())
    return {"img_name":filename}, 200

@app.route('/plant', methods=["POST"])
@jwt_required()
def add_plant():
    values = request.values
    name = values['name']
    description = values['description']
    water_interval = values["water_interval"]
    img = request.files['plant_img'].read()

    filename = str(uuid.uuid4()) + '.jpg'
    img_url = 'https://storage.googleapis.com/webdev-final-279420.appspot.com/User_img/' + filename
    
    #upload img to storage 
    storage_client = storage.Client()
    bucket = storage_client.bucket('webdev-final-279420.appspot.com')
    blob = bucket.blob('User_img/'+filename)
    blob.upload_from_string(img, content_type='image/jpeg')

    new_plant = Plant(id=None,user_id=current_identity.id, name=name, img_path=str(img_url), water_interval=water_interval, days_until_water=water_interval, notes=description)
    try:
        db.add_plant(current_identity, new_plant)
    except:
        return jsonify("couldn't add plant"), 500
    return jsonify("added the plant"), 200

@app.route('/plants', methods=["GET"])
@jwt_required()
def get_plants():
    plants = db.get_plants(current_identity)
    all_plants = []
    for plant in plants:
        all_plants.append(plant.__dict__)
    return jsonify(all_plants), 200

@app.route('/feedback', methods=["POST"])
def submit_feedback():
    values = request.json
    name = values['name']
    email = values['email']
    message = values["message"]
    try:
        db.add_feedback(name,email,message)
    except:
        return jsonify("there was an issue saving your feedback"), 500
    return jsonify("success"), 200

@app.route('/dashboard', methods=["GET"])
@jwt_required()
def dashboard():
    user = current_identity
    return jsonify(user.username)


if __name__ == "__main__":
    app.run(host='0.0.0.0', debug=True)