Cleanup: More minor dusting, commenting, clarity renaming.
This commit is contained in:
		
							parent
							
								
									bb5fa816d8
								
							
						
					
					
						commit
						96856e9361
					
				
					 4 changed files with 43 additions and 31 deletions
				
			
		|  | @ -5,8 +5,8 @@ import traceback | |||
| 
 | ||||
| try: | ||||
|     ffmpeg = converter.Converter( | ||||
|         ffmpeg_path='C:\\software\\ffmpeg\\bin\\ffmpeg.exe', | ||||
|         ffprobe_path='C:\\software\\ffmpeg\\bin\\ffprobe.exe', | ||||
|         ffmpeg_path='D:\\software\\ffmpeg\\bin\\ffmpeg.exe', | ||||
|         ffprobe_path='D:\\software\\ffmpeg\\bin\\ffprobe.exe', | ||||
|     ) | ||||
| except converter.ffmpeg.FFMpegError: | ||||
|     traceback.print_exc() | ||||
|  |  | |||
|  | @ -249,6 +249,11 @@ class Album(ObjectBase, GroupableMixin): | |||
|     @decorators.required_feature('album.edit') | ||||
|     @decorators.transaction | ||||
|     def add_associated_directory(self, filepath, *, commit=True): | ||||
|         ''' | ||||
|         Add a directory from which this album will pull files during rescans. | ||||
|         These relationships are not unique and multiple albums | ||||
|         can associate with the same directory if desired. | ||||
|         ''' | ||||
|         filepath = pathclass.Path(filepath) | ||||
|         if not filepath.is_dir: | ||||
|             raise ValueError('%s is not a directory' % filepath) | ||||
|  |  | |||
|  | @ -811,8 +811,7 @@ class PDBPhotoMixin: | |||
|                         #print('Failed tag expression') | ||||
|                         continue | ||||
| 
 | ||||
|                 elif is_must_may_forbid: | ||||
|                     pass | ||||
|                 # elif is_must_may_forbid: | ||||
|                 #     if photo.id not in mmf_results: | ||||
|                 #         #print('Failed tag mmf') | ||||
|                 #         continue | ||||
|  | @ -991,21 +990,27 @@ class PDBUserMixin: | |||
|         else: | ||||
|             raise exceptions.NoSuchUser(username or id) | ||||
| 
 | ||||
|     def get_user_id_or_none(self, user): | ||||
|     def get_user_id_or_none(self, user_obj_or_id): | ||||
|         ''' | ||||
|         For methods that create photos, albums, etc., we sometimes associate | ||||
|         them with an author but sometimes not. This method hides validation | ||||
|         that those methods would otherwise have to duplicate. | ||||
|         them with an author but sometimes not. The callers of those methods | ||||
|         might be trying to use a User object, or a user's ID, or maybe they | ||||
|         left it None. | ||||
|         This method hides validation that those methods would otherwise | ||||
|         have to duplicate. | ||||
|         ''' | ||||
|         if isinstance(user, objects.User): | ||||
|             if user.photodb != self: | ||||
|                 raise ValueError('That user does not belong to this photodb') | ||||
|             author_id = user.id | ||||
|         elif user is not None: | ||||
|             # Confirm that this string is an ID and not junk. | ||||
|             author_id = self.get_user(id=user).id | ||||
|         else: | ||||
|         if user_obj_or_id is None: | ||||
|             author_id = None | ||||
| 
 | ||||
|         elif isinstance(user_obj_or_id, objects.User): | ||||
|             if user_obj_or_id.photodb != self: | ||||
|                 raise ValueError('That user does not belong to this photodb') | ||||
|             author_id = user_obj_or_id.id | ||||
| 
 | ||||
|         else: | ||||
|             # Confirm that this string is a valid ID and not junk. | ||||
|             author_id = self.get_user(id=user_obj_or_id).id | ||||
| 
 | ||||
|         return author_id | ||||
| 
 | ||||
|     @decorators.required_feature('user.login') | ||||
|  | @ -1017,11 +1022,11 @@ class PDBUserMixin: | |||
|         if fetch is None: | ||||
|             raise exceptions.WrongLogin() | ||||
| 
 | ||||
|         stored_password = fetch[constants.SQL_USER['password']] | ||||
| 
 | ||||
|         if not isinstance(password, bytes): | ||||
|             password = password.encode('utf-8') | ||||
| 
 | ||||
|         stored_password = fetch[constants.SQL_USER['password']] | ||||
| 
 | ||||
|         success = bcrypt.checkpw(password, stored_password) | ||||
|         if not success: | ||||
|             raise exceptions.WrongLogin() | ||||
|  |  | |||
|  | @ -27,7 +27,9 @@ def build_query( | |||
| 
 | ||||
|     if mmf_results: | ||||
|         # "id IN/NOT IN (1, 2, 3)" | ||||
|         wheres.add('id %s %s' % (mmf_results['operator'], helpers.sql_listify(mmf_results['photoids']))) | ||||
|         operator = mmf_results['operator'] | ||||
|         photo_ids = helpers.sql_listify(mmf_results['photo_ids']) | ||||
|         wheres.add('id %s %s' % (operator, photo_ids)) | ||||
| 
 | ||||
|     if orderby: | ||||
|         orderby = [o.split('-') for o in orderby] | ||||
|  |  | |||
		Loading…
	
		Reference in a new issue