|
|
@ -295,6 +295,39 @@ class RegistrationWorkerStore(SQLBaseStore): |
|
|
|
return ret['user_id'] |
|
|
|
return ret['user_id'] |
|
|
|
return None |
|
|
|
return None |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@defer.inlineCallbacks |
|
|
|
|
|
|
|
def user_add_threepid(self, user_id, medium, address, validated_at, added_at): |
|
|
|
|
|
|
|
yield self._simple_upsert("user_threepids", { |
|
|
|
|
|
|
|
"medium": medium, |
|
|
|
|
|
|
|
"address": address, |
|
|
|
|
|
|
|
}, { |
|
|
|
|
|
|
|
"user_id": user_id, |
|
|
|
|
|
|
|
"validated_at": validated_at, |
|
|
|
|
|
|
|
"added_at": added_at, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@defer.inlineCallbacks |
|
|
|
|
|
|
|
def user_get_threepids(self, user_id): |
|
|
|
|
|
|
|
ret = yield self._simple_select_list( |
|
|
|
|
|
|
|
"user_threepids", { |
|
|
|
|
|
|
|
"user_id": user_id |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
['medium', 'address', 'validated_at', 'added_at'], |
|
|
|
|
|
|
|
'user_get_threepids' |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
defer.returnValue(ret) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def user_delete_threepid(self, user_id, medium, address): |
|
|
|
|
|
|
|
return self._simple_delete( |
|
|
|
|
|
|
|
"user_threepids", |
|
|
|
|
|
|
|
keyvalues={ |
|
|
|
|
|
|
|
"user_id": user_id, |
|
|
|
|
|
|
|
"medium": medium, |
|
|
|
|
|
|
|
"address": address, |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
desc="user_delete_threepids", |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RegistrationStore(RegistrationWorkerStore, |
|
|
|
class RegistrationStore(RegistrationWorkerStore, |
|
|
|
background_updates.BackgroundUpdateStore): |
|
|
|
background_updates.BackgroundUpdateStore): |
|
|
@ -632,39 +665,6 @@ class RegistrationStore(RegistrationWorkerStore, |
|
|
|
|
|
|
|
|
|
|
|
defer.returnValue(res if res else False) |
|
|
|
defer.returnValue(res if res else False) |
|
|
|
|
|
|
|
|
|
|
|
@defer.inlineCallbacks |
|
|
|
|
|
|
|
def user_add_threepid(self, user_id, medium, address, validated_at, added_at): |
|
|
|
|
|
|
|
yield self._simple_upsert("user_threepids", { |
|
|
|
|
|
|
|
"medium": medium, |
|
|
|
|
|
|
|
"address": address, |
|
|
|
|
|
|
|
}, { |
|
|
|
|
|
|
|
"user_id": user_id, |
|
|
|
|
|
|
|
"validated_at": validated_at, |
|
|
|
|
|
|
|
"added_at": added_at, |
|
|
|
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@defer.inlineCallbacks |
|
|
|
|
|
|
|
def user_get_threepids(self, user_id): |
|
|
|
|
|
|
|
ret = yield self._simple_select_list( |
|
|
|
|
|
|
|
"user_threepids", { |
|
|
|
|
|
|
|
"user_id": user_id |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
['medium', 'address', 'validated_at', 'added_at'], |
|
|
|
|
|
|
|
'user_get_threepids' |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
defer.returnValue(ret) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def user_delete_threepid(self, user_id, medium, address): |
|
|
|
|
|
|
|
return self._simple_delete( |
|
|
|
|
|
|
|
"user_threepids", |
|
|
|
|
|
|
|
keyvalues={ |
|
|
|
|
|
|
|
"user_id": user_id, |
|
|
|
|
|
|
|
"medium": medium, |
|
|
|
|
|
|
|
"address": address, |
|
|
|
|
|
|
|
}, |
|
|
|
|
|
|
|
desc="user_delete_threepids", |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@defer.inlineCallbacks |
|
|
|
@defer.inlineCallbacks |
|
|
|
def save_or_get_3pid_guest_access_token( |
|
|
|
def save_or_get_3pid_guest_access_token( |
|
|
|
self, medium, address, access_token, inviter_user_id |
|
|
|
self, medium, address, access_token, inviter_user_id |
|
|
|