Simplify code

This commit is contained in:
Daniel Wagner-Hall 2015-11-19 10:07:21 -05:00
parent bed7889703
commit 9da4c5340d
2 changed files with 6 additions and 8 deletions

View File

@ -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
)

View File

@ -318,7 +318,7 @@ class RoomMemberStore(SQLBaseStore):
def f(txn):
sql = (
"SELECT"
" COUNT(*)"
" forgotten"
" FROM"
" room_memberships"
" WHERE"
@ -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)