tellicoViewer/test.py

#!/usr/bin/env python3

from xml.etree import ElementTree
import zipfile
from flask import Flask, render_template
app = Flask(__name__)

data = []
attributes = ["title",
              "authors",
              "isbn"]

fields = {}

@app.route("/r")
def update_data():
    global data
    out = []

    f = zipfile.ZipFile("library.tc")
    tree = ElementTree.parse(f.open("tellico.xml"))
    collection = tree.getroot()[0]
    ns = "{http://periapsis.org/tellico/}"

    for field in collection.find(ns + "fields"):
        #check if multiple entries are allowed
        if (int(field.attrib["flags"]) & 0x01 & 0xff) != 0:
            #change name to match
            name = field.attrib["name"] + "s"
        else:
            name = field.attrib["name"]
        fields[name] = field.attrib

    for entry in collection.iter(ns + "entry"):
        bookout = {}
        for datum_name in fields.keys():
            datum = entry.find(ns + datum_name)
            if datum != None:
                datum_type = int(fields[datum_name]["type"])
                #simple text, paragraph, multiple choice, image
                if datum_type == 1 or datum_type == 2 or datum_type == 3 or datum_type == 10:
                    #check if multiple entries are allowed
                    #if so, put entries in a list
                    if (int(fields[datum_name]["flags"]) & 0x01 & 0xff) != 0:
                        bookout[datum_name] = []
                        for sub in datum:
                            bookout[datum_name].append(sub.text)
                    else:
                        bookout[datum_name] = datum.text

                #date
                elif datum_type == 12:
                    bookout[datum_name] = ""
                    if datum.find(ns + "year") != None:
                        bookout[datum_name] += datum.find(ns + "year").text
                        if datum.find(ns + "month") != None:
                            bookout[datum_name] += "-" + datum.find(ns + "month").text
                            if datum.find(ns + "day") != None:
                                bookout[datum_name] += "-" + datum.find(ns + "day").text

                elif datum_type == 6: #Number
                    bookout[datum_name] = datum.text
                else:
                    print("Did not parse:" + datum_name + ", type:" + str(datum_type))
                    bookout[datum_name] = ""
            else:
                bookout[datum_name] = ""
        out.append(bookout.copy())
    data = out.copy()
    return "success"

@app.route("/")
def main():
    if data == []:
        update_data()
    return render_template("main.html", books=data, attributes=attributes)

@app.route("/book/<int:id_num>")
def book(id_num):
    if data == []:
        update_data()
    entry = next((item for item in data if int(item["id"]) == id_num), {})
    return render_template("book.html", book=entry, fields=fields)

if __name__ == "__main__":
    app.run(debug=True, host='0.0.0.0')