I am building an application using Flask and SQLAlchemy. I basically have 3 tables: users, friendships and best friends:
A user can have many friends, but only one best friend. Therefore, I want my model to be relational. A one-to-many relationship between users and friends and a one-to-one relationship between users and best friends.
This is my model:
from app import db
from sqlalchemy.orm import relationship, backref
from sqlalchemy import Table, Column, Integer, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
class users(db.Model):
__tablename__ = "Users"
id = db.Column(db.Integer, primary_key=True)
userName = db.Column(db.String, nullable=False)
userEmail = db.Column(db.String, nullable=False)
userPhone = db.Column(db.String, nullable=False)
userPass = db.Column(db.String, nullable=False)
friendsR = db.relationship('friendships', backref='friendships.friend_id', primaryjoin='users.id==friendships.user_id', lazy='joined')
def __init__(self, userName, userEmail, userPhone, userPass):
self.userName = userName
self.userEmail = userEmail
self.userPhone = userPhone
self.userPass = userPass
def __repr__(self):
return '{}-{}-{}-{}'.format(self.id, self.userName, self.userEmail, self.userPhone)
class friendships(db.Model):
__tablename__ = "Friendships"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, ForeignKey('Users.id'), nullable=False)
friend_id = db.Column(db.Integer, ForeignKey('Users.id'), nullable=False)
userR = relationship('users', foreign_keys='friendships.user_id')
friendR = relationship('users', foreign_keys='friendships.friend_id')
def __init__(self, user_id, friend_id):
self.user_id = user_id
self.friend_id = friend_id
def __repr__(self):
return '{}-{}-{}-{}'.format(self.user_id, self.friend_id)
class bestFriends(db.Model):
__tablename__ = "BestFriends"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, ForeignKey('Users.id'), nullable=False)
best_friend_id = db.Column(db.Integer, ForeignKey('Users.id'), nullable=False)
user = relationship('users', foreign_keys='bestFriends.user_id')
best_friend = relationship('users', foreign_keys='bestFriends.best_friend_id')
def __init__(self, user_id, best_friend_id):
self.user_id = user_id
self.best_friend_id = best_friend_id
def __repr__(self):
return '{}-{}-{}-{}'.format(self.user_id, self.best_friend_id)
I need to be able to request the list of friends of the user who is logged in, as well as the best friend of this user, if available. I also need the PAGINATE result:
Here is my app.py function to display user friends:
@app.route('/friendList<int:page>', methods=['GET', 'POST'])
@app.route('/friends')
def friendList(page=1):
if not session.get('logged_in'):
return render_template('login.html')
else:
userID = session['user_id']
userList = users.query.join(friendships).add_columns(users.id, users.userName, users.userEmail, friendships.user_id, friendships.friend_id).filter(users.id == friendships.friend_id).filter(friendships.user_id == userID).paginate(page, 1, False)
return render_template(
'friends.html', userList=userList)
And this will be the side of Jinja code:
{% extends "layout.html" %}
{% block body %}
<div id="pagination">
{% if userList.has_prev %}
<a href="{{ url_for('friendList', page=userList.prev_num) }}">Back</a>
{% endif %}
{% if userList.has_next %}
<a href="{{ url_for('friendList', page=userList.next_num) }}">Next</a>
{% endif %}
</div>
<div style="clear:both;"></div>
<div id="innerContent">
{% if userList %}
{% for friends in userList %}
<div class="contentUsers">
{{ friends.userName }}
</div>
<br><br><br><br>
{% endfor %}{% else %}<div>No friends</div>
{% endif %}
</div>
{% endblock %}
And if I ask like this:
userList = db.session.query(users,friendships).filter(users.id == friendships.friend_id).filter(friendships.user_id == userID).paginate(page, 1, False)
:
InvalidRequestError: Could not find a FROM clause to join from. Tried joining to <class 'models.friendships'>, but got: Can't determine join between 'Users' and 'Friendships'; tables have more than one foreign key constraint relationship between them. Please specify the 'onclause' of this join explicitly.
:
userList = users.query.join(friendships, users.id==friendships.user_id).add_columns(users.id, users.userName, users.userEmail, friendships.user_id, friendships.friend_id).filter(users.id == friendships.friend_id).filter(friendships.user_id == userID).paginate(page, 1, False)
:
TypeError: 'Pagination' object is not iterable
, - , , - / !!!
Jinja .items :
{% if userList.items %}
{% for friends in userList.items %}
<div class="contentUsers">
{{ friends.userName }}
</div>
<br><br><br><br>
{% endfor %}{% else %}<div>No friends</div>
{% endif%}
else "no friends"