iUpdates storygraph tests

This commit is contained in:
Mouse Reeve
2021-11-11 12:39:12 -08:00
parent 5967834805
commit efcf7824dd
2 changed files with 12 additions and 6 deletions

View File

@ -77,7 +77,9 @@ class Importer:
def get_shelf(self, normalized_row):
"""determine which shelf to use"""
shelf_name = normalized_row["shelf"]
shelf = [s for (s, gs) in self.shelf_mapping_guesses if shelf_name in gs]
shelf = [
s for (s, gs) in self.shelf_mapping_guesses.items() if shelf_name in gs
]
return shelf[0] if shelf else None
def normalize_row(self, entry, mappings): # pylint: disable=no-self-use