tracker.py

import datetime
from model import db, User, Image
import requests
import os
from passlib.hash import bcrypt
from PIL import Image as pilImage
from flask import flash, session

def check_login(user, password):
    """ check if a users login credetials are correct """

    # query for any users with that username
    active_user = db.session.query(User.password, User).filter(User.username == user,
                                    ).first()

    # if there is a matching user
    if active_user and bcrypt.verify(password, active_user[0]):
        flash( "Login Successful")
        session['user'] = active_user[1].user_id
        return '/user'
    # if there is not a user with that username and password combo
    else:
        flash("Login Failed")
        return '/login'


def time_difference_now(time):  # pragma: no cover
    """ Find how much time has passed since a datetime in days"""

    NOW = datetime.datetime.now()

    diff = NOW - time

    # day_seconds = diff.seconds/(24.00 * 60 * 60.00)

    # days = diff.days + day_seconds

    return int(diff.days)


def sort_projects_by_update(projects, time):
    """ sort a list of project objects into 2 lists based on update """

    need_update = []
    update = []

    for project in projects:
        since_update = time_difference_now(project.updated_at)
        if since_update > time:
            need_update.append((project, since_update))
        else:
            update.append((project, since_update))

    return need_update, update


def check_username(user):
    """ check if a user is in the database """

    # active_user = User.query.filter_by(username = user).first()

    # if active_user:
    #     flash( "Login Successful")
    #     session['user'] = active_user.user_id
    # else:
    #     flash("Login Failed")


def post_project_db_update(project, notes, status, image, user, progress):
    """ update a project in the db and ravelry site """

    project.notes = notes
    project.progress = progress
    if status:
        project.status_id = int(status)
    project.updated_at = datetime.datetime.now()
    if image:
        photo = Image(url=image, project_id=project.project_id)
        db.session.add(photo)
    db.session.commit()