LibraryHippo 2020 - A Small Heroku Datastore

def check_card(self, patron, number, pin):
    session = Session()
    summary_page = self.login(session, patron, number, pin)

    holds_url = urllib.parse.urljoin(
        self.login_url(),
        summary_page.find(name="a", href=re.compile("/holds$"))["href"],
    )
    checkouts_url = urllib.parse.urljoin(
        self.login_url(),
        summary_page.find(name="a", href=re.compile("/items$"))["href"],
    )

    holds = self.get_holds(session, holds_url)
    checkouts = self.get_checkouts(session, checkouts_url)

    return {
        "holds": holds,
        "checkouts": checkouts,
    }

@app.route("/check")
def check():
    card_check_result = WPL().check_card(
        Config.PATRON_NAME, Config.CARD_NUMBER, Config.PIN
    )

    result = "<h1>Holds</h1>"
    for hold in card_check_result["holds"]:
        result += "<dl>"
        for k, v in hold.items():
            result += f"<dt>{k}</dt><dd>{v}</dd>"
        result += "</dl><hr>"

    result += "<h1>Checkouts</h1>"
    for checkout in card_check_result["checkouts"]:
        result += "<dl>"
        for k, v in checkout.items():
            result += f"<dt>{k}</dt><dd>{v}</dd>"
        result += "</dl><hr>"

    return result

class Config(object):
    # …
    # Remove PATRON_NAME, CARD_NUMBER, and PIN, as they'll move to the database

    SQLALCHEMY_DATABASE_URI = os.environ.get(
        "DATABASE_URL"
    ) or "sqlite:///" + os.path.join(basedir, "app.db")
    SQLALCHEMY_TRACK_MODIFICATIONS = False

# …
from flask_migrate import Migrate
from flask_sqlalchemy import SQLAlchemy

# …
app = Flask(__name__)
app.config.from_object(Config)

mail = Mail(app)

db = SQLAlchemy(app)
migrate = Migrate(app, db)

from app import routes, models

class Card(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    patron_name = db.Column(db.String(64))
    number = db.Column(db.String(32))
    pin = db.Column(db.String(16))
    last_state = db.Column(db.Text())

    def __repr__(self):
        return f"<Patron {self.patron_name}>"

# …
import json
from app.models import Card

# …

@app.route("/check")
def check():
    card = Card.query.get(1)  # a hack - we know there's only 1 card for now
    card_check_result = WPL().check_card(card)
    card.last_state = json.dumps(card_check_result)
    db.session.commit()
    # rendering code…

# …
from app import mail, models

# …

def register(app):
    @app.cli.command("notify-all")
    def notify_all():
        card = models.Card.query.get(1)  # a hack - we know there's only 1 card for now
        last_card_state = json.loads(card.last_state)

        html_body = "<h1>Holds</h1>"
        for hold in last_card_state["holds"]:
            html_body += "<dl>"
            for k, v in hold.items():
                html_body += f"<dt>{k}</dt><dd>{v}</dd>"
            html_body += "</dl><hr>"

        html_body += "<h1>Checkouts</h1>"
        for checkout in last_card_state["checkouts"]:
            html_body += "<dl>"
            for k, v in checkout.items():
                html_body += f"<dt>{k}</dt><dd>{v}</dd>"
            html_body += "</dl><hr>"

        now = datetime.now().isoformat()

        msg = Message(
            "LibraryHippo starting notifications", recipients=["[email protected]"]
        )
        msg.body = f"starting notifications at {now}"
        msg.html = html_body

        mail.send(msg)

        # …