def test_convert(self): self.code1 = base62_urlsafe.encode("1234") self.code2 = base62_urlsafe.encode("4567") self.code3 = base62_urlsafe.encode("8901") self.assertEqual(base62_urlsafe.decode(self.code1), "1234") self.assertEqual(base62_urlsafe.decode(self.code2), "4567") self.assertEqual(base62_urlsafe.decode(self.code3), "8901")
def add_code(signal, instance, sender, **kwargs): if not instance.short_code: instance.short_code = base62_urlsafe.encode(instance.id) instance.save()