import os

import requests

import json

from flask import Flask, render_template, session, redirect, request, url_for, flash

from flask_script import Manager, Shell

from flask_wtf import FlaskForm

from wtforms import StringField, SubmitField, FileField, PasswordField, BooleanField, SelectMultipleField, ValidationError

from wtforms.validators import Required, Length, Email, Regexp, EqualTo

from flask_sqlalchemy import SQLAlchemy

import random

from flask_migrate import Migrate, MigrateCommand

from flask_mail import Mail, Message

from threading import Thread

from werkzeug import secure_filename

from werkzeug.security import generate_password_hash, check_password_hash

# Imports for login management

from flask_login import LoginManager, login_required, logout_user, login_user, UserMixin, current_user

# Configure base directory of app

basedir = os.path.abspath(os.path.dirname(__file__))

app = Flask(__name__)

app.static_folder = 'static'

app.config['SECRET_KEY'] = 'hardtoguessstring'

app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get('DATABASE_URL') or "postgresql://localhost/articles"

app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

app.config['HEROKU_ON'] = os.environ.get('HEROKU')

# Email configuration

app.config['MAIL_SERVER'] = 'smtp.googlemail.com'

app.config['MAIL_PORT'] = 587 #default

app.config['MAIL_USE_TLS'] = True

app.config['MAIL_USERNAME'] = os.environ.get('MAIL_USERNAME')

app.config['MAIL_PASSWORD'] = os.environ.get('MAIL_PASSWORD')

app.config['MAIL_SUBJECT_PREFIX'] = '[NYTimes Articles App]'

app.config['MAIL_SENDER'] = 'Admin <ameliagc364@gmail.com>'

app.config['ADMIN'] = os.environ.get('ADMIN') or "ameliagc364@gmail.com"

app.config['HEROKU_ON'] = os.environ.get('HEROKU')

# Set up Flask debug and necessary additions to app

manager = Manager(app)

db = SQLAlchemy(app) # For database use

migrate = Migrate(app, db) # For database use/updating

manager.add_command('db', MigrateCommand) # Add migrate command to manager

mail = Mail(app) # For email sending

# Login configurations setup

login_manager = LoginManager()

login_manager.session_protection = 'strong'

login_manager.login_view = 'login'

login_manager.init_app(app)

# Shell function

def make_shell_context():

return dict( app=app, db=db, User=User, FavoriteArticles=FavoriteArticles, Article=Article, Search=Search)

manager.add_command("shell", Shell(make_context=make_shell_context))

# Send email functions

def send_async_email(app, msg):

with app.app_context():

mail.send(msg)

def send_email(to, subject, template, **kwargs):

return thr

# Set up for tables

# Association table between search terms and articles

titles = db.Table('titles',db.Column('search_id',db.Integer, db.ForeignKey('search.id')),db.Column('articles_id',db.Integer, db.ForeignKey('articles.id')))

# Assosciation table between articles and user's favorites collection

user_collection = db.Table('user_collection',db.Column('user_id', db.Integer, db.ForeignKey('articles.id')),db.Column('collection_id',db.Integer, db.ForeignKey('favoriteArticles.id')))

class User(UserMixin, db.Model):

return True

class FavoriteArticles(db.Model):

articles = db.relationship('Article', secondary=user_collection,backref=db.backref('favoriteArticles',lazy='dynamic'),lazy='dynamic')

class Article(db.Model):

return "{} : {}".format(self.headline,self.url)

class Search(db.Model):

return "{} : {}".format(self.id, self.term)

@login_manager.user_loader

def load_user(user_id):

return User.query.get(int(user_id))

# Form to register new user

class RegistrationForm(FlaskForm):

raise ValidationError('Username already taken')

# Login form

class LoginForm(FlaskForm):

submit = SubmitField('Log In')

# Form to search archive for an article

class ArticleSearchForm(FlaskForm):

search = StringField("Enter a year and month in the following format: YYYY M (ex. 2017 2 for February 2017).", validators=[Required()])

submit = SubmitField('Submit')

# Form to choose favorite articles

class SaveFavoriteForm(FlaskForm):

name = StringField('Collection Name',validators=[Required()])

favorite_articles = SelectMultipleField('Articles to save')

submit = SubmitField("Save")

# Get or create functions

def get_article_by_id(id):

a = Article.query.filter_by(id=id).first()

return a

def get_or_create_search_term(db_session, term, article_list = []):

return searchTerm

def get_or_create_article(db_session, headline, url):

return article

def get_or_create_personal_collection(db_session, name, article_list, current_user):

return savedArticles

# Error handling routes

@app.errorhandler(404)

def page_not_found(e):

return render_template('404.html'), 404

@app.errorhandler(500)

def internal_server_error(e):

return render_template('500.html'), 500

## Login routes

@app.route('/login',methods=["GET","POST"])

def login():

return render_template('login.html',form=form)

@app.route('/logout')

@login_required

def logout():

logout_user()

flash('You have been logged out')

return redirect(url_for('index'))

@app.route('/register',methods=["GET","POST"])

def register():

return render_template('register.html',form=form)

# View routes

@app.route('/', methods=['GET', 'POST'])

def index():

return render_template('index.html', form=form)

@app.route('/all_articles')

def see_all():

return render_template('all_articles.html', all_articles=all_articles)

@app.route('/create_article_collection',methods=["GET","POST"])

@login_required

def save_favorites():

return render_template('save_articles.html',form=form)

@app.route('/see_collection', methods=["GET", "POST"])

@login_required

def seeFavArticles():

return render_template('view_favorites.html', fav_articles=fav_articles)

if __name__ == '__main__':

db.create_all()

manager.run()