Skip to content

Commit

Permalink
updating SQL query strings
Browse files Browse the repository at this point in the history
  • Loading branch information
ktabouguia committed Jan 5, 2019
1 parent 66d04df commit 40485c2
Showing 1 changed file with 12 additions and 12 deletions.
24 changes: 12 additions & 12 deletions bounce/db/membership.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,14 +45,14 @@ def insert(session, club_name, user_id):
club. """
# For now we do nothing on conflict, but when we have roles on these
# memberships we need to update on conflict.
query = f"""
query = """
INSERT INTO memberships (user_id, club_id) VALUES (
'{user_id}',
(SELECT id FROM clubs WHERE name = '{club_name}')
:user_id,
(SELECT id FROM clubs WHERE name = :club_name)
)
ON CONFLICT DO NOTHING
"""
session.execute(query)
session.execute(query, {'user_id' : user_id, 'club_name' : club_name})
session.commit()


Expand All @@ -61,17 +61,17 @@ def select(session, club_name, user_id=None):
Returns all memberships for the given club. If user_id is given, returns
only the membership for the given user.
"""
query = f"""
query = """
SELECT users.id AS user_id,
memberships.created_at, users.full_name, users.username FROM
memberships INNER JOIN users ON (memberships.user_id = users.id)
WHERE memberships.club_id IN (
SELECT id FROM clubs WHERE name = '{club_name}'
SELECT id FROM clubs WHERE name = :club_name
)
"""
if user_id:
query += f' AND user_id = {user_id}'
result_proxy = session.execute(query)
query += ' AND user_id = :user_id'
result_proxy = session.execute(query, {'user_id' : user_id, 'club_name' : club_name})
results = []
for row in result_proxy.fetchall():
results.append(
Expand All @@ -85,13 +85,13 @@ def delete(session, club_name, user_id=None):
Deletes all memberships for the given club. If user_id is given, deletes
only the membership for the given user.
"""
query = f"""
query = """
DELETE FROM memberships
WHERE memberships.club_id IN (
SELECT id FROM clubs WHERE name = '{club_name}'
SELECT id FROM clubs WHERE name = :club_name
)
"""
if user_id:
query += f' AND user_id = {user_id}'
session.execute(query)
query += ' AND user_id = :user_id'
session.execute(query, {'user_id' : user_id, 'club_name' : club_name})
session.commit()

0 comments on commit 40485c2

Please sign in to comment.