"""Common database model definitions. These models are 'generic' and do not fit a particular business logic object. """ import base64 import decimal import hashlib import hmac import json import logging import math import os import re import uuid from datetime import datetime, timedelta from enum import Enum from secrets import compare_digest from typing import Any, Callable, Dict, List, Tuple, TypedDict, Union from django.apps import apps from django.conf import settings from django.contrib.auth.models import Group, User from django.contrib.contenttypes.fields import GenericForeignKey from django.contrib.contenttypes.models import ContentType from django.contrib.humanize.templatetags.humanize import naturaltime from django.contrib.sites.models import Site from django.core.cache import cache from django.core.exceptions import AppRegistryNotReady, ValidationError from django.core.validators import (MaxValueValidator, MinValueValidator, URLValidator) from django.db import models, transaction from django.db.utils import IntegrityError, OperationalError from django.urls import reverse from django.utils.timezone import now from django.utils.translation import gettext_lazy as _ from djmoney.contrib.exchange.exceptions import MissingRate from djmoney.contrib.exchange.models import convert_money from djmoney.settings import CURRENCY_CHOICES from rest_framework.exceptions import PermissionDenied import build.validators import InvenTree.fields import InvenTree.helpers import InvenTree.models import InvenTree.ready import InvenTree.tasks import InvenTree.validators import order.validators from plugin import registry logger = logging.getLogger('inventree') class MetaMixin(models.Model): """A base class for InvenTree models to include shared meta fields. Attributes: - updated: The last time this object was updated """ class Meta: """Meta options for MetaMixin.""" abstract = True updated = models.DateTimeField( verbose_name=_('Updated'), help_text=_('Timestamp of last update'), auto_now=True, null=True, ) class EmptyURLValidator(URLValidator): """Validator for filed with url - that can be empty.""" def __call__(self, value): """Make sure empty values pass.""" value = str(value).strip() if len(value) == 0: pass else: super().__call__(value) class ProjectCode(InvenTree.models.MetadataMixin, models.Model): """A ProjectCode is a unique identifier for a project.""" @staticmethod def get_api_url(): """Return the API URL for this model.""" return reverse('api-project-code-list') def __str__(self): """String representation of a ProjectCode.""" return self.code code = models.CharField( max_length=50, unique=True, verbose_name=_('Project Code'), help_text=_('Unique project code'), ) description = models.CharField( max_length=200, blank=True, verbose_name=_('Description'), help_text=_('Project description'), ) class SettingsKeyType(TypedDict, total=False): """Type definitions for a SettingsKeyType Attributes: name: Translatable string name of the setting (required) description: Translatable string description of the setting (required) units: Units of the particular setting (optional) validator: Validation function/list of functions for the setting (optional, default: None, e.g: bool, int, str, MinValueValidator, ...) default: Default value or function that returns default value (optional) choices: (Function that returns) Tuple[str: key, str: display value] (optional) hidden: Hide this setting from settings page (optional) before_save: Function that gets called after save with *args, **kwargs (optional) after_save: Function that gets called after save with *args, **kwargs (optional) protected: Protected values are not returned to the client, instead "***" is returned (optional, default: False) """ name: str description: str units: str validator: Union[Callable, List[Callable], Tuple[Callable]] default: Union[Callable, Any] choices: Union[Tuple[str, str], Callable[[], Tuple[str, str]]] hidden: bool before_save: Callable[..., None] after_save: Callable[..., None] protected: bool class BaseInvenTreeSetting(models.Model): """An base InvenTreeSetting object is a key:value pair used for storing single values (e.g. one-off settings values). Attributes: SETTINGS: definition of all available settings extra_unique_fields: List of extra fields used to be unique, e.g. for PluginConfig -> plugin """ SETTINGS: Dict[str, SettingsKeyType] = {} extra_unique_fields: List[str] = [] class Meta: """Meta options for BaseInvenTreeSetting -> abstract stops creation of database entry.""" abstract = True def save(self, *args, **kwargs): """Enforce validation and clean before saving.""" self.key = str(self.key).upper() do_cache = kwargs.pop('cache', True) self.clean(**kwargs) self.validate_unique() # Execute before_save action self._call_settings_function('before_save', args, kwargs) # Update this setting in the cache if do_cache: self.save_to_cache() super().save() # Execute after_save action self._call_settings_function('after_save', args, kwargs) def _call_settings_function(self, reference: str, args, kwargs): """Call a function associated with a particular setting. Args: reference (str): The name of the function to call args: Positional arguments to pass to the function kwargs: Keyword arguments to pass to the function """ # Get action setting = self.get_setting_definition(self.key, *args, **kwargs) settings_fnc = setting.get(reference, None) # Execute if callable if callable(settings_fnc): settings_fnc(self) @property def cache_key(self): """Generate a unique cache key for this settings object""" return self.__class__.create_cache_key(self.key, **self.get_filters_for_instance()) def save_to_cache(self): """Save this setting object to cache""" ckey = self.cache_key logger.debug(f"Saving setting '{ckey}' to cache") try: cache.set( ckey, self, timeout=3600 ) except TypeError: # Some characters cause issues with caching; ignore and move on pass @classmethod def create_cache_key(cls, setting_key, **kwargs): """Create a unique cache key for a particular setting object. The cache key uses the following elements to ensure the key is 'unique': - The name of the class - The unique KEY string - Any key:value kwargs associated with the particular setting type (e.g. user-id) """ key = f"{str(cls.__name__)}:{setting_key}" for k, v in kwargs.items(): key += f"_{k}:{v}" return key.replace(" ", "") @classmethod def get_filters(cls, **kwargs): """Enable to filter by other kwargs defined in cls.extra_unique_fields""" return {key: value for key, value in kwargs.items() if key in cls.extra_unique_fields} def get_filters_for_instance(self): """Enable to filter by other fields defined in self.extra_unique_fields""" return {key: getattr(self, key, None) for key in self.extra_unique_fields if hasattr(self, key)} @classmethod def allValues(cls, exclude_hidden=False, **kwargs): """Return a dict of "all" defined global settings. This performs a single database lookup, and then any settings which are not *in* the database are assigned their default values """ results = cls.objects.all() if exclude_hidden: # Keys which start with an undersore are used for internal functionality results = results.exclude(key__startswith='_') # Optionally filter by other keys results = results.filter(**cls.get_filters(**kwargs)) # Query the database settings = {} for setting in results: if setting.key: settings[setting.key.upper()] = setting.value # Specify any "default" values which are not in the database for key in cls.SETTINGS.keys(): if key.upper() not in settings: settings[key.upper()] = cls.get_setting_default(key) if exclude_hidden: hidden = cls.SETTINGS[key].get('hidden', False) if hidden: # Remove hidden items del settings[key.upper()] for key, value in settings.items(): validator = cls.get_setting_validator(key) if cls.is_protected(key): value = '***' elif cls.validator_is_bool(validator): value = InvenTree.helpers.str2bool(value) elif cls.validator_is_int(validator): try: value = int(value) except ValueError: value = cls.get_setting_default(key) settings[key] = value return settings @classmethod def get_setting_definition(cls, key, **kwargs): """Return the 'definition' of a particular settings value, as a dict object. - The 'settings' dict can be passed as a kwarg - If not passed, look for cls.SETTINGS - Returns an empty dict if the key is not found """ settings = kwargs.get('settings', cls.SETTINGS) key = str(key).strip().upper() if settings is not None and key in settings: return settings[key] else: return {} @classmethod def get_setting_name(cls, key, **kwargs): """Return the name of a particular setting. If it does not exist, return an empty string. """ setting = cls.get_setting_definition(key, **kwargs) return setting.get('name', '') @classmethod def get_setting_description(cls, key, **kwargs): """Return the description for a particular setting. If it does not exist, return an empty string. """ setting = cls.get_setting_definition(key, **kwargs) return setting.get('description', '') @classmethod def get_setting_units(cls, key, **kwargs): """Return the units for a particular setting. If it does not exist, return an empty string. """ setting = cls.get_setting_definition(key, **kwargs) return setting.get('units', '') @classmethod def get_setting_validator(cls, key, **kwargs): """Return the validator for a particular setting. If it does not exist, return None """ setting = cls.get_setting_definition(key, **kwargs) return setting.get('validator', None) @classmethod def get_setting_default(cls, key, **kwargs): """Return the default value for a particular setting. If it does not exist, return an empty string """ setting = cls.get_setting_definition(key, **kwargs) default = setting.get('default', '') if callable(default): return default() else: return default @classmethod def get_setting_choices(cls, key, **kwargs): """Return the validator choices available for a particular setting.""" setting = cls.get_setting_definition(key, **kwargs) choices = setting.get('choices', None) if callable(choices): # Evaluate the function (we expect it will return a list of tuples...) return choices() return choices @classmethod def get_setting_object(cls, key, **kwargs): """Return an InvenTreeSetting object matching the given key. - Key is case-insensitive - Returns None if no match is made First checks the cache to see if this object has recently been accessed, and returns the cached version if so. """ key = str(key).strip().upper() filters = { 'key__iexact': key, # Optionally filter by other keys **cls.get_filters(**kwargs), } # Perform cache lookup by default do_cache = kwargs.pop('cache', True) ckey = cls.create_cache_key(key, **kwargs) if do_cache: try: # First attempt to find the setting object in the cache cached_setting = cache.get(ckey) if cached_setting is not None: return cached_setting except AppRegistryNotReady: # Cache is not ready yet do_cache = False try: settings = cls.objects.all() setting = settings.filter(**filters).first() except (ValueError, cls.DoesNotExist): setting = None except (IntegrityError, OperationalError): setting = None # Setting does not exist! (Try to create it) if not setting: # Unless otherwise specified, attempt to create the setting create = kwargs.pop('create', True) # Prevent creation of new settings objects when importing data if InvenTree.ready.isImportingData() or not InvenTree.ready.canAppAccessDatabase(allow_test=True, allow_shell=True): create = False if create: # Attempt to create a new settings object default_value = cls.get_setting_default(key, **kwargs) setting = cls( key=key, value=default_value, **kwargs ) try: # Wrap this statement in "atomic", so it can be rolled back if it fails with transaction.atomic(): setting.save(**kwargs) except (IntegrityError, OperationalError): # It might be the case that the database isn't created yet pass except ValidationError: # The setting failed validation - might be due to duplicate keys pass if setting and do_cache: # Cache this setting object setting.save_to_cache() return setting @classmethod def get_setting(cls, key, backup_value=None, **kwargs): """Get the value of a particular setting. If it does not exist, return the backup value (default = None) """ # If no backup value is specified, atttempt to retrieve a "default" value if backup_value is None: backup_value = cls.get_setting_default(key, **kwargs) setting = cls.get_setting_object(key, **kwargs) if setting: value = setting.value # Cast to boolean if necessary if setting.is_bool(): value = InvenTree.helpers.str2bool(value) # Cast to integer if necessary if setting.is_int(): try: value = int(value) except (ValueError, TypeError): value = backup_value else: value = backup_value return value @classmethod def set_setting(cls, key, value, change_user, create=True, **kwargs): """Set the value of a particular setting. If it does not exist, option to create it. Args: key: settings key value: New value change_user: User object (must be staff member to update a core setting) create: If True, create a new setting if the specified key does not exist. """ if change_user is not None and not change_user.is_staff: return filters = { 'key__iexact': key, # Optionally filter by other keys **cls.get_filters(**kwargs), } try: setting = cls.objects.get(**filters) except cls.DoesNotExist: if create: setting = cls(key=key, **kwargs) else: return # Enforce standard boolean representation if setting.is_bool(): value = InvenTree.helpers.str2bool(value) setting.value = str(value) setting.save() key = models.CharField(max_length=50, blank=False, unique=False, help_text=_('Settings key (must be unique - case insensitive)')) value = models.CharField(max_length=200, blank=True, unique=False, help_text=_('Settings value')) @property def name(self): """Return name for setting.""" return self.__class__.get_setting_name(self.key, **self.get_filters_for_instance()) @property def default_value(self): """Return default_value for setting.""" return self.__class__.get_setting_default(self.key, **self.get_filters_for_instance()) @property def description(self): """Return description for setting.""" return self.__class__.get_setting_description(self.key, **self.get_filters_for_instance()) @property def units(self): """Return units for setting.""" return self.__class__.get_setting_units(self.key, **self.get_filters_for_instance()) def clean(self, **kwargs): """If a validator (or multiple validators) are defined for a particular setting key, run them against the 'value' field.""" super().clean() # Encode as native values if self.is_int(): self.value = self.as_int() elif self.is_bool(): self.value = self.as_bool() validator = self.__class__.get_setting_validator(self.key, **kwargs) if validator is not None: self.run_validator(validator) options = self.valid_options() if options and self.value not in options: raise ValidationError(_("Chosen value is not a valid option")) def run_validator(self, validator): """Run a validator against the 'value' field for this InvenTreeSetting object.""" if validator is None: return value = self.value # Boolean validator if validator is bool: # Value must "look like" a boolean value if InvenTree.helpers.is_bool(value): # Coerce into either "True" or "False" value = InvenTree.helpers.str2bool(value) else: raise ValidationError({ 'value': _('Value must be a boolean value') }) # Integer validator if validator is int: try: # Coerce into an integer value value = int(value) except (ValueError, TypeError): raise ValidationError({ 'value': _('Value must be an integer value'), }) # If a list of validators is supplied, iterate through each one if type(validator) in [list, tuple]: for v in validator: self.run_validator(v) if callable(validator): # We can accept function validators with a single argument if self.is_bool(): value = self.as_bool() if self.is_int(): value = self.as_int() validator(value) def validate_unique(self, exclude=None): """Ensure that the key:value pair is unique. In addition to the base validators, this ensures that the 'key' is unique, using a case-insensitive comparison. Note that sub-classes (UserSetting, PluginSetting) use other filters to determine if the setting is 'unique' or not """ super().validate_unique(exclude) filters = { 'key__iexact': self.key, # Optionally filter by other keys **self.get_filters_for_instance(), } try: # Check if a duplicate setting already exists setting = self.__class__.objects.filter(**filters).exclude(id=self.id) if setting.exists(): raise ValidationError({'key': _('Key string must be unique')}) except self.DoesNotExist: pass def choices(self): """Return the available choices for this setting (or None if no choices are defined).""" return self.__class__.get_setting_choices(self.key, **self.get_filters_for_instance()) def valid_options(self): """Return a list of valid options for this setting.""" choices = self.choices() if not choices: return None return [opt[0] for opt in choices] def is_choice(self): """Check if this setting is a "choice" field.""" return self.__class__.get_setting_choices(self.key, **self.get_filters_for_instance()) is not None def as_choice(self): """Render this setting as the "display" value of a choice field. E.g. if the choices are: [('A4', 'A4 paper'), ('A3', 'A3 paper')], and the value is 'A4', then display 'A4 paper' """ choices = self.get_setting_choices(self.key, **self.get_filters_for_instance()) if not choices: return self.value for value, display in choices: if value == self.value: return display return self.value def is_model(self): """Check if this setting references a model instance in the database.""" return self.model_name() is not None def model_name(self): """Return the model name associated with this setting.""" setting = self.get_setting_definition(self.key, **self.get_filters_for_instance()) return setting.get('model', None) def model_class(self): """Return the model class associated with this setting. If (and only if): - It has a defined 'model' parameter - The 'model' parameter is of the form app.model - The 'model' parameter has matches a known app model """ model_name = self.model_name() if not model_name: return None try: (app, mdl) = model_name.strip().split('.') except ValueError: logger.error(f"Invalid 'model' parameter for setting {self.key} : '{model_name}'") return None app_models = apps.all_models.get(app, None) if app_models is None: logger.error(f"Error retrieving model class '{model_name}' for setting '{self.key}' - no app named '{app}'") return None model = app_models.get(mdl, None) if model is None: logger.error(f"Error retrieving model class '{model_name}' for setting '{self.key}' - no model named '{mdl}'") return None # Looks like we have found a model! return model def api_url(self): """Return the API url associated with the linked model, if provided, and valid!""" model_class = self.model_class() if model_class: # If a valid class has been found, see if it has registered an API URL try: return model_class.get_api_url() except Exception: pass # Some other model types are hard-coded hardcoded_models = { 'auth.user': 'api-user-list', 'auth.group': 'api-group-list', } model_table = f'{model_class._meta.app_label}.{model_class._meta.model_name}' if url := hardcoded_models[model_table]: return reverse(url) return None def is_bool(self): """Check if this setting is required to be a boolean value.""" validator = self.__class__.get_setting_validator(self.key, **self.get_filters_for_instance()) return self.__class__.validator_is_bool(validator) def as_bool(self): """Return the value of this setting converted to a boolean value. Warning: Only use on values where is_bool evaluates to true! """ return InvenTree.helpers.str2bool(self.value) def setting_type(self): """Return the field type identifier for this setting object.""" if self.is_bool(): return 'boolean' elif self.is_int(): return 'integer' elif self.is_model(): return 'related field' else: return 'string' @classmethod def validator_is_bool(cls, validator): """Return if validator is for bool.""" if validator == bool: return True if type(validator) in [list, tuple]: for v in validator: if v == bool: return True return False def is_int(self,): """Check if the setting is required to be an integer value.""" validator = self.__class__.get_setting_validator(self.key, **self.get_filters_for_instance()) return self.__class__.validator_is_int(validator) @classmethod def validator_is_int(cls, validator): """Return if validator is for int.""" if validator == int: return True if type(validator) in [list, tuple]: for v in validator: if v == int: return True return False def as_int(self): """Return the value of this setting converted to a boolean value. If an error occurs, return the default value """ try: value = int(self.value) except (ValueError, TypeError): value = self.default_value return value @classmethod def is_protected(cls, key, **kwargs): """Check if the setting value is protected.""" setting = cls.get_setting_definition(key, **kwargs) return setting.get('protected', False) @property def protected(self): """Returns if setting is protected from rendering.""" return self.__class__.is_protected(self.key, **self.get_filters_for_instance()) def settings_group_options(): """Build up group tuple for settings based on your choices.""" return [('', _('No group')), *[(str(a.id), str(a)) for a in Group.objects.all()]] def update_instance_url(setting): """Update the first site objects domain to url.""" site_obj = Site.objects.all().order_by('id').first() site_obj.domain = setting.value site_obj.save() def update_instance_name(setting): """Update the first site objects name to instance name.""" site_obj = Site.objects.all().order_by('id').first() site_obj.name = setting.value site_obj.save() def validate_email_domains(setting): """Validate the email domains setting.""" if not setting.value: return domains = setting.value.split(',') for domain in domains: if not domain: raise ValidationError(_('An empty domain is not allowed.')) if not re.match(r'^@[a-zA-Z0-9\.\-_]+$', domain): raise ValidationError(_(f'Invalid domain name: {domain}')) def update_exchange_rates(setting): """Update exchange rates when base currency is changed""" if InvenTree.ready.isImportingData(): return if not InvenTree.ready.canAppAccessDatabase(): return InvenTree.tasks.update_exchange_rates() class InvenTreeSetting(BaseInvenTreeSetting): """An InvenTreeSetting object is a key:value pair used for storing single values (e.g. one-off settings values). The class provides a way of retrieving the value for a particular key, even if that key does not exist. """ class Meta: """Meta options for InvenTreeSetting.""" verbose_name = "InvenTree Setting" verbose_name_plural = "InvenTree Settings" def save(self, *args, **kwargs): """When saving a global setting, check to see if it requires a server restart. If so, set the "SERVER_RESTART_REQUIRED" setting to True """ super().save() if self.requires_restart() and not InvenTree.ready.isImportingData(): InvenTreeSetting.set_setting('SERVER_RESTART_REQUIRED', True, None) """ Dict of all global settings values: The key of each item is the name of the value as it appears in the database. Each global setting has the following parameters: - name: Translatable string name of the setting (required) - description: Translatable string description of the setting (required) - default: Default value (optional) - units: Units of the particular setting (optional) - validator: Validation function for the setting (optional) The keys must be upper-case """ SETTINGS = { 'SERVER_RESTART_REQUIRED': { 'name': _('Restart required'), 'description': _('A setting has been changed which requires a server restart'), 'default': False, 'validator': bool, 'hidden': True, }, 'INVENTREE_INSTANCE': { 'name': _('Server Instance Name'), 'default': 'InvenTree', 'description': _('String descriptor for the server instance'), 'after_save': update_instance_name, }, 'INVENTREE_INSTANCE_TITLE': { 'name': _('Use instance name'), 'description': _('Use the instance name in the title-bar'), 'validator': bool, 'default': False, }, 'INVENTREE_RESTRICT_ABOUT': { 'name': _('Restrict showing `about`'), 'description': _('Show the `about` modal only to superusers'), 'validator': bool, 'default': False, }, 'INVENTREE_COMPANY_NAME': { 'name': _('Company name'), 'description': _('Internal company name'), 'default': 'My company name', }, 'INVENTREE_BASE_URL': { 'name': _('Base URL'), 'description': _('Base URL for server instance'), 'validator': EmptyURLValidator(), 'default': '', 'after_save': update_instance_url, }, 'INVENTREE_DEFAULT_CURRENCY': { 'name': _('Default Currency'), 'description': _('Select base currency for pricing calculations'), 'default': 'USD', 'choices': CURRENCY_CHOICES, 'after_save': update_exchange_rates, }, 'INVENTREE_DOWNLOAD_FROM_URL': { 'name': _('Download from URL'), 'description': _('Allow download of remote images and files from external URL'), 'validator': bool, 'default': False, }, 'INVENTREE_DOWNLOAD_IMAGE_MAX_SIZE': { 'name': _('Download Size Limit'), 'description': _('Maximum allowable download size for remote image'), 'units': 'MB', 'default': 1, 'validator': [ int, MinValueValidator(1), MaxValueValidator(25), ] }, 'INVENTREE_DOWNLOAD_FROM_URL_USER_AGENT': { 'name': _('User-agent used to download from URL'), 'description': _('Allow to override the user-agent used to download images and files from external URL (leave blank for the default)'), 'default': '', }, 'INVENTREE_REQUIRE_CONFIRM': { 'name': _('Require confirm'), 'description': _('Require explicit user confirmation for certain action.'), 'validator': bool, 'default': True, }, 'INVENTREE_TREE_DEPTH': { 'name': _('Tree Depth'), 'description': _('Default tree depth for treeview. Deeper levels can be lazy loaded as they are needed.'), 'default': 1, 'validator': [ int, MinValueValidator(0), ] }, 'INVENTREE_UPDATE_CHECK_INTERVAL': { 'name': _('Update Check Interval'), 'description': _('How often to check for updates (set to zero to disable)'), 'validator': [ int, MinValueValidator(0), ], 'default': 7, 'units': _('days'), }, 'INVENTREE_BACKUP_ENABLE': { 'name': _('Automatic Backup'), 'description': _('Enable automatic backup of database and media files'), 'validator': bool, 'default': False, }, 'INVENTREE_BACKUP_DAYS': { 'name': _('Auto Backup Interval'), 'description': _('Specify number of days between automated backup events'), 'validator': [ int, MinValueValidator(1), ], 'default': 1, 'units': _('days'), }, 'INVENTREE_DELETE_TASKS_DAYS': { 'name': _('Task Deletion Interval'), 'description': _('Background task results will be deleted after specified number of days'), 'default': 30, 'units': _('days'), 'validator': [ int, MinValueValidator(7), ] }, 'INVENTREE_DELETE_ERRORS_DAYS': { 'name': _('Error Log Deletion Interval'), 'description': _('Error logs will be deleted after specified number of days'), 'default': 30, 'units': _('days'), 'validator': [ int, MinValueValidator(7) ] }, 'INVENTREE_DELETE_NOTIFICATIONS_DAYS': { 'name': _('Notification Deletion Interval'), 'description': _('User notifications will be deleted after specified number of days'), 'default': 30, 'units': _('days'), 'validator': [ int, MinValueValidator(7), ] }, 'BARCODE_ENABLE': { 'name': _('Barcode Support'), 'description': _('Enable barcode scanner support'), 'default': True, 'validator': bool, }, 'BARCODE_INPUT_DELAY': { 'name': _('Barcode Input Delay'), 'description': _('Barcode input processing delay time'), 'default': 50, 'validator': [ int, MinValueValidator(1), ], 'units': 'ms', }, 'BARCODE_WEBCAM_SUPPORT': { 'name': _('Barcode Webcam Support'), 'description': _('Allow barcode scanning via webcam in browser'), 'default': True, 'validator': bool, }, 'PART_ENABLE_REVISION': { 'name': _('Part Revisions'), 'description': _('Enable revision field for Part'), 'validator': bool, 'default': True, }, 'PART_IPN_REGEX': { 'name': _('IPN Regex'), 'description': _('Regular expression pattern for matching Part IPN') }, 'PART_ALLOW_DUPLICATE_IPN': { 'name': _('Allow Duplicate IPN'), 'description': _('Allow multiple parts to share the same IPN'), 'default': True, 'validator': bool, }, 'PART_ALLOW_EDIT_IPN': { 'name': _('Allow Editing IPN'), 'description': _('Allow changing the IPN value while editing a part'), 'default': True, 'validator': bool, }, 'PART_COPY_BOM': { 'name': _('Copy Part BOM Data'), 'description': _('Copy BOM data by default when duplicating a part'), 'default': True, 'validator': bool, }, 'PART_COPY_PARAMETERS': { 'name': _('Copy Part Parameter Data'), 'description': _('Copy parameter data by default when duplicating a part'), 'default': True, 'validator': bool, }, 'PART_COPY_TESTS': { 'name': _('Copy Part Test Data'), 'description': _('Copy test data by default when duplicating a part'), 'default': True, 'validator': bool }, 'PART_CATEGORY_PARAMETERS': { 'name': _('Copy Category Parameter Templates'), 'description': _('Copy category parameter templates when creating a part'), 'default': True, 'validator': bool }, 'PART_TEMPLATE': { 'name': _('Template'), 'description': _('Parts are templates by default'), 'default': False, 'validator': bool, }, 'PART_ASSEMBLY': { 'name': _('Assembly'), 'description': _('Parts can be assembled from other components by default'), 'default': False, 'validator': bool, }, 'PART_COMPONENT': { 'name': _('Component'), 'description': _('Parts can be used as sub-components by default'), 'default': True, 'validator': bool, }, 'PART_PURCHASEABLE': { 'name': _('Purchaseable'), 'description': _('Parts are purchaseable by default'), 'default': True, 'validator': bool, }, 'PART_SALABLE': { 'name': _('Salable'), 'description': _('Parts are salable by default'), 'default': False, 'validator': bool, }, 'PART_TRACKABLE': { 'name': _('Trackable'), 'description': _('Parts are trackable by default'), 'default': False, 'validator': bool, }, 'PART_VIRTUAL': { 'name': _('Virtual'), 'description': _('Parts are virtual by default'), 'default': False, 'validator': bool, }, 'PART_SHOW_IMPORT': { 'name': _('Show Import in Views'), 'description': _('Display the import wizard in some part views'), 'default': False, 'validator': bool, }, 'PART_SHOW_RELATED': { 'name': _('Show related parts'), 'description': _('Display related parts for a part'), 'default': True, 'validator': bool, }, 'PART_CREATE_INITIAL': { 'name': _('Initial Stock Data'), 'description': _('Allow creation of initial stock when adding a new part'), 'default': False, 'validator': bool, }, 'PART_CREATE_SUPPLIER': { 'name': _('Initial Supplier Data'), 'description': _('Allow creation of initial supplier data when adding a new part'), 'default': True, 'validator': bool, }, 'PART_NAME_FORMAT': { 'name': _('Part Name Display Format'), 'description': _('Format to display the part name'), 'default': "{{ part.IPN if part.IPN }}{{ ' | ' if part.IPN }}{{ part.name }}{{ ' | ' if part.revision }}" "{{ part.revision if part.revision }}", 'validator': InvenTree.validators.validate_part_name_format }, 'PART_CATEGORY_DEFAULT_ICON': { 'name': _('Part Category Default Icon'), 'description': _('Part category default icon (empty means no icon)'), 'default': '', }, 'PRICING_DECIMAL_PLACES_MIN': { 'name': _('Minimum Pricing Decimal Places'), 'description': _('Minimum number of decimal places to display when rendering pricing data'), 'default': 0, 'validator': [ int, MinValueValidator(0), MaxValueValidator(4), ] }, 'PRICING_DECIMAL_PLACES': { 'name': _('Maximum Pricing Decimal Places'), 'description': _('Maximum number of decimal places to display when rendering pricing data'), 'default': 6, 'validator': [ int, MinValueValidator(2), MaxValueValidator(6) ] }, 'PRICING_USE_SUPPLIER_PRICING': { 'name': _('Use Supplier Pricing'), 'description': _('Include supplier price breaks in overall pricing calculations'), 'default': True, 'validator': bool, }, 'PRICING_PURCHASE_HISTORY_OVERRIDES_SUPPLIER': { 'name': _('Purchase History Override'), 'description': _('Historical purchase order pricing overrides supplier price breaks'), 'default': False, 'validator': bool, }, 'PRICING_USE_STOCK_PRICING': { 'name': _('Use Stock Item Pricing'), 'description': _('Use pricing from manually entered stock data for pricing calculations'), 'default': True, 'validator': bool, }, 'PRICING_STOCK_ITEM_AGE_DAYS': { 'name': _('Stock Item Pricing Age'), 'description': _('Exclude stock items older than this number of days from pricing calculations'), 'default': 0, 'units': _('days'), 'validator': [ int, MinValueValidator(0), ] }, 'PRICING_USE_VARIANT_PRICING': { 'name': _('Use Variant Pricing'), 'description': _('Include variant pricing in overall pricing calculations'), 'default': True, 'validator': bool, }, 'PRICING_ACTIVE_VARIANTS': { 'name': _('Active Variants Only'), 'description': _('Only use active variant parts for calculating variant pricing'), 'default': False, 'validator': bool, }, 'PRICING_UPDATE_DAYS': { 'name': _('Pricing Rebuild Interval'), 'description': _('Number of days before part pricing is automatically updated'), 'units': _('days'), 'default': 30, 'validator': [ int, MinValueValidator(10), ] }, 'PART_INTERNAL_PRICE': { 'name': _('Internal Prices'), 'description': _('Enable internal prices for parts'), 'default': False, 'validator': bool }, 'PART_BOM_USE_INTERNAL_PRICE': { 'name': _('Internal Price Override'), 'description': _('If available, internal prices override price range calculations'), 'default': False, 'validator': bool }, 'LABEL_ENABLE': { 'name': _('Enable label printing'), 'description': _('Enable label printing from the web interface'), 'default': True, 'validator': bool, }, 'LABEL_DPI': { 'name': _('Label Image DPI'), 'description': _('DPI resolution when generating image files to supply to label printing plugins'), 'default': 300, 'validator': [ int, MinValueValidator(100), ] }, 'REPORT_ENABLE': { 'name': _('Enable Reports'), 'description': _('Enable generation of reports'), 'default': False, 'validator': bool, }, 'REPORT_DEBUG_MODE': { 'name': _('Debug Mode'), 'description': _('Generate reports in debug mode (HTML output)'), 'default': False, 'validator': bool, }, 'REPORT_DEFAULT_PAGE_SIZE': { 'name': _('Page Size'), 'description': _('Default page size for PDF reports'), 'default': 'A4', 'choices': [ ('A4', 'A4'), ('Legal', 'Legal'), ('Letter', 'Letter') ], }, 'REPORT_ENABLE_TEST_REPORT': { 'name': _('Enable Test Reports'), 'description': _('Enable generation of test reports'), 'default': True, 'validator': bool, }, 'REPORT_ATTACH_TEST_REPORT': { 'name': _('Attach Test Reports'), 'description': _('When printing a Test Report, attach a copy of the Test Report to the associated Stock Item'), 'default': False, 'validator': bool, }, 'SERIAL_NUMBER_GLOBALLY_UNIQUE': { 'name': _('Globally Unique Serials'), 'description': _('Serial numbers for stock items must be globally unique'), 'default': False, 'validator': bool, }, 'SERIAL_NUMBER_AUTOFILL': { 'name': _('Autofill Serial Numbers'), 'description': _('Autofill serial numbers in forms'), 'default': False, 'validator': bool, }, 'STOCK_DELETE_DEPLETED_DEFAULT': { 'name': _('Delete Depleted Stock'), 'description': _('Determines default behaviour when a stock item is depleted'), 'default': True, 'validator': bool, }, 'STOCK_BATCH_CODE_TEMPLATE': { 'name': _('Batch Code Template'), 'description': _('Template for generating default batch codes for stock items'), 'default': '', }, 'STOCK_ENABLE_EXPIRY': { 'name': _('Stock Expiry'), 'description': _('Enable stock expiry functionality'), 'default': False, 'validator': bool, }, 'STOCK_ALLOW_EXPIRED_SALE': { 'name': _('Sell Expired Stock'), 'description': _('Allow sale of expired stock'), 'default': False, 'validator': bool, }, 'STOCK_STALE_DAYS': { 'name': _('Stock Stale Time'), 'description': _('Number of days stock items are considered stale before expiring'), 'default': 0, 'units': _('days'), 'validator': [int], }, 'STOCK_ALLOW_EXPIRED_BUILD': { 'name': _('Build Expired Stock'), 'description': _('Allow building with expired stock'), 'default': False, 'validator': bool, }, 'STOCK_OWNERSHIP_CONTROL': { 'name': _('Stock Ownership Control'), 'description': _('Enable ownership control over stock locations and items'), 'default': False, 'validator': bool, }, 'STOCK_LOCATION_DEFAULT_ICON': { 'name': _('Stock Location Default Icon'), 'description': _('Stock location default icon (empty means no icon)'), 'default': '', }, 'BUILDORDER_REFERENCE_PATTERN': { 'name': _('Build Order Reference Pattern'), 'description': _('Required pattern for generating Build Order reference field'), 'default': 'BO-{ref:04d}', 'validator': build.validators.validate_build_order_reference_pattern, }, 'RETURNORDER_ENABLED': { 'name': _('Enable Return Orders'), 'description': _('Enable return order functionality in the user interface'), 'validator': bool, 'default': False, }, 'RETURNORDER_REFERENCE_PATTERN': { 'name': _('Return Order Reference Pattern'), 'description': _('Required pattern for generating Return Order reference field'), 'default': 'RMA-{ref:04d}', 'validator': order.validators.validate_return_order_reference_pattern, }, 'RETURNORDER_EDIT_COMPLETED_ORDERS': { 'name': _('Edit Completed Return Orders'), 'description': _('Allow editing of return orders after they have been completed'), 'default': False, 'validator': bool, }, 'SALESORDER_REFERENCE_PATTERN': { 'name': _('Sales Order Reference Pattern'), 'description': _('Required pattern for generating Sales Order reference field'), 'default': 'SO-{ref:04d}', 'validator': order.validators.validate_sales_order_reference_pattern, }, 'SALESORDER_DEFAULT_SHIPMENT': { 'name': _('Sales Order Default Shipment'), 'description': _('Enable creation of default shipment with sales orders'), 'default': False, 'validator': bool, }, 'SALESORDER_EDIT_COMPLETED_ORDERS': { 'name': _('Edit Completed Sales Orders'), 'description': _('Allow editing of sales orders after they have been shipped or completed'), 'default': False, 'validator': bool, }, 'PURCHASEORDER_REFERENCE_PATTERN': { 'name': _('Purchase Order Reference Pattern'), 'description': _('Required pattern for generating Purchase Order reference field'), 'default': 'PO-{ref:04d}', 'validator': order.validators.validate_purchase_order_reference_pattern, }, 'PURCHASEORDER_EDIT_COMPLETED_ORDERS': { 'name': _('Edit Completed Purchase Orders'), 'description': _('Allow editing of purchase orders after they have been shipped or completed'), 'default': False, 'validator': bool, }, # login / SSO 'LOGIN_ENABLE_PWD_FORGOT': { 'name': _('Enable password forgot'), 'description': _('Enable password forgot function on the login pages'), 'default': True, 'validator': bool, }, 'LOGIN_ENABLE_REG': { 'name': _('Enable registration'), 'description': _('Enable self-registration for users on the login pages'), 'default': False, 'validator': bool, }, 'LOGIN_ENABLE_SSO': { 'name': _('Enable SSO'), 'description': _('Enable SSO on the login pages'), 'default': False, 'validator': bool, }, 'LOGIN_ENABLE_SSO_REG': { 'name': _('Enable SSO registration'), 'description': _('Enable self-registration via SSO for users on the login pages'), 'default': False, 'validator': bool, }, 'LOGIN_MAIL_REQUIRED': { 'name': _('Email required'), 'description': _('Require user to supply mail on signup'), 'default': False, 'validator': bool, }, 'LOGIN_SIGNUP_SSO_AUTO': { 'name': _('Auto-fill SSO users'), 'description': _('Automatically fill out user-details from SSO account-data'), 'default': True, 'validator': bool, }, 'LOGIN_SIGNUP_MAIL_TWICE': { 'name': _('Mail twice'), 'description': _('On signup ask users twice for their mail'), 'default': False, 'validator': bool, }, 'LOGIN_SIGNUP_PWD_TWICE': { 'name': _('Password twice'), 'description': _('On signup ask users twice for their password'), 'default': True, 'validator': bool, }, 'LOGIN_SIGNUP_MAIL_RESTRICTION': { 'name': _('Allowed domains'), 'description': _('Restrict signup to certain domains (comma-separated, starting with @)'), 'default': '', 'before_save': validate_email_domains, }, 'SIGNUP_GROUP': { 'name': _('Group on signup'), 'description': _('Group to which new users are assigned on registration'), 'default': '', 'choices': settings_group_options }, 'LOGIN_ENFORCE_MFA': { 'name': _('Enforce MFA'), 'description': _('Users must use multifactor security.'), 'default': False, 'validator': bool, }, 'PLUGIN_ON_STARTUP': { 'name': _('Check plugins on startup'), 'description': _('Check that all plugins are installed on startup - enable in container environments'), 'default': str(os.getenv('INVENTREE_DOCKER', False)).lower() in ['1', 'true'], 'validator': bool, 'requires_restart': True, }, 'PLUGIN_CHECK_SIGNATURES': { 'name': _('Check plugin signatures'), 'description': _('Check and show signatures for plugins'), 'default': False, 'validator': bool, }, # Settings for plugin mixin features 'ENABLE_PLUGINS_URL': { 'name': _('Enable URL integration'), 'description': _('Enable plugins to add URL routes'), 'default': False, 'validator': bool, 'requires_restart': True, }, 'ENABLE_PLUGINS_NAVIGATION': { 'name': _('Enable navigation integration'), 'description': _('Enable plugins to integrate into navigation'), 'default': False, 'validator': bool, 'requires_restart': True, }, 'ENABLE_PLUGINS_APP': { 'name': _('Enable app integration'), 'description': _('Enable plugins to add apps'), 'default': False, 'validator': bool, 'requires_restart': True, }, 'ENABLE_PLUGINS_SCHEDULE': { 'name': _('Enable schedule integration'), 'description': _('Enable plugins to run scheduled tasks'), 'default': False, 'validator': bool, 'requires_restart': True, }, 'ENABLE_PLUGINS_EVENTS': { 'name': _('Enable event integration'), 'description': _('Enable plugins to respond to internal events'), 'default': False, 'validator': bool, 'requires_restart': True, }, "PROJECT_CODES_ENABLED": { 'name': _('Enable project codes'), 'description': _('Enable project codes for tracking projects'), 'default': False, 'validator': bool, }, 'STOCKTAKE_ENABLE': { 'name': _('Stocktake Functionality'), 'description': _('Enable stocktake functionality for recording stock levels and calculating stock value'), 'validator': bool, 'default': False, }, 'STOCKTAKE_AUTO_DAYS': { 'name': _('Automatic Stocktake Period'), 'description': _('Number of days between automatic stocktake recording (set to zero to disable)'), 'validator': [ int, MinValueValidator(0), ], 'default': 0, }, 'STOCKTAKE_DELETE_REPORT_DAYS': { 'name': _('Report Deletion Interval'), 'description': _('Stocktake reports will be deleted after specified number of days'), 'default': 30, 'units': _('days'), 'validator': [ int, MinValueValidator(7), ] }, } typ = 'inventree' key = models.CharField( max_length=50, blank=False, unique=True, help_text=_('Settings key (must be unique - case insensitive'), ) def to_native_value(self): """Return the "pythonic" value, e.g. convert "True" to True, and "1" to 1.""" return self.__class__.get_setting(self.key) def requires_restart(self): """Return True if this setting requires a server restart after changing.""" options = InvenTreeSetting.SETTINGS.get(self.key, None) if options: return options.get('requires_restart', False) else: return False def label_printer_options(): """Build a list of available label printer options.""" printers = [('', _('No Printer (Export to PDF)'))] label_printer_plugins = registry.with_mixin('labels') if label_printer_plugins: printers.extend([(p.slug, p.name + ' - ' + p.human_name) for p in label_printer_plugins]) return printers class InvenTreeUserSetting(BaseInvenTreeSetting): """An InvenTreeSetting object with a usercontext.""" class Meta: """Meta options for InvenTreeUserSetting.""" verbose_name = "InvenTree User Setting" verbose_name_plural = "InvenTree User Settings" constraints = [ models.UniqueConstraint(fields=['key', 'user'], name='unique key and user') ] SETTINGS = { 'HOMEPAGE_HIDE_INACTIVE': { 'name': _('Hide inactive parts'), 'description': _('Hide inactive parts in results displayed on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_PART_STARRED': { 'name': _('Show subscribed parts'), 'description': _('Show subscribed parts on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_CATEGORY_STARRED': { 'name': _('Show subscribed categories'), 'description': _('Show subscribed part categories on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_PART_LATEST': { 'name': _('Show latest parts'), 'description': _('Show latest parts on the homepage'), 'default': True, 'validator': bool, }, 'PART_RECENT_COUNT': { 'name': _('Recent Part Count'), 'description': _('Number of recent parts to display on index page'), 'default': 10, 'validator': [int, MinValueValidator(1)] }, 'HOMEPAGE_BOM_REQUIRES_VALIDATION': { 'name': _('Show unvalidated BOMs'), 'description': _('Show BOMs that await validation on the homepage'), 'default': False, 'validator': bool, }, 'HOMEPAGE_STOCK_RECENT': { 'name': _('Show recent stock changes'), 'description': _('Show recently changed stock items on the homepage'), 'default': True, 'validator': bool, }, 'STOCK_RECENT_COUNT': { 'name': _('Recent Stock Count'), 'description': _('Number of recent stock items to display on index page'), 'default': 10, 'validator': [int, MinValueValidator(1)] }, 'HOMEPAGE_STOCK_LOW': { 'name': _('Show low stock'), 'description': _('Show low stock items on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_SHOW_STOCK_DEPLETED': { 'name': _('Show depleted stock'), 'description': _('Show depleted stock items on the homepage'), 'default': False, 'validator': bool, }, 'HOMEPAGE_BUILD_STOCK_NEEDED': { 'name': _('Show needed stock'), 'description': _('Show stock items needed for builds on the homepage'), 'default': False, 'validator': bool, }, 'HOMEPAGE_STOCK_EXPIRED': { 'name': _('Show expired stock'), 'description': _('Show expired stock items on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_STOCK_STALE': { 'name': _('Show stale stock'), 'description': _('Show stale stock items on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_BUILD_PENDING': { 'name': _('Show pending builds'), 'description': _('Show pending builds on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_BUILD_OVERDUE': { 'name': _('Show overdue builds'), 'description': _('Show overdue builds on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_PO_OUTSTANDING': { 'name': _('Show outstanding POs'), 'description': _('Show outstanding POs on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_PO_OVERDUE': { 'name': _('Show overdue POs'), 'description': _('Show overdue POs on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_SO_OUTSTANDING': { 'name': _('Show outstanding SOs'), 'description': _('Show outstanding SOs on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_SO_OVERDUE': { 'name': _('Show overdue SOs'), 'description': _('Show overdue SOs on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_SO_SHIPMENTS_PENDING': { 'name': _('Show pending SO shipments'), 'description': _('Show pending SO shipments on the homepage'), 'default': True, 'validator': bool, }, 'HOMEPAGE_NEWS': { 'name': _('Show News'), 'description': _('Show news on the homepage'), 'default': False, 'validator': bool, }, "LABEL_INLINE": { 'name': _('Inline label display'), 'description': _('Display PDF labels in the browser, instead of downloading as a file'), 'default': True, 'validator': bool, }, "LABEL_DEFAULT_PRINTER": { 'name': _('Default label printer'), 'description': _('Configure which label printer should be selected by default'), 'default': '', 'choices': label_printer_options }, "REPORT_INLINE": { 'name': _('Inline report display'), 'description': _('Display PDF reports in the browser, instead of downloading as a file'), 'default': False, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_PARTS': { 'name': _('Search Parts'), 'description': _('Display parts in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_SUPPLIER_PARTS': { 'name': _('Search Supplier Parts'), 'description': _('Display supplier parts in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_MANUFACTURER_PARTS': { 'name': _('Search Manufacturer Parts'), 'description': _('Display manufacturer parts in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_HIDE_INACTIVE_PARTS': { 'name': _("Hide Inactive Parts"), 'description': _('Excluded inactive parts from search preview window'), 'default': False, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_CATEGORIES': { 'name': _('Search Categories'), 'description': _('Display part categories in search preview window'), 'default': False, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_STOCK': { 'name': _('Search Stock'), 'description': _('Display stock items in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_HIDE_UNAVAILABLE_STOCK': { 'name': _('Hide Unavailable Stock Items'), 'description': _('Exclude stock items which are not available from the search preview window'), 'validator': bool, 'default': False, }, 'SEARCH_PREVIEW_SHOW_LOCATIONS': { 'name': _('Search Locations'), 'description': _('Display stock locations in search preview window'), 'default': False, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_COMPANIES': { 'name': _('Search Companies'), 'description': _('Display companies in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_BUILD_ORDERS': { 'name': _('Search Build Orders'), 'description': _('Display build orders in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_PURCHASE_ORDERS': { 'name': _('Search Purchase Orders'), 'description': _('Display purchase orders in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_EXCLUDE_INACTIVE_PURCHASE_ORDERS': { 'name': _('Exclude Inactive Purchase Orders'), 'description': _('Exclude inactive purchase orders from search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_SHOW_SALES_ORDERS': { 'name': _('Search Sales Orders'), 'description': _('Display sales orders in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_EXCLUDE_INACTIVE_SALES_ORDERS': { 'name': _('Exclude Inactive Sales Orders'), 'description': _('Exclude inactive sales orders from search preview window'), 'validator': bool, 'default': True, }, 'SEARCH_PREVIEW_SHOW_RETURN_ORDERS': { 'name': _('Search Return Orders'), 'description': _('Display return orders in search preview window'), 'default': True, 'validator': bool, }, 'SEARCH_PREVIEW_EXCLUDE_INACTIVE_RETURN_ORDERS': { 'name': _('Exclude Inactive Return Orders'), 'description': _('Exclude inactive return orders from search preview window'), 'validator': bool, 'default': True, }, 'SEARCH_PREVIEW_RESULTS': { 'name': _('Search Preview Results'), 'description': _('Number of results to show in each section of the search preview window'), 'default': 10, 'validator': [int, MinValueValidator(1)] }, 'SEARCH_REGEX': { 'name': _('Regex Search'), 'description': _('Enable regular expressions in search queries'), 'default': False, 'validator': bool, }, 'SEARCH_WHOLE': { 'name': _('Whole Word Search'), 'description': _('Search queries return results for whole word matches'), 'default': False, 'validator': bool, }, 'PART_SHOW_QUANTITY_IN_FORMS': { 'name': _('Show Quantity in Forms'), 'description': _('Display available part quantity in some forms'), 'default': True, 'validator': bool, }, 'FORMS_CLOSE_USING_ESCAPE': { 'name': _('Escape Key Closes Forms'), 'description': _('Use the escape key to close modal forms'), 'default': False, 'validator': bool, }, 'STICKY_HEADER': { 'name': _('Fixed Navbar'), 'description': _('The navbar position is fixed to the top of the screen'), 'default': False, 'validator': bool, }, 'DATE_DISPLAY_FORMAT': { 'name': _('Date Format'), 'description': _('Preferred format for displaying dates'), 'default': 'YYYY-MM-DD', 'choices': [ ('YYYY-MM-DD', '2022-02-22'), ('YYYY/MM/DD', '2022/22/22'), ('DD-MM-YYYY', '22-02-2022'), ('DD/MM/YYYY', '22/02/2022'), ('MM-DD-YYYY', '02-22-2022'), ('MM/DD/YYYY', '02/22/2022'), ('MMM DD YYYY', 'Feb 22 2022'), ] }, 'DISPLAY_SCHEDULE_TAB': { 'name': _('Part Scheduling'), 'description': _('Display part scheduling information'), 'default': True, 'validator': bool, }, 'DISPLAY_STOCKTAKE_TAB': { 'name': _('Part Stocktake'), 'description': _('Display part stocktake information (if stocktake functionality is enabled)'), 'default': True, 'validator': bool, }, 'TABLE_STRING_MAX_LENGTH': { 'name': _('Table String Length'), 'description': _('Maximimum length limit for strings displayed in table views'), 'validator': [ int, MinValueValidator(0), ], 'default': 100, } } typ = 'user' extra_unique_fields = ['user'] key = models.CharField( max_length=50, blank=False, unique=False, help_text=_('Settings key (must be unique - case insensitive'), ) user = models.ForeignKey( User, on_delete=models.CASCADE, blank=True, null=True, verbose_name=_('User'), help_text=_('User'), ) def to_native_value(self): """Return the "pythonic" value, e.g. convert "True" to True, and "1" to 1.""" return self.__class__.get_setting(self.key, user=self.user) class PriceBreak(MetaMixin): """Represents a PriceBreak model.""" class Meta: """Define this as abstract -> no DB entry is created.""" abstract = True quantity = InvenTree.fields.RoundingDecimalField( max_digits=15, decimal_places=5, default=1, validators=[MinValueValidator(1)], verbose_name=_('Quantity'), help_text=_('Price break quantity'), ) price = InvenTree.fields.InvenTreeModelMoneyField( max_digits=19, decimal_places=6, null=True, verbose_name=_('Price'), help_text=_('Unit price at specified quantity'), ) def convert_to(self, currency_code): """Convert the unit-price at this price break to the specified currency code. Args: currency_code: The currency code to convert to (e.g "USD" or "AUD") """ try: converted = convert_money(self.price, currency_code) except MissingRate: logger.warning(f"No currency conversion rate available for {self.price_currency} -> {currency_code}") return self.price.amount return converted.amount def get_price(instance, quantity, moq=True, multiples=True, currency=None, break_name: str = 'price_breaks'): """Calculate the price based on quantity price breaks. - Don't forget to add in flat-fee cost (base_cost field) - If MOQ (minimum order quantity) is required, bump quantity - If order multiples are to be observed, then we need to calculate based on that, too """ from common.settings import currency_code_default if hasattr(instance, break_name): price_breaks = getattr(instance, break_name).all() else: price_breaks = [] # No price break information available? if len(price_breaks) == 0: return None # Check if quantity is fraction and disable multiples multiples = (quantity % 1 == 0) # Order multiples if multiples: quantity = int(math.ceil(quantity / instance.multiple) * instance.multiple) pb_found = False pb_quantity = -1 pb_cost = 0.0 if currency is None: # Default currency selection currency = currency_code_default() pb_min = None for pb in price_breaks: # Store smallest price break if not pb_min: pb_min = pb # Ignore this pricebreak (quantity is too high) if pb.quantity > quantity: continue pb_found = True # If this price-break quantity is the largest so far, use it! if pb.quantity > pb_quantity: pb_quantity = pb.quantity # Convert everything to the selected currency pb_cost = pb.convert_to(currency) # Use smallest price break if not pb_found and pb_min: # Update price break information pb_quantity = pb_min.quantity pb_cost = pb_min.convert_to(currency) # Trigger cost calculation using smallest price break pb_found = True # Convert quantity to decimal.Decimal format quantity = decimal.Decimal(f'{quantity}') if pb_found: cost = pb_cost * quantity return InvenTree.helpers.normalize(cost + instance.base_cost) else: return None class ColorTheme(models.Model): """Color Theme Setting.""" name = models.CharField(max_length=20, default='', blank=True) user = models.CharField(max_length=150, unique=True) @classmethod def get_color_themes_choices(cls): """Get all color themes from static folder.""" if not settings.STATIC_COLOR_THEMES_DIR.exists(): logger.error('Theme directory does not exsist') return [] # Get files list from css/color-themes/ folder files_list = [] for file in settings.STATIC_COLOR_THEMES_DIR.iterdir(): files_list.append([file.stem, file.suffix]) # Get color themes choices (CSS sheets) choices = [(file_name.lower(), _(file_name.replace('-', ' ').title())) for file_name, file_ext in files_list if file_ext == '.css'] return choices @classmethod def is_valid_choice(cls, user_color_theme): """Check if color theme is valid choice.""" try: user_color_theme_name = user_color_theme.name except AttributeError: return False for color_theme in cls.get_color_themes_choices(): if user_color_theme_name == color_theme[0]: return True return False class VerificationMethod(Enum): """Class to hold method references.""" NONE = 0 TOKEN = 1 HMAC = 2 class WebhookEndpoint(models.Model): """Defines a Webhook entdpoint. Attributes: endpoint_id: Path to the webhook, name: Name of the webhook, active: Is this webhook active?, user: User associated with webhook, token: Token for sending a webhook, secret: Shared secret for HMAC verification, """ # Token TOKEN_NAME = "Token" VERIFICATION_METHOD = VerificationMethod.NONE MESSAGE_OK = "Message was received." MESSAGE_TOKEN_ERROR = "Incorrect token in header." endpoint_id = models.CharField( max_length=255, verbose_name=_('Endpoint'), help_text=_('Endpoint at which this webhook is received'), default=uuid.uuid4, editable=False, ) name = models.CharField( max_length=255, blank=True, null=True, verbose_name=_('Name'), help_text=_('Name for this webhook') ) active = models.BooleanField( default=True, verbose_name=_('Active'), help_text=_('Is this webhook active') ) user = models.ForeignKey( User, on_delete=models.SET_NULL, blank=True, null=True, verbose_name=_('User'), help_text=_('User'), ) token = models.CharField( max_length=255, blank=True, null=True, verbose_name=_('Token'), help_text=_('Token for access'), default=uuid.uuid4, ) secret = models.CharField( max_length=255, blank=True, null=True, verbose_name=_('Secret'), help_text=_('Shared secret for HMAC'), ) # To be overridden def init(self, request, *args, **kwargs): """Set verification method. Args: request: Original request object. """ self.verify = self.VERIFICATION_METHOD def process_webhook(self): """Process the webhook incomming. This does not deal with the data itself - that happens in process_payload. Do not touch or pickle data here - it was not verified to be safe. """ if self.token: self.verify = VerificationMethod.TOKEN if self.secret: self.verify = VerificationMethod.HMAC return True def validate_token(self, payload, headers, request): """Make sure that the provided token (if any) confirms to the setting for this endpoint. This can be overridden to create your own token validation method. """ token = headers.get(self.TOKEN_NAME, "") # no token if self.verify == VerificationMethod.NONE: # do nothing as no method was chosen pass # static token elif self.verify == VerificationMethod.TOKEN: if not compare_digest(token, self.token): raise PermissionDenied(self.MESSAGE_TOKEN_ERROR) # hmac token elif self.verify == VerificationMethod.HMAC: digest = hmac.new(self.secret.encode('utf-8'), request.body, hashlib.sha256).digest() computed_hmac = base64.b64encode(digest) if not hmac.compare_digest(computed_hmac, token.encode('utf-8')): raise PermissionDenied(self.MESSAGE_TOKEN_ERROR) return True def save_data(self, payload=None, headers=None, request=None): """Safes payload to database. Args: payload (optional): Payload that was send along. Defaults to None. headers (optional): Headers that were send along. Defaults to None. request (optional): Original request object. Defaults to None. """ return WebhookMessage.objects.create( host=request.get_host(), header=json.dumps(dict(headers.items())), body=payload, endpoint=self, ) def process_payload(self, message, payload=None, headers=None) -> bool: """Process a payload. Args: message: DB entry for this message mm payload (optional): Payload that was send along. Defaults to None. headers (optional): Headers that were included. Defaults to None. Returns: bool: Was the message processed """ return True def get_return(self, payload=None, headers=None, request=None) -> str: """Returns the message that should be returned to the endpoint caller. Args: payload (optional): Payload that was send along. Defaults to None. headers (optional): Headers that were send along. Defaults to None. request (optional): Original request object. Defaults to None. Returns: str: Message for caller. """ return self.MESSAGE_OK class WebhookMessage(models.Model): """Defines a webhook message. Attributes: message_id: Unique identifier for this message, host: Host from which this message was received, header: Header of this message, body: Body of this message, endpoint: Endpoint on which this message was received, worked_on: Was the work on this message finished? """ message_id = models.UUIDField( verbose_name=_('Message ID'), help_text=_('Unique identifier for this message'), primary_key=True, default=uuid.uuid4, editable=False, ) host = models.CharField( max_length=255, verbose_name=_('Host'), help_text=_('Host from which this message was received'), editable=False, ) header = models.CharField( max_length=255, blank=True, null=True, verbose_name=_('Header'), help_text=_('Header of this message'), editable=False, ) body = models.JSONField( blank=True, null=True, verbose_name=_('Body'), help_text=_('Body of this message'), editable=False, ) endpoint = models.ForeignKey( WebhookEndpoint, on_delete=models.SET_NULL, blank=True, null=True, verbose_name=_('Endpoint'), help_text=_('Endpoint on which this message was received'), ) worked_on = models.BooleanField( default=False, verbose_name=_('Worked on'), help_text=_('Was the work on this message finished?'), ) class NotificationEntry(MetaMixin): """A NotificationEntry records the last time a particular notifaction was sent out. It is recorded to ensure that notifications are not sent out "too often" to users. Attributes: - key: A text entry describing the notification e.g. 'part.notify_low_stock' - uid: An (optional) numerical ID for a particular instance - date: The last time this notification was sent """ class Meta: """Meta options for NotificationEntry.""" unique_together = [ ('key', 'uid'), ] key = models.CharField( max_length=250, blank=False, ) uid = models.IntegerField( ) @classmethod def check_recent(cls, key: str, uid: int, delta: timedelta): """Test if a particular notification has been sent in the specified time period.""" since = datetime.now().date() - delta entries = cls.objects.filter( key=key, uid=uid, updated__gte=since ) return entries.exists() @classmethod def notify(cls, key: str, uid: int): """Notify the database that a particular notification has been sent out.""" entry, created = cls.objects.get_or_create( key=key, uid=uid ) entry.save() class NotificationMessage(models.Model): """A NotificationMessage is a message sent to a particular user, notifying them of some *important information* Notification messages can be generated by a variety of sources. Attributes: target_object: The 'target' of the notification message source_object: The 'source' of the notification message """ # generic link to target target_content_type = models.ForeignKey( ContentType, on_delete=models.CASCADE, related_name='notification_target', ) target_object_id = models.PositiveIntegerField() target_object = GenericForeignKey('target_content_type', 'target_object_id') # generic link to source source_content_type = models.ForeignKey( ContentType, on_delete=models.SET_NULL, related_name='notification_source', null=True, blank=True, ) source_object_id = models.PositiveIntegerField( null=True, blank=True, ) source_object = GenericForeignKey('source_content_type', 'source_object_id') # user that receives the notification user = models.ForeignKey( User, on_delete=models.CASCADE, verbose_name=_('User'), help_text=_('User'), null=True, blank=True, ) category = models.CharField( max_length=250, blank=False, ) name = models.CharField( max_length=250, blank=False, ) message = models.CharField( max_length=250, blank=True, null=True, ) creation = models.DateTimeField( auto_now_add=True, ) read = models.BooleanField( default=False, ) @staticmethod def get_api_url(): """Return API endpoint.""" return reverse('api-notifications-list') def age(self): """Age of the message in seconds.""" delta = now() - self.creation return delta.seconds def age_human(self): """Humanized age.""" return naturaltime(self.creation) class NewsFeedEntry(models.Model): """A NewsFeedEntry represents an entry on the RSS/Atom feed that is generated for InvenTree news. Attributes: - feed_id: Unique id for the news item - title: Title for the news item - link: Link to the news item - published: Date of publishing of the news item - author: Author of news item - summary: Summary of the news items content - read: Was this iteam already by a superuser? """ feed_id = models.CharField( verbose_name=_('Id'), unique=True, max_length=250, ) title = models.CharField( verbose_name=_('Title'), max_length=250, ) link = models.URLField( verbose_name=_('Link'), max_length=250, ) published = models.DateTimeField( verbose_name=_('Published'), max_length=250, ) author = models.CharField( verbose_name=_('Author'), max_length=250, ) summary = models.CharField( verbose_name=_('Summary'), max_length=250, ) read = models.BooleanField( verbose_name=_('Read'), help_text=_('Was this news item read?'), default=False ) def rename_notes_image(instance, filename): """Function for renaming uploading image file. Will store in the 'notes' directory.""" fname = os.path.basename(filename) return os.path.join('notes', fname) class NotesImage(models.Model): """Model for storing uploading images for the 'notes' fields of various models. Simply stores the image file, for use in the 'notes' field (of any models which support markdown) """ image = models.ImageField( upload_to=rename_notes_image, verbose_name=_('Image'), help_text=_('Image file'), ) user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True) date = models.DateTimeField(auto_now_add=True)