summary refs log tree commit diff
path: root/synapse
diff options
context:
space:
mode:
authorDaniel Wagner-Hall <dawagner@gmail.com>2015-11-19 10:07:21 -0500
committerDaniel Wagner-Hall <dawagner@gmail.com>2015-11-19 10:07:21 -0500
commit9da4c5340da7e5a8e03a3bd7e028a1c862ce9616 (patch)
tree628617bb92d56391b71ab62079c81e5767dfa951 /synapse
parentApply forgetting properly to historical events (diff)
downloadsynapse-9da4c5340da7e5a8e03a3bd7e028a1c862ce9616.tar.xz
Simplify code
Diffstat (limited to '')
-rw-r--r--synapse/handlers/_base.py2
-rw-r--r--synapse/storage/roommember.py12
2 files changed, 6 insertions, 8 deletions
diff --git a/synapse/handlers/_base.py b/synapse/handlers/_base.py
index 95bb06395a..5fd20285d2 100644
--- a/synapse/handlers/_base.py
+++ b/synapse/handlers/_base.py
@@ -93,7 +93,7 @@ class BaseHandler(object):
             membership_event = state.get((EventTypes.Member, user_id), None)
             if membership_event:
                 was_forgotten_at_event = yield self.store.was_forgotten_at(
-                    membership_event.user_id,
+                    membership_event.state_key,
                     membership_event.room_id,
                     membership_event.event_id
                 )
diff --git a/synapse/storage/roommember.py b/synapse/storage/roommember.py
index 5e92cdc811..c3e11b91da 100644
--- a/synapse/storage/roommember.py
+++ b/synapse/storage/roommember.py
@@ -295,7 +295,7 @@ class RoomMemberStore(SQLBaseStore):
             sql = (
                 "SELECT"
                 "  COUNT(*)"
-                "FROM"
+                " FROM"
                 "  room_memberships"
                 " WHERE"
                 "  user_id = ?"
@@ -318,8 +318,8 @@ class RoomMemberStore(SQLBaseStore):
         def f(txn):
             sql = (
                 "SELECT"
-                "  COUNT(*)"
-                "FROM"
+                "  forgotten"
+                " FROM"
                 "  room_memberships"
                 " WHERE"
                 "  user_id = ?"
@@ -327,11 +327,9 @@ class RoomMemberStore(SQLBaseStore):
                 "  room_id = ?"
                 " AND"
                 "  event_id = ?"
-                " AND"
-                "  forgotten = 1"
             )
             txn.execute(sql, (user_id, room_id, event_id))
             rows = txn.fetchall()
             return rows[0][0]
-        count = yield self.runInteraction("did_forget_membership_at", f)
-        defer.returnValue(count == 1)
+        forgot = yield self.runInteraction("did_forget_membership_at", f)
+        defer.returnValue(forgot == 1)