voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add exceptions.GenerateIDFailed.

Browse source
This commit is contained in:
voussoir 2022-11-11 15:29:22 -08:00
parent d1b0fb6aec
commit b99c594931
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
etiquette/exceptions.py
View file

@ -187,6 +187,9 @@ class FeatureDisabled(EtiquetteException):
''' '''
error_message = 'This feature has been disabled. Requires {requires}.' error_message = 'This feature has been disabled. Requires {requires}.'
class GenerateIDFailed(EtiquetteException):
error_message = 'Could not generate unique ID on {table}. You should increase ID bits.'
class MinMaxInvalid(EtiquetteException): class MinMaxInvalid(EtiquetteException):
''' '''
For when the user searches for e.g. width=a-b but the a-b can't be parsed. For when the user searches for e.g. width=a-b but the a-b can't be parsed.

3
etiquette/photodb.py
View file

@ -1236,10 +1236,11 @@ class PhotoDB(
table = thing_class.table table = thing_class.table
length = self.config['id_bits'] length = self.config['id_bits']
while True: for retry in range(10):
id = RNG.getrandbits(length) id = RNG.getrandbits(length)
if not self.exists(f'SELECT 1 FROM {table} WHERE id == ?', [id]): if not self.exists(f'SELECT 1 FROM {table} WHERE id == ?', [id]):
return id return id
raise exceptions.GenerateIDFailed(table=table)
def load_config(self) -> None: def load_config(self) -> None:
log.debug('Loading config file.') log.debug('Loading config file.')