Toot toot! It's the refactor tractor.

- Create helper functions to qualify media and static files
This commit is contained in:
Oliver Walters 2020-04-07 11:38:57 +10:00
parent ae9ef04013
commit 95032141ce
3 changed files with 46 additions and 10 deletions

View File

@ -14,6 +14,41 @@ from django.core.exceptions import ValidationError
from django.utils.translation import ugettext as _
from .version import inventreeVersion, inventreeInstanceName
from .settings import MEDIA_URL, STATIC_URL
def getMediaUrl(filename):
"""
Return the qualified access path for the given file,
under the media directory.
"""
return os.path.join(MEDIA_URL, str(filename))
def getStaticUrl(filename):
"""
Return the qualified access path for the given file,
under the static media directory.
"""
return os.path.join(STATIC_URL, str(filename))
def getBlankImage():
"""
Return the qualified path for the 'blank image' placeholder.
"""
return getStaticUrl("img/blank_image.png")
def getBlankThumbnail():
"""
Return the qualified path for the 'blank image' thumbnail placeholder.
"""
return getStaticUrl("img/blank_image.thumbnail.png")
def TestIfImage(img):

View File

@ -23,6 +23,7 @@ from markdownx.models import MarkdownxField
from stdimage.models import StdImageField
from InvenTree.helpers import getMediaUrl, getBlankImage, getBlankThumbnail
from InvenTree.fields import InvenTreeURLField, RoundingDecimalField
from InvenTree.status_codes import OrderStatus
from common.models import Currency
@ -118,17 +119,17 @@ class Company(models.Model):
""" Return the URL of the image for this company """
if self.image:
return os.path.join(settings.MEDIA_URL, str(self.image.url))
return getMediaUrl(self.image.url)
else:
return os.path.join(settings.STATIC_URL, 'img/blank_image.png')
return getBlankImage()
def get_thumbnail_url(self):
""" Return the URL for the thumbnail image for this Company """
if self.image:
return os.path.join(settings.MEDIA_URL, str(self.image.thumbnail.url))
return getMediaUrl(self.image.thumbnail.url)
else:
return os.path.join(settings.STATIC_URL, 'img/blank_image.thumbnail.png')
return getBlankThumbnail()
@property
def part_count(self):

View File

@ -300,9 +300,9 @@ class Part(models.Model):
""" Return the URL of the image for this part """
if self.image:
return os.path.join(settings.MEDIA_URL, str(self.image.url))
return helpers.getMediaUrl(self.image.url)
else:
return os.path.join(settings.STATIC_URL, 'img/blank_image.png')
return helpers.getBlankImage()
def get_thumbnail_url(self):
"""
@ -310,9 +310,9 @@ class Part(models.Model):
"""
if self.image:
return os.path.join(settings.MEDIA_URL, str(self.image.thumbnail.url))
return helpers.getMediaUrl(self.image.thumbnail.url)
else:
return os.path.join(settings.STATIC_URL, 'img/blank_image.thumbnail.png')
return helpers.getBlankThumbnail()
def validate_unique(self, exclude=None):
""" Validate that a part is 'unique'.