Mercurial > kallithea

comparison rhodecode/model/user.py @ 1593:92a4f7c496a5 beta

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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 ==