Mercurial > kallithea
comparison rhodecode/model/user.py @ 1593:92a4f7c496a5 beta
Correct code style
author | Nicolas VINOT <aeris@imirhil.fr> |
---|---|
date | Sun, 23 Oct 2011 01:16:55 +0200 |
parents | 10d117545a7e |
children | 9dae92a65e40 |
comparison
equal
deleted
inserted
replaced
1592:8628c8706bf8 | 1593:92a4f7c496a5 |
---|---|
47 'repository.write': 3, | 47 'repository.write': 3, |
48 'repository.admin': 3} | 48 'repository.admin': 3} |
49 | 49 |
50 | 50 |
51 class UserModel(BaseModel): | 51 class UserModel(BaseModel): |
52 def get(self, user_id, cache=False): | 52 def get(self, user_id, cache = False): |
53 user = self.sa.query(User) | 53 user = self.sa.query(User) |
54 if cache: | 54 if cache: |
55 user = user.options(FromCache("sql_cache_short", | 55 user = user.options(FromCache("sql_cache_short", |
56 "get_user_%s" % user_id)) | 56 "get_user_%s" % user_id)) |
57 return user.get(user_id) | 57 return user.get(user_id) |
58 | 58 |
59 def get_by_username(self, username, cache=False, case_insensitive=False): | 59 def get_by_username(self, username, cache = False, case_insensitive = False): |
60 | 60 |
61 if case_insensitive: | 61 if case_insensitive: |
62 user = self.sa.query(User).filter(User.username.ilike(username)) | 62 user = self.sa.query(User).filter(User.username.ilike(username)) |
63 else: | 63 else: |
64 user = self.sa.query(User)\ | 64 user = self.sa.query(User)\ |
66 if cache: | 66 if cache: |
67 user = user.options(FromCache("sql_cache_short", | 67 user = user.options(FromCache("sql_cache_short", |
68 "get_user_%s" % username)) | 68 "get_user_%s" % username)) |
69 return user.scalar() | 69 return user.scalar() |
70 | 70 |
71 def get_by_api_key(self, api_key, cache=False): | 71 def get_by_api_key(self, api_key, cache = False): |
72 | 72 |
73 user = self.sa.query(User)\ | 73 user = self.sa.query(User)\ |
74 .filter(User.api_key == api_key) | 74 .filter(User.api_key == api_key) |
75 if cache: | 75 if cache: |
76 user = user.options(FromCache("sql_cache_short", | 76 user = user.options(FromCache("sql_cache_short", |
101 :param user_dn: | 101 :param user_dn: |
102 :param attrs: | 102 :param attrs: |
103 """ | 103 """ |
104 from rhodecode.lib.auth import get_crypt_password | 104 from rhodecode.lib.auth import get_crypt_password |
105 log.debug('Checking for such ldap account in RhodeCode database') | 105 log.debug('Checking for such ldap account in RhodeCode database') |
106 if self.get_by_username(username, case_insensitive=True) is None: | 106 if self.get_by_username(username, case_insensitive = True) is None: |
107 try: | 107 try: |
108 new_user = User() | 108 new_user = User() |
109 # add ldap account always lowercase | 109 # add ldap account always lowercase |
110 new_user.username = username.lower() | 110 new_user.username = username.lower() |
111 new_user.password = get_crypt_password(password) | 111 new_user.password = get_crypt_password(password) |
150 self.sa.rollback() | 150 self.sa.rollback() |
151 raise | 151 raise |
152 | 152 |
153 def update(self, user_id, form_data): | 153 def update(self, user_id, form_data): |
154 try: | 154 try: |
155 user = self.get(user_id, cache=False) | 155 user = self.get(user_id, cache = False) |
156 if user.username == 'default': | 156 if user.username == 'default': |
157 raise DefaultUserException( | 157 raise DefaultUserException( |
158 _("You can't Edit this user since it's" | 158 _("You can't Edit this user since it's" |
159 " crucial for entire application")) | 159 " crucial for entire application")) |
160 | 160 |
172 self.sa.rollback() | 172 self.sa.rollback() |
173 raise | 173 raise |
174 | 174 |
175 def update_my_account(self, user_id, form_data): | 175 def update_my_account(self, user_id, form_data): |
176 try: | 176 try: |
177 user = self.get(user_id, cache=False) | 177 user = self.get(user_id, cache = False) |
178 if user.username == 'default': | 178 if user.username == 'default': |
179 raise DefaultUserException( | 179 raise DefaultUserException( |
180 _("You can't Edit this user since it's" | 180 _("You can't Edit this user since it's" |
181 " crucial for entire application")) | 181 " crucial for entire application")) |
182 for k, v in form_data.items(): | 182 for k, v in form_data.items(): |
194 self.sa.rollback() | 194 self.sa.rollback() |
195 raise | 195 raise |
196 | 196 |
197 def delete(self, user_id): | 197 def delete(self, user_id): |
198 try: | 198 try: |
199 user = self.get(user_id, cache=False) | 199 user = self.get(user_id, cache = False) |
200 if user.username == 'default': | 200 if user.username == 'default': |
201 raise DefaultUserException( | 201 raise DefaultUserException( |
202 _("You can't remove this user since it's" | 202 _("You can't remove this user since it's" |
203 " crucial for entire application")) | 203 " crucial for entire application")) |
204 if user.repositories: | 204 if user.repositories: |
220 | 220 |
221 def reset_password(self, data): | 221 def reset_password(self, data): |
222 from rhodecode.lib.celerylib import tasks, run_task | 222 from rhodecode.lib.celerylib import tasks, run_task |
223 run_task(tasks.reset_user_password, data['email']) | 223 run_task(tasks.reset_user_password, data['email']) |
224 | 224 |
225 def fill_data(self, auth_user, user_id=None, api_key=None): | 225 def fill_data(self, auth_user, user_id = None, api_key = None): |
226 """ | 226 """ |
227 Fetches auth_user by user_id,or api_key if present. | 227 Fetches auth_user by user_id,or api_key if present. |
228 Fills auth_user attributes with those taken from database. | 228 Fills auth_user attributes with those taken from database. |
229 Additionally set's is_authenitated if lookup fails | 229 Additionally set's is_authenitated if lookup fails |
230 present in database | 230 present in database |
266 user.permissions['global'] = set() | 266 user.permissions['global'] = set() |
267 | 267 |
268 #====================================================================== | 268 #====================================================================== |
269 # fetch default permissions | 269 # fetch default permissions |
270 #====================================================================== | 270 #====================================================================== |
271 default_user = self.get_by_username('default', cache=True) | 271 default_user = self.get_by_username('default', cache = True) |
272 | 272 |
273 default_perms = self.sa.query(RepoToPerm, Repository, Permission)\ | 273 default_perms = self.sa.query(RepoToPerm, Repository, Permission)\ |
274 .join((Repository, RepoToPerm.repository_id == | 274 .join((Repository, RepoToPerm.repository_id == |
275 Repository.repo_id))\ | 275 Repository.repo_id))\ |
276 .join((Permission, RepoToPerm.permission_id == | 276 .join((Permission, RepoToPerm.permission_id == |