diff --git a/sr_session.py b/sr_session.py index 4a40660..a829c9d 100644 --- a/sr_session.py +++ b/sr_session.py @@ -5,6 +5,9 @@ from sqlalchemy import desc from share import share from create_db import Card, Rating, get_session from settings import get_settings +from link_handler import render_embed_link + + """ testing """ @@ -266,5 +269,19 @@ def sr_session(): show_card = dbsession.query(Card).get(next_card_id) interval = get_interval(next_card_id) - #these factors should be better packaged - return render_template("deck.html", username=username, card=show_card, maybe_in=round(interval*maybe_factor), no_in=round(no_factor*interval)) \ No newline at end of file + card_type = show_card.card_type + print("card type: ", card_type) + + if card_type == 'PDF': + #these factors should be better packaged + return render_template("deck.html", username=username, card=show_card, maybe_in=round(interval*maybe_factor), no_in=round(no_factor*interval)) + elif card_type == 'URL': + #templates for different kinds of links + return render_embed_link(cm_username=username, + card=show_card, + maybe_in=round(interval*maybe_factor), + no_in=round(no_factor*interval)) + + + + #return render_template("link.html", username=username, card=show_card, maybe_in=round(interval*maybe_factor), no_in=round(no_factor*interval)) \ No newline at end of file