# HG changeset patch # User Søren Løvborg # Date 2015-07-26 13:58:50 # Node ID 598d0d0c4190320c507726d99b71d3921242cda3 # Parent 102bbfe8af43fe63edb4a3e5f1acf3b8d54818e9 UserModel: remove methods that are redundant with db.User UserModel().get_by_api_key is exactly equivalent to User.get_by_api_key. UserModel's get_by_username and get_by_email are not exactly identical to their db.User counterparts, due to a difference in the order of optional arguments. Fortunately, these optional arguments are never used. diff --git a/kallithea/controllers/api/api.py b/kallithea/controllers/api/api.py --- a/kallithea/controllers/api/api.py +++ b/kallithea/controllers/api/api.py @@ -672,10 +672,10 @@ class ApiController(JSONRPCController): """ - if UserModel().get_by_username(username): + if User.get_by_username(username): raise JSONRPCError("user `%s` already exist" % (username,)) - if UserModel().get_by_email(email, case_insensitive=True): + if User.get_by_email(email, case_insensitive=True): raise JSONRPCError("email `%s` already exist" % (email,)) if Optional.extract(extern_name): diff --git a/kallithea/model/user.py b/kallithea/model/user.py --- a/kallithea/model/user.py +++ b/kallithea/model/user.py @@ -59,15 +59,6 @@ class UserModel(BaseModel): def get_user(self, user): return self._get_user(user) - def get_by_username(self, username, cache=False, case_insensitive=False): - return User.get_by_username(username, case_insensitive, cache) - - def get_by_email(self, email, cache=False, case_insensitive=False): - return User.get_by_email(email, case_insensitive, cache) - - def get_by_api_key(self, api_key, cache=False): - return User.get_by_api_key(api_key, cache) - def create(self, form_data, cur_user=None): if not cur_user: cur_user = getattr(get_current_authuser(), 'username', None) @@ -348,9 +339,9 @@ class UserModel(BaseModel): if user_id is not None: dbuser = self.get(user_id) elif api_key is not None: - dbuser = self.get_by_api_key(api_key) + dbuser = User.get_by_api_key(api_key) elif username is not None: - dbuser = self.get_by_username(username) + dbuser = User.get_by_username(username) if dbuser is not None and dbuser.active: log.debug('filling %s data' % dbuser) diff --git a/kallithea/tests/api/api_base.py b/kallithea/tests/api/api_base.py --- a/kallithea/tests/api/api_base.py +++ b/kallithea/tests/api/api_base.py @@ -94,7 +94,7 @@ class _BaseTestApi(object): @classmethod def setup_class(cls): - cls.usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + cls.usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) cls.apikey = cls.usr.api_key cls.test_user = UserModel().create_or_update( username='test-api', @@ -233,7 +233,7 @@ class _BaseTestApi(object): userid=TEST_USER_ADMIN_LOGIN) response = api_call(self, params) - usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) ret = usr.get_api_data() ret['permissions'] = AuthUser(usr.user_id).permissions @@ -252,7 +252,7 @@ class _BaseTestApi(object): id_, params = _build_data(self.apikey, 'get_user') response = api_call(self, params) - usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) ret = usr.get_api_data() ret['permissions'] = AuthUser(usr.user_id).permissions @@ -263,7 +263,7 @@ class _BaseTestApi(object): id_, params = _build_data(self.apikey_regular, 'get_user') response = api_call(self, params) - usr = UserModel().get_by_username(self.TEST_USER_LOGIN) + usr = User.get_by_username(self.TEST_USER_LOGIN) ret = usr.get_api_data() ret['permissions'] = AuthUser(usr.user_id).permissions @@ -580,7 +580,7 @@ class _BaseTestApi(object): password='trololo') response = api_call(self, params) - usr = UserModel().get_by_username(username) + usr = User.get_by_username(username) ret = dict( msg='created new user `%s`' % username, user=jsonify(usr.get_api_data()) @@ -601,7 +601,7 @@ class _BaseTestApi(object): email=email) response = api_call(self, params) - usr = UserModel().get_by_username(username) + usr = User.get_by_username(username) ret = dict( msg='created new user `%s`' % username, user=jsonify(usr.get_api_data()) @@ -621,7 +621,7 @@ class _BaseTestApi(object): email=email, extern_name='internal') response = api_call(self, params) - usr = UserModel().get_by_username(username) + usr = User.get_by_username(username) ret = dict( msg='created new user `%s`' % username, user=jsonify(usr.get_api_data()) @@ -697,7 +697,7 @@ class _BaseTestApi(object): ('password', 'newpass') ]) def test_api_update_user(self, name, expected): - usr = UserModel().get_by_username(self.TEST_USER_LOGIN) + usr = User.get_by_username(self.TEST_USER_LOGIN) kw = {name: expected, 'userid': usr.user_id} id_, params = _build_data(self.apikey, 'update_user', **kw) @@ -706,7 +706,7 @@ class _BaseTestApi(object): ret = { 'msg': 'updated user ID:%s %s' % ( usr.user_id, self.TEST_USER_LOGIN), - 'user': jsonify(UserModel() \ + 'user': jsonify(User \ .get_by_username(self.TEST_USER_LOGIN) \ .get_api_data()) } @@ -715,7 +715,7 @@ class _BaseTestApi(object): self._compare_ok(id_, expected, given=response.body) def test_api_update_user_no_changed_params(self): - usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) ret = jsonify(usr.get_api_data()) id_, params = _build_data(self.apikey, 'update_user', userid=TEST_USER_ADMIN_LOGIN) @@ -730,7 +730,7 @@ class _BaseTestApi(object): self._compare_ok(id_, expected, given=response.body) def test_api_update_user_by_user_id(self): - usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) ret = jsonify(usr.get_api_data()) id_, params = _build_data(self.apikey, 'update_user', userid=usr.user_id) @@ -755,7 +755,7 @@ class _BaseTestApi(object): @mock.patch.object(UserModel, 'update_user', crash) def test_api_update_user_when_exception_happens(self): - usr = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN) + usr = User.get_by_username(TEST_USER_ADMIN_LOGIN) ret = jsonify(usr.get_api_data()) id_, params = _build_data(self.apikey, 'update_user', userid=usr.user_id)