read-ti-later/app.py

from utils import clean_articles
import ldap as l
from ldap3 import Server, Connection, ALL, MODIFY_REPLACE
from flask import Flask, g, request, session, redirect, url_for, render_template
from flask_simpleldap import LDAP
from flask_bootstrap import Bootstrap
from readability import Document
from readabilipy import simple_json_from_html_string
import os
import sqlite3
import requests
from requests.api import head
from utils import clean_articles

app = Flask(__name__)
Bootstrap(app)
app.secret_key = 'asdf'
app.debug = True

# Base
app.config['LDAP_REALM_NAME'] = 'OpenLDAP Authentication'
app.config['LDAP_HOST'] = os.environ.get('LDAP_HOST')
app.config['LDAP_BASE_DN'] = os.environ.get('LDAP_BASE_DN')
app.config['LDAP_USERNAME'] = os.environ.get('LDAP_USERNAME')
app.config['LDAP_PASSWORD'] = os.environ.get('LDAP_PASSWORD')

# OpenLDAP 
app.config['LDAP_OBJECTS_DN'] = 'dn'
app.config['LDAP_OPENLDAP'] = True
app.config['LDAP_USER_OBJECT_FILTER'] = '(&(objectclass=posixAccount)(uid=%s))'

short_domain = os.environ.get('SHORT_DOMAIN')

ldap = LDAP(app)

server = Server(app.config['LDAP_HOST'])
conn = Connection(server, app.config['LDAP_USERNAME'], app.config['LDAP_PASSWORD'], auto_bind=True)

@app.before_request
def before_request():
    g.user = None
    if 'user_id' in session:
        # This is where you'd query your database to get the user info.
        g.user = {}


@app.route('/')
@ldap.login_required
def index():
    user_dict = ldap.get_object_details(session['user_id'])

    if 'user_id' in session:
        user = {'dn': 'cn={},cn=usergroup,ou=users,dc=technicalincompetence,dc=club'.format(user_dict['cn'][0].decode('ascii')),
                'firstName': user_dict['givenName'][0].decode('ascii'),
                'lastName': user_dict['sn'][0].decode('ascii'),
                'email': user_dict['mail'][0].decode('ascii'),
                'userName': user_dict['uid'][0].decode('ascii'),
        }

    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    c.execute("SELECT article_id, url, title, byline FROM saved_articles INNER JOIN articles on saved_articles.article_id = articles.id WHERE user=? AND read=0 OR read IS NULL", (session['user_id'], ))
    rows = c.fetchall()

    conn.commit()
    conn.close()

    return render_template('list.j2', articles = clean_articles(rows))


@app.route('/archived')
@ldap.login_required
def archived():
    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    c.execute("SELECT article_id, url, title, byline FROM saved_articles INNER JOIN articles on saved_articles.article_id = articles.id WHERE user=? AND read=1", (session['user_id'], ))
    rows = c.fetchall()
    print(rows)

    conn.commit()
    conn.close()

    return render_template('list.j2', articles = clean_articles(rows))


@app.route('/save')
@ldap.login_required
def save():
    return render_template('save.j2')


@app.route('/login', methods=['GET', 'POST'])
def login():
    if g.user:
        return redirect(url_for('index'))
    if request.method == 'POST':
        user = request.form['user']
        passwd = request.form['passwd']
        test = ldap.bind_user(user, passwd)
        if test is None or passwd == '':
            return render_template('login.j2', error='Invalid credentials')
        else:
            session['user_id'] = request.form['user']
            session['passwd'] = request.form['passwd']

            if session['next']:
                next = session['next']
                session['next'] = ''
                return redirect(next)
            return redirect('/')
    return render_template('login.j2')


@ldap.login_required
@app.route('/article/<int:article_id>')
def read_article(article_id):
    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    c.execute("SELECT * FROM articles where id=?", (article_id,))
    rows = c.fetchall()
    conn.commit()
    conn.close()
    
    if (len(rows) > 0):
        return render_template('article.j2', article=rows[0])

    return render_template('article.j2', article=())


@ldap.login_required
@app.route('/add', methods=['GET', 'POST'])
def add_url():
    if not 'user_id' in session:
        session['next'] = request.url
        return redirect(url_for('login'))
    if request.method == 'POST':
        url = request.form['url']
        close = None
    else:
        url = request.args.get('url')
        close = request.args.get('close')
    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    if url is not None and len(url) > 0:
        headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 Safari/537.36'}
        response = requests.get(url, headers=headers)

        article = simple_json_from_html_string(response.text, use_readability=True)

        c.execute("SELECT * FROM articles WHERE url=?", (url,))
        rows = c.fetchall()
        
        if (len(rows) == 0):
            c.execute("INSERT INTO articles (url, content, title, byline) VALUES (?, ?, ?, ?)", (url, article['content'], article['title'], article['byline']))
            c.execute("SELECT * FROM articles WHERE url=?", (url,))
            rows = c.fetchall()

        article_id = rows[0][0]

        c.execute("SELECT * FROM saved_articles WHERE user=? AND article_id=?", (session['user_id'], article_id))
        rows = c.fetchall()

        if (len(rows) == 0):
            c.execute("INSERT INTO saved_articles (user, article_id) VALUES (?, ?)", (session['user_id'], article_id))
        conn.commit()
        conn.close()

        if close is not None and close == '1':
            return render_template('close.j2')
        return 'Saved'
    conn.commit()
    conn.close()
    return 'Error'

@ldap.login_required
@app.route('/delete/<int:article_id>')
def delete_article(article_id):
    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    c.execute("DELETE FROM saved_articles WHERE user=? AND article_id=?", (session['user_id'], article_id))
    c.execute("SELECT * FROM saved_articles WHERE article_id=?", (article_id, ))
    rows = c.fetchall()
    
    if (len(rows) == 0):
        c.execute("DELETE FROM articles WHERE id=?", (article_id,))

    conn.commit()
    conn.close()

    return redirect(url_for('index'))

@ldap.login_required
@app.route('/archive/<int:article_id>')
def archive_article(article_id):
    conn = sqlite3.connect('pocket/readitlater.db')
    c = conn.cursor()

    c.execute("UPDATE saved_articles SET read=1 WHERE user=? AND article_id=?", (session['user_id'], article_id))

    conn.commit()
    conn.close()

    return redirect(url_for('index'))


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


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