index.py

from flask import Flask, render_template, request, redirect, url_for, jsonify, session
from authlib.integrations.flask_client import OAuth
from loginpass import create_flask_blueprint, Discord, Google
from databases import Database
import os
from dotenv import load_dotenv

if os.path.exists('.env'):
    load_dotenv()

app = Flask(__name__)
oauth = OAuth(app)

config = {
    "SECRET_KEY": os.getenv('SECRET_KEY'),
    "DISCORD_CLIENT_ID": os.getenv('DISCORD_CLIENT_ID'),
    "DISCORD_CLIENT_SECRET": os.getenv('DISCORD_CLIENT_SECRET'),
    "GOOGLE_CLIENT_ID": os.getenv('GOOGLE_CLIENT_ID'),
    "GOOGLE_CLIENT_SECRET": os.getenv('GOOGLE_CLIENT_SECRET'),
    "MONGO_URI": os.getenv('MONGO_URI'),
}

app.config.from_mapping(config)
backends = [Discord, Google]
database = Database(app.config['MONGO_URI'])

@app.route('/')
def home():
    if 'user' in session:
        user = session['user']
        books = database.getUser(user['email'])['books']
        books_ = []
        for book in books:
            books_.append(database.getBook(book))
        return render_template('user.html', books=books_, user=user)
    return render_template('index.html')

@app.route('/logout')
def logout():
    session.pop('user', None)
    return redirect(url_for('home'))

@app.route('/changename', methods=['POST'])
def changename():
    if 'user' in session:
        name = request.form['name']
        database.updateName(session['user']['email'], name)
        email = session['user']['email']
        session.pop('user', None)
        session['user'] = database.getUser(email)
        return redirect(url_for('home'))
    return redirect(url_for('home'))

@app.route('/newBook', methods=['POST'])
def newBook():
    if 'user' in session:
        name = request.form['name']
        if name.replace(' ', '') == '':
            return redirect(url_for('home'))
        database.createBook(session['user']['_id'], name)
        return redirect(url_for('home'))
    return redirect(url_for('home'))

@app.route('/book/<bookID>')
def book(bookID):
    if 'user' in session:
        book = database.getBook(bookID)
        chapters = book['chapterOrder']
        chapters_ = []
        for chapter in chapters:
            chapters_.append(database.getBook(bookID)['chapters'][chapter])
        return render_template('book.html', book=book, chapters=chapters_, user=session['user'])
    return redirect(url_for('home'))

@app.route('/newchapter/<bookid>', methods=['POST'])
def newChapter(bookid):
    if 'user' in session:
        name = request.form['name']
        if name.replace(' ', '') == '':
            return redirect(url_for('book', bookID=bookid))
        database.createChapter(bookid, name)
        return redirect(url_for('book', bookID=bookid))
    return redirect(url_for('home'))

@app.route('/book/<bookID>/<chapterID>')
def chapter(bookID, chapterID):
    if 'user' in session:
        book = database.getBook(bookID)
        chapter = book['chapters'][chapterID]
        snippets = chapter['snippetOrder']
        snippets_ = []
        for snippet in snippets:
            snippets_.append(database.getSnippet(bookID, chapterID, snippet))
        return render_template('chapter.html', book=book, chapter=chapter, snippets=snippets_, user=session['user'])

@app.route('/newSnippet/<bookid>/<chapterid>', methods=['POST'])
def newSnippet(bookid, chapterid):
    if 'user' in session:
        name = request.form['name']
        if name.replace(' ', '') == '':
            return redirect(url_for('book', bookID=bookid))
        database.createSnippet(bookid, chapterid, name)
        return redirect(url_for('chapter', bookID=bookid, chapterID=chapterid))
    return redirect(url_for('home'))

@app.route('/book/<bookID>/<chapterID>/<snippetID>')
def snippet(bookID, chapterID, snippetID):
    if 'user' in session:
        snippet = database.getSnippet(bookID, chapterID, snippetID)
        return render_template('snippet.html', snippet=snippet, user=session['user'], bookID=bookID, chapterID=chapterID)

@app.route('/share/<bookID>/<chapterID>/<snippetID>', methods=['POST'])
def share(bookID, chapterID, snippetID):
    if 'user' in session:
        time = request.form['time']
        shareID = database.share(session['user']['_id'], bookID, chapterID, snippetID, time)
        return redirect(url_for('snip', shareID=shareID))
    return redirect(url_for('home'))

@app.route('/snip/<shareID>')
def snip(shareID):
    share = database.getShared(shareID)
    if share:
        return render_template('shared.html', snippet=share)
    return redirect(url_for('home'))

@app.route('/delete/<item>', methods=['POST'])
def delete(item):
    if 'user' in session:
        if item == 'book':
            bookID = request.args.get('bookID')
            database.deleteBook(bookID)
            return redirect(url_for('home'))
        elif item == 'chapter':
            bookID = request.args.get('bookID')
            chapterID = request.args.get('chapterID')
            database.deleteChapter(bookID, chapterID)
            return redirect(url_for('book', bookID=bookID))
        elif item == 'snippet':
            bookID = request.args.get('bookID')
            chapterID = request.args.get('chapterID')
            snippetID = request.args.get('snippetID')
            database.deleteSnippet(bookID, chapterID, snippetID)
            return redirect(url_for('chapter', bookID=bookID, chapterID=chapterID))
    return redirect(url_for('home'))

@app.route('/updatesnip', methods=['POST'])
def updatesnip():
    if 'user' in session:
        bookID = request.args.get('bookID')
        chapterID = request.args.get('chapterID')
        snippetID = request.args.get('snippetID')
        content = request.form['content']
        database.updateSnippetContent(bookID, chapterID, snippetID, content)
        return redirect(url_for('chapter', bookID=bookID, chapterID=chapterID))
    return redirect(url_for('home'))

@app.route('/edit/<item>', methods=['POST'])
def edit(item):
    if 'user' in session:
        name = request.args.get('name')
        print(name)
        if name == None or name == "null" or name.replace(' ', '') == '':
            return redirect(url_for('home'))
        if item == 'book':
            bookID = request.args.get('bookID')
            database.updateBook(bookID, name)
            return redirect(url_for('book', bookID=bookID))
        elif item == 'chapter':
            bookID = request.args.get('bookID')
            chapterID = request.args.get('chapterID')
            database.updateChapter(bookID, chapterID, name)
            return redirect(url_for('chapter', bookID=bookID, chapterID=chapterID))
        elif item == 'snippet':
            bookID = request.args.get('bookID')
            chapterID = request.args.get('chapterID')
            snippetID = request.args.get('snippetID')
            database.updateSnippet(bookID, chapterID, snippetID, name)
            return redirect(url_for('snippet', bookID=bookID, chapterID=chapterID, snippetID=snippetID))
    return redirect(url_for('home'))

# @app.route("/snip/<id>")


def handle_authorize(remote, token, user_info):
    if database.userExists(user_info['email']):
        session['user'] = database.getUser(user_info['email'])
    else:
        database.addUser(user_info['email'])
        session['user'] = database.getUser(user_info['email'])
    return redirect(url_for('home'))


bp = create_flask_blueprint(backends, oauth, handle_authorize)
app.register_blueprint(bp, url_prefix='/')

if __name__ == '__main__':
    app.run(debug=True)