def _get_suggested_extension(name, format): original_extension = os.path.splitext(name)[1] try: suggested_extension = format_to_extension(format) except UnknownFormatError: extension = original_extension else: if suggested_extension.lower() == original_extension.lower(): extension = original_extension else: try: original_format = extension_to_format(original_extension) except UnknownExtensionError: extension = suggested_extension else: # If the formats match, give precedence to the original extension. if format.lower() == original_format.lower(): extension = original_extension else: extension = suggested_extension return extension
def test_format_to_extension_no_init(self): self.assertEqual(utils.format_to_extension('PNG'), '.png') self.assertEqual(utils.format_to_extension('ICO'), '.ico') with self.assertRaises(utils.UnknownFormatError): utils.format_to_extension('TXT')
def hash(generator): format = getattr(generator, 'format', None) ext = format_to_extension(format) if format else '' return os.path.normpath(os.path.join(settings.IMAGEKIT_CACHEFILE_DIR, '%s%s' % (generator.get_hash(), ext)))
def test_format_to_extension_no_init(self): self.assertEqual(utils.format_to_extension("PNG"), ".png") self.assertEqual(utils.format_to_extension("ICO"), ".ico") self.assertRaises(utils.UnknownFormatError, lambda: utils.format_to_extension("TXT"))