photoserver/framesets.py at main · EdLeafe/photoserver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
from __future__ import absolute_import, print_function, unicode_literals

from dataclasses import dataclass
from datetime import datetime

from flask import (
    Flask,
    abort,
    g,
    make_response,
    redirect,
    render_template,
    request,
    session,
    url_for,
)

import albums
import entities
import frames
import utils


def GET_list():
    crs = utils.get_cursor()
    framesets = entities.Frameset.list()
    g.framesets = sorted([fs.to_dict() for fs in framesets], key=lambda x: x["name"].upper())
    albums = entities.Album.list()
    g.albums = sorted(
        [ab.to_dict() for ab in albums if not ab.parent_id], key=lambda x: x["name"].upper()
    )
    g.request_string = str(request.headers)
    return render_template("frameset_list.html")


def set_album(frameset_id, album_id):
    fs = entities.Frameset.get(frameset_id)
    fs.assign_album(album_id)
    return ""


def show(frameset_id):
    if frameset_id is None:
        g.frameset = {"name": "", "pkid": "", "orientation": ""}
    else:
        crs = utils.get_cursor()
        crs.execute("select * from frameset where pkid = %s", (frameset_id,))
        g.frameset = crs.fetchall()[0]
        sql = """select frame.pkid, frame.name, frame.description
                from frame
                where frame.frameset_id = %s;"""
        crs.execute(sql, (frameset_id,))
        g.frames = crs.fetchall()
    return render_template("frameset_detail.html")


def delete(pkid=None):
    if pkid is None:
        # Form
        pkid = request.form["pkid"]
    crs = utils.get_cursor()
    # Get the file name
    sql = "select name from frameset where pkid = %s"
    res = crs.execute(sql, (pkid,))
    if not res:
        abort(404)
    # Blank out the frameset_id from any children
    sql = "update frame set frameset_id = '' where frameset_id = %s"
    crs.execute(sql, (pkid,))
    sql = "delete from frameset where pkid = %s"
    crs.execute(sql, (pkid,))
    utils.commit()
    return redirect(url_for("list_framesets"))


def update():
    rf = request.form
    if "delete" in rf:
        return delete()
    pkid = rf["pkid"]
    name = rf["name"]
    orientation = rf["orientation"]
    crs = utils.get_cursor()
    if not pkid:
        # New frameset
        pkid = utils.gen_uuid()
        sql = """insert into frameset (pkid, name, orientation)
                 values (%s, %s, %s); """
        vals = (pkid, name, orientation)
    else:
        sql = """update frameset set name = %s, orientation = %s
                 where pkid = %s; """
        vals = (name, orientation, pkid)
    crs.execute(sql, vals)
    utils.commit()
    return redirect(url_for("list_framesets"))


def manage_frames(frameset_id):
    fs = entities.Frameset.get(frameset_id)
    g.frameset = fs.to_dict()
    frames = [frm.to_dict() for frm in entities.Frame.list()]
    for frm in frames:
        frm["selected"] = 1 if frm["frameset_id"] == frameset_id else 0
    g.frames = sorted(frames, key=lambda k: -k["selected"])
    return render_template("frameset_frames.html")


def manage_frames_POST(frameset_id):
    frame_ids = request.form.getlist("selected")
    utils.debugout("FRAMEIDS", frame_ids)
    fs = entities.Frameset.get(frameset_id)
    fs.set_frames(frame_ids)
    return redirect("/framesets")