Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check if Company thumbnails are created on Company app start

Browse Source
This commit is contained in:
Oliver Walters 2020-04-07 11:23:30 +10:00
parent 1a233e7949
commit e0655f61d8
3 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
InvenTree/company/apps.py
View File

@ -1,7 +1,36 @@
from __future__ import unicode_literals
from django.apps import AppConfig
import os
from django.apps import AppConfig
from django.db.utils import OperationalError, ProgrammingError
from django.conf import settings
class CompanyConfig(AppConfig):
name = 'company'
def ready(self):
"""
This function is called whenever the Company app is loaded.
"""
self.generate_company_thumbs()
def generate_company_thumbs(self):
from .models import Company
print("InvenTree: Checking Company image thumbnails")
try:
for company in Company.objects.all():
if company.image:
url = company.image.thumbnail.name
loc = os.path.join(settings.MEDIA_ROOT, url)
if not os.path.exists(loc):
print("InvenTree: Generating thumbnail for Company '{c}'".format(c=company.name))
company.image.render_variations(replace=False)
except (OperationalError, ProgrammingError):
print("Could not generate Company thumbnails")

2
InvenTree/company/models.py
View File

@ -96,7 +96,7 @@ class Company(models.Model):
upload_to=rename_company_image,
null=True,
blank=True,
variations={'thumnbnail': (128, 128)},
variations={'thumbnail': (128, 128)},
delete_orphans=True,
)

3
InvenTree/part/apps.py
View File

@ -20,13 +20,14 @@ class PartConfig(AppConfig):
def generate_part_thumbnails(self):
from .models import Part
print("Checking Part image thumbnails")
print("InvenTree: Checking Part image thumbnails")
try:
for part in Part.objects.all():
if part.image:
url = part.image.thumbnail.name
loc = os.path.join(settings.MEDIA_ROOT, url)
if not os.path.exists(loc):
print("InvenTree: Generating thumbnail for Part '{p}'".format(p=part.name))
part.image.render_variations(replace=False)