def set_package_ratio(): """Walk through products and set `package` and `package_ratio`""" keeper = StorageManager(FileStorage('storage.fs')) products = Product.fetch_all(keeper) for product in products: try: product.package = product.get_package() product.package_ratio = product.package.get_ratio(product.category) print(green(u'Product "{}" updated'.format(product))) except AttributeError, e: print(red(u'{} removed'.format(product))) keeper.delete(product) except PackageLookupError, e: logging.debug(e.message.encode('utf-8')) print(yellow(e.message))
def test_fetch_all(self): products = Product.fetch_all(self.keeper) for product in products: self.assertIsInstance(product, Product)