205 lines
7.7 KiB
Python
205 lines
7.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# all the imports
|
|
from __future__ import with_statement
|
|
from contextlib import closing
|
|
import sqlite3
|
|
from flask import Flask, request, session, g, redirect, url_for, abort, render_template, flash, jsonify
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
import re
|
|
from email.mime.text import MIMEText
|
|
from subprocess import Popen, PIPE
|
|
import smtplib
|
|
from note import Note, Page, convert_datestring_to_date
|
|
|
|
# configuration
|
|
DATABASE = 'notes.db'
|
|
DEBUG = True
|
|
SECRET_KEY = 'development key'
|
|
USERNAME = 'admin'
|
|
PASSWORD = 'dupa.8'
|
|
|
|
app = Flask(__name__)
|
|
app.config.from_object(__name__)
|
|
|
|
class Thumb:
|
|
def __init__(self, source, href, tags):
|
|
self.src = source
|
|
self.href = href
|
|
self.tags = tags
|
|
|
|
#DB functions
|
|
def connect_db():
|
|
return sqlite3.connect(app.config['DATABASE'])
|
|
|
|
def init_db():
|
|
with closing(connect_db()) as db:
|
|
with app.open_resource('schema.sql') as f:
|
|
db.cursor().executescript(f.read())
|
|
db.commit()
|
|
|
|
def query_db(query, args=(), one=False):
|
|
"""
|
|
Simple wrapper around query
|
|
"""
|
|
cur = g.db.execute(query, args)
|
|
rv = [dict((cur.description[idx][0], value)
|
|
for idx, value in enumerate(row)) for row in cur.fetchall()]
|
|
return (rv[0] if rv else None) if one else rv
|
|
|
|
@app.before_request
|
|
def before_request():
|
|
try:
|
|
g.db = g._db.cursor()
|
|
except AttributeError:
|
|
g._db = connect_db()
|
|
g.db = g._db.cursor()
|
|
|
|
@app.teardown_request
|
|
def teardown_request(exception):
|
|
try:
|
|
g.db.close()
|
|
except AttributeError:
|
|
pass
|
|
|
|
#VIEWS
|
|
@app.route('/')
|
|
@app.route('/view')
|
|
@app.route('/view/<subject>/<datestring>')
|
|
def home(subject= "mechanika", datestring= "14grudnia2012"):
|
|
other_dates = prepare_other_dates(subject)
|
|
all_dates = prepare_all_dates()
|
|
all_subjects = [sub["subject"] for sub in get_all_subjects()]
|
|
thumbs = get_thumbs(subject, datestring)
|
|
if datestring in other_dates:
|
|
other_dates.remove(datestring)
|
|
thumbs_info = {"subject": subject, "datestring": datestring, "other_dates": other_dates}
|
|
|
|
thumbs = [dict(href=thumb.href, src=thumb.src) for thumb in thumbs]
|
|
main_page_src=thumbs[0]["href"]
|
|
return render_template('index.html', thumbs=thumbs, main_page_src=main_page_src,thumbs_info=thumbs_info,
|
|
all_dates=all_dates, all_subjects=all_subjects)
|
|
|
|
|
|
@app.route('/subject/<subject>')
|
|
def show_subject(subject):
|
|
dates = prepare_other_dates(subject)
|
|
all_dates = prepare_all_dates()
|
|
all_subjects = [sub["subject"] for sub in get_all_subjects()]
|
|
return render_template('subject.html', subject=subject, dates=dates, all_dates=all_dates, all_subjects=all_subjects)
|
|
|
|
@app.route('/date/<datestring>')
|
|
def show_date(datestring):
|
|
all_dates = prepare_all_dates()
|
|
all_subjects = [sub["subject"] for sub in get_all_subjects()]
|
|
notes = get_notes_for_day(datestring)
|
|
separated_notes = {}
|
|
for note in notes:
|
|
if not note["subject"] in separated_notes:
|
|
separated_notes[note["subject"]] = []
|
|
separated_notes[note["subject"]].append(note)
|
|
|
|
return render_template('date.html', date=datestring, all_dates=all_dates, all_subjects=all_subjects, separated_notes=separated_notes)
|
|
|
|
@app.route('/search')
|
|
def show_search():
|
|
all_dates = prepare_all_dates()
|
|
all_subjects = [sub["subject"] for sub in get_all_subjects()]
|
|
return render_template('search.html', all_dates=all_dates, all_subjects=all_subjects )
|
|
|
|
@app.route('/about')
|
|
def about():
|
|
all_dates = prepare_all_dates()
|
|
all_subjects = [sub["subject"] for sub in get_all_subjects()]
|
|
return render_template('about.html', all_dates=all_dates, all_subjects=all_subjects )
|
|
|
|
#HELPERS
|
|
def add_note(subject, datestring, notes):
|
|
g.db.execute('insert into notes (subject, datestring ) values (?, ?)', [subject, datestring])
|
|
note_id = g.db.lastrowid
|
|
for note in notes:
|
|
g.db.execute('insert into pages (note_id, name, tags ) values (?, ?, ?)', [note_id, note["name"], note["tags"]])
|
|
g._db.commit()
|
|
|
|
def create_thumb(subject, datestring, name, tags, author_string):
|
|
src = "notes/" + author_string + "/mini/" + subject + "/" + datestring + "/" + name
|
|
href = "notes/" + author_string + "/" + subject + "/" + datestring + "/" + name
|
|
thumb = Thumb(src, href, tags)
|
|
return thumb
|
|
|
|
def get_thumbs(subject, datestring=""):
|
|
if(datestring==""):
|
|
query_result = query_db("select pages.* from pages left join notes on notes.id = pages.note_id where notes.subject = ?", [subject,])
|
|
else:
|
|
query_result = query_db("select pages.*, notes.author_string from pages left join notes on notes.id = pages.note_id \
|
|
where notes.subject = ? and datestring = ?", [subject, datestring])
|
|
return [create_thumb(subject, datestring, t["name"], t["tags"], t["author_string"]) for t in query_result ]
|
|
|
|
def get_possible_dates(subject):
|
|
dates = query_db("select distinct datestring from notes where subject = ? ", [subject])
|
|
return dates
|
|
|
|
def get_notes_for_day(datestring):
|
|
"""return pages which where written that day (daystring) with other info"""
|
|
query_result = query_db("select pages.*,notes.subject,notes.author_string, notes.author from pages left join notes on notes.id = pages.note_id \
|
|
where datestring = ?", [datestring])
|
|
return [{"subject": t["subject"], "author_string":t["author_string"], "author":t["author"], "datestring": datestring, "name": t["name"]}for t in query_result ]
|
|
|
|
def get_all_dates():
|
|
"""return all dates from database"""
|
|
dates = query_db("select distinct datestring from notes ")
|
|
return dates
|
|
|
|
def get_all_subjects():
|
|
"""return all subject from database"""
|
|
subjects = query_db("select distinct subject from notes ")
|
|
return subjects
|
|
|
|
def prepare_all_dates():
|
|
"""return all dates in well formated form of nested dictionaries"""
|
|
datestrings = [date["datestring"] for date in get_all_dates()]
|
|
all_dates = [convert_datestring_to_date(date) for date in datestrings]
|
|
return sort_dates(datestrings, all_dates)
|
|
|
|
def prepare_other_dates(subject):
|
|
datestrings = [date["datestring"] for date in get_possible_dates(subject)]
|
|
dates = [convert_datestring_to_date(date) for date in datestrings]
|
|
return sort_dates(datestrings, dates)
|
|
|
|
def sort_dates(datestrings, dates):
|
|
sorted_dates = {}
|
|
for date, datestring in zip(dates, datestrings):
|
|
if not date.datetime.year in sorted_dates:
|
|
sorted_dates[date.datetime.year] = {}
|
|
if not date.datetime.month in sorted_dates[date.datetime.year]:
|
|
sorted_dates[date.datetime.year][date.datetime.month] = []
|
|
sorted_dates[date.datetime.year][date.datetime.month].append({"day" : date.datetime.day, "source": datestring,
|
|
"weekday": weekday(date.datetime.weekday())})
|
|
sorted = sort_number_strings(sorted_dates[date.datetime.year][date.datetime.month])
|
|
sorted_dates[date.datetime.year][date.datetime.month] = sorted
|
|
return sorted_dates
|
|
|
|
def sort_number_strings(data):
|
|
"""return array of dictionaries sorted by day treated as a number, i.e. 12 is after 3."""
|
|
number_data = [int(day["day"]) for day in data]
|
|
number_data.sort()
|
|
return [{"day": number, "source": data[get_index_for(number, data)]["source"],
|
|
"weekday": data[get_index_for(number, data)]["weekday"] } for i, number in enumerate(number_data)]
|
|
|
|
def weekday(number):
|
|
"""return name of the weekday for a specific number in range 0-6"""
|
|
weekdays = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
|
|
return weekdays[number]
|
|
|
|
def get_index_for(number, data):
|
|
"""get index of dictionary which day value equals number"""
|
|
index = 0
|
|
for i, day in enumerate(data):
|
|
if day["day"] == number:
|
|
index = i
|
|
return index
|
|
|
|
if __name__ == '__main__':
|
|
app.run()
|
|
|