Skip to content

Commit

Permalink
Standardize instances of default thumbnail names
Browse files Browse the repository at this point in the history
Now where instances of the string literal "reedphoto" would exist have
been replaced with references to a constant DEFAULT_THUMBNAIL_NAME

Effort Towards Bug #16
  • Loading branch information
liam-middlebrook committed Dec 29, 2017
1 parent 1c226ba commit f44f223
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 8 deletions.
15 changes: 8 additions & 7 deletions gallery/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@
from gallery.models import File
from gallery.models import Tag

from gallery.util import DEFAULT_THUMBNAIL_NAME
from gallery.util import get_dir_file_contents
from gallery.util import get_dir_tree_dict
from gallery.util import get_full_dir_path
Expand Down Expand Up @@ -271,7 +272,7 @@ def add_directory(parent_id, name, description, owner):
if sibling.get_name() == name:
return None

uuid_thumbnail = "reedphoto"
uuid_thumbnail = DEFAULT_THUMBNAIL_NAME
dir_model = Directory(parent_id, name, description, owner,
uuid_thumbnail, "{\"g\":[]}")
db.session.add(dir_model)
Expand All @@ -282,7 +283,7 @@ def add_directory(parent_id, name, description, owner):
return dir_model.id

def add_file(file_name, path, dir_id, description, owner):
uuid_thumbnail = "reedphoto"
uuid_thumbnail = DEFAULT_THUMBNAIL_NAME

file_path = os.path.join('/', path, file_name)

Expand All @@ -305,18 +306,18 @@ def refresh_thumbnail_helper(dir_model):
dir_children = [d for d in Directory.query.filter(Directory.parent == dir_model.id).all()]
file_children = [f for f in File.query.filter(File.parent == dir_model.id).all()]
for file in file_children:
if file.thumbnail_uuid != "reedphoto":
if file.thumbnail_uuid != DEFAULT_THUMBNAIL_NAME:
return file.thumbnail_uuid
for d in dir_children:
if d.thumbnail_uuid != "reedphoto":
if d.thumbnail_uuid != DEFAULT_THUMBNAIL_NAME:
return d.thumbnail_uuid
# WE HAVE TO GO DEEPER (inception noise)
for d in dir_children:
return refresh_thumbnail_helper(d)
# No thumbnail found
return "reedphoto"
return DEFAULT_THUMBNAIL_NAME

missing_thumbnails = File.query.filter(File.thumbnail_uuid == "reedphoto").all()
missing_thumbnails = File.query.filter(File.thumbnail_uuid == DEFAULT_THUMBNAIL_NAME).all()
for file_model in missing_thumbnails:
dir_path = get_full_dir_path(file_model.parent)
file_path = os.path.join(dir_path, file_model.name)
Expand All @@ -326,7 +327,7 @@ def refresh_thumbnail_helper(dir_model):
db.session.commit()
db.session.refresh(file_model)

missing_thumbnails = Directory.query.filter(Directory.thumbnail_uuid == "reedphoto").all()
missing_thumbnails = Directory.query.filter(Directory.thumbnail_uuid == DEFAULT_THUMBNAIL_NAME).all()
for dir_model in missing_thumbnails:
dir_model.thumbnail_uuid = refresh_thumbnail_helper(dir_model)
db.session.flush()
Expand Down
3 changes: 2 additions & 1 deletion gallery/file_modules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from wand.image import Image
from wand.color import Color

from gallery.util import DEFAULT_THUMBNAIL_NAME
from gallery.util import hash_file
from gallery.util import convert_bytes_to_utf8

Expand All @@ -11,7 +12,7 @@ class FileModule:
dir_path = None
exif_dict = {'Exif': {}}
mime_type = None
thumbnail_uuid = "reedphoto"
thumbnail_uuid = DEFAULT_THUMBNAIL_NAME
file_name = None

def __init__(self, file_path, dir_path):
Expand Down
2 changes: 2 additions & 0 deletions gallery/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@
from gallery.models import Directory
from gallery.models import File

DEFAULT_THUMBNAIL_NAME = 'reedphoto'

def get_dir_file_contents(dir_id):
contents = [f for f in File.query.filter(File.parent == dir_id).all()]
contents.sort(key=lambda x: x.get_name())
Expand Down

0 comments on commit f44f223

Please sign in to comment.