paulo@60: import datetime paulo@60: import cgi paulo@60: import urlparse paulo@60: import csv paulo@60: import traceback paulo@60: paulo@60: import html paulo@60: paulo@78: import pinlib paulo@78: paulo@60: paulo@60: DEBUG = True paulo@60: STRTIME_FMT = "%Y-%m-%d %H:%M:%S" paulo@60: paulo@60: paulo@60: class LLDialect(csv.Dialect): paulo@60: delimiter = '\t' paulo@60: quoting = csv.QUOTE_NONE paulo@60: lineterminator = '\n' paulo@60: paulo@60: paulo@60: LLDIALECT = LLDialect() paulo@60: LLDB_FN = "lldb.tsv" paulo@60: LLDB_UNREAD_FN = "lldb_unread.tsv" paulo@60: paulo@60: paulo@60: class LLError(Exception): paulo@60: pass paulo@60: paulo@60: class PinFailError(LLError): paulo@60: def __str__(self): paulo@60: return "PIN FAIL!" paulo@60: paulo@60: class MissingFieldsError(LLError): paulo@60: def __str__(self): paulo@60: return "MISSING FIELD(s)!" paulo@60: paulo@60: paulo@60: def lldb_unread_load(): paulo@60: return csv.reader(open(LLDB_UNREAD_FN), LLDIALECT) paulo@60: paulo@60: paulo@60: def lldb_add(inp): paulo@60: try: paulo@60: title = inp["title"][0] paulo@60: url = inp["url"][0] paulo@60: except (KeyError, IndexError): paulo@60: raise MissingFieldsError() paulo@60: paulo@60: dt_str = datetime.datetime.now().strftime(STRTIME_FMT) paulo@60: with open(LLDB_FN, 'a') as lldb_f: paulo@60: csv.writer(lldb_f, LLDIALECT).writerow([title, url, dt_str]) paulo@60: with open(LLDB_UNREAD_FN, 'a') as lldb_f: paulo@60: csv.writer(lldb_f, LLDIALECT).writerow([title, url, dt_str]) paulo@60: paulo@60: paulo@60: def lldb_unread_delete(inp): paulo@60: try: paulo@60: delete = inp["delete"] paulo@60: except KeyError: paulo@60: raise MissingFieldsError() paulo@60: paulo@60: lldb_unread = [i for i in lldb_unread_load()] paulo@60: lldb_unread_f = open(LLDB_UNREAD_FN, 'w') paulo@60: paulo@60: paulo@60: try: paulo@60: for i in delete: paulo@60: for j in lldb_unread: paulo@60: dt_str = j[2] paulo@60: if i == dt_str: paulo@60: lldb_unread.remove(j) paulo@60: finally: paulo@60: csv.writer(lldb_unread_f, LLDIALECT).writerows(lldb_unread) paulo@60: lldb_unread_f.close() paulo@60: paulo@60: paulo@60: paulo@60: def parse_wsgi_input(environ): paulo@60: return urlparse.parse_qs(environ["wsgi.input"].read()) paulo@60: paulo@60: paulo@60: paulo@60: def main(environ): paulo@60: is_post = (environ["REQUEST_METHOD"] == "POST") paulo@78: cookies = pinlib.parse_cookies(environ) paulo@60: inp = parse_wsgi_input(environ) paulo@60: paulo@60: if is_post: paulo@78: try: paulo@78: pinlib.check(cookies) paulo@78: except pinlib.PinFailError: paulo@78: raise PinFailError paulo@60: if inp["submit"][0] == "Add": paulo@60: lldb_add(inp) paulo@60: elif inp["submit"][0] == "Delete": paulo@60: lldb_unread_delete(inp) paulo@60: paulo@60: title = "later links..." paulo@60: root = html.HTML("html") paulo@60: paulo@60: header = root.header paulo@60: header.link(rel="stylesheet", type="text/css", href="index.css") paulo@60: header.title(title) paulo@60: paulo@60: body = root.body paulo@60: body.h1(title) paulo@60: paulo@60: paulo@60: if (DEBUG): paulo@60: debug = body.pre paulo@60: for i in environ.items(): paulo@60: debug += cgi.escape("%s = %s \n" % i) paulo@60: paulo@78: debug += cgi.escape("wsgi.input.read = %s \n" % inp) paulo@78: debug += cgi.escape("cookies = %s \n" % cookies) paulo@60: paulo@60: form = body.form(action="index.fcgi", method="post") paulo@60: paulo@60: table = form.table paulo@60: hrow = table.tr paulo@60: hrow.th("Link") paulo@60: hrow.th("Created") paulo@60: hrow.th.input(type="submit", name="submit", value="Delete") paulo@60: paulo@60: for i in lldb_unread_load(): paulo@60: (title, url, dt_str) = (j.decode("utf-8") for j in i) paulo@60: row = table.tr paulo@60: row.td.a(title, href=url) paulo@60: row.td(dt_str) paulo@60: row.td.input(type="checkbox", name="delete", value=dt_str) paulo@60: paulo@60: p1 = form.p paulo@60: p1.label("Title").input(type="text", name="title", size="64") paulo@60: p1.br paulo@60: p1.label("URL").input(type="text", name="url", size="64") paulo@60: p1.br paulo@60: p1.input(type="submit", name="submit", value="Add") paulo@60: paulo@60: return unicode(root).encode("utf-8") paulo@60: paulo@60: paulo@60: def app(environ, start_response): paulo@60: response_code = "500 Internal Server Error" paulo@60: response_type = "text/plain; charset=UTF-8" paulo@60: paulo@60: try: paulo@60: response_body = main(environ) paulo@78: response_code = "200 OK" paulo@60: response_type = "text/html; charset=UTF-8" paulo@60: except LLError as e: paulo@60: response_body = str(e) paulo@60: except: paulo@60: response_body = traceback.format_exc() paulo@60: paulo@60: response_headers = [ paulo@60: ("Content-Type", response_type), paulo@60: ("Content-Length", str(len(response_body))), paulo@60: ] paulo@60: start_response(response_code, response_headers) paulo@60: paulo@60: return [response_body]