62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from flask import (
|
|
Blueprint, redirect, render_template, request, session, url_for
|
|
)
|
|
from requests import get
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from config import CONFIG
|
|
|
|
from create_db import Card, User
|
|
|
|
bp = Blueprint('matches', __name__, url_prefix='/matches')
|
|
|
|
|
|
engine = create_engine(CONFIG['DB_CONNECTION'])
|
|
dbsessionmaker = sessionmaker(bind=engine)
|
|
dbsession = dbsessionmaker()
|
|
|
|
|
|
def get_matches(user_id):
|
|
list_of_matches = []
|
|
|
|
#in all shared cards for the ones you votes yes
|
|
all_cards = dbsession.query(Card)
|
|
all_shared_of_user_of_ir_1 = all_cards.filter(Card.share_id != "0", Card.owner_id == user_id, Card.interest_rate == 1).all()
|
|
if all_shared_of_user_of_ir_1 == []:
|
|
return list_of_matches
|
|
|
|
# see who else voted yes
|
|
for c in all_shared_of_user_of_ir_1:
|
|
others_yes = all_cards.filter(Card.share_id == c.share_id, Card.interest_rate == 1).all()
|
|
if others_yes == []:
|
|
pass
|
|
else:
|
|
list_of_matches.append(others_yes)
|
|
|
|
return list_of_matches
|
|
|
|
@bp.route("/", methods=("GET", "POST"))
|
|
def index():
|
|
if not 'user_id' in session:
|
|
redirect(url_for('index'))
|
|
user_id = session["user_id"]
|
|
username = session["username"]
|
|
|
|
list_of_matches = get_matches(user_id)
|
|
|
|
#@TODO tu bi morali dodati še nek users object, da own_id pretvorimo v username
|
|
#relevant userids, but might be uselessly duplicated
|
|
user_ids = []
|
|
for match in list_of_matches:
|
|
for card in match:
|
|
user_ids.append(card.owner_id)
|
|
|
|
users = dbsession.query(User)
|
|
names_by_ids = {}
|
|
|
|
for id in user_ids:
|
|
|
|
names_by_ids[id] = users.get(id).username
|
|
|
|
return render_template("matches.html", username=username, list_of_matches=list_of_matches, names_by_ids=names_by_ids) |