mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import logging
|
|
|
|
from PIL import UnidentifiedImageError
|
|
|
|
from django.apps import AppConfig
|
|
from django.db.utils import OperationalError, ProgrammingError
|
|
from django.conf import settings
|
|
|
|
from InvenTree.ready import canAppAccessDatabase
|
|
|
|
|
|
logger = logging.getLogger("inventree")
|
|
|
|
|
|
class CompanyConfig(AppConfig):
|
|
name = 'company'
|
|
|
|
def ready(self):
|
|
"""
|
|
This function is called whenever the Company app is loaded.
|
|
"""
|
|
|
|
if canAppAccessDatabase():
|
|
self.generate_company_thumbs()
|
|
|
|
def generate_company_thumbs(self):
|
|
|
|
from .models import Company
|
|
|
|
logger.debug("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):
|
|
logger.info("InvenTree: Generating thumbnail for Company '{c}'".format(c=company.name))
|
|
try:
|
|
company.image.render_variations(replace=False)
|
|
except FileNotFoundError:
|
|
logger.warning(f"Image file '{company.image}' missing")
|
|
company.image = None
|
|
company.save()
|
|
except UnidentifiedImageError:
|
|
logger.warning(f"Image file '{company.image}' is invalid")
|
|
except (OperationalError, ProgrammingError):
|
|
# Getting here probably meant the database was in test mode
|
|
pass
|