def save_category(event_request, category_id): category = Category() category.id = event_request.username + str(category_id) category.category_id = category_id category.username = User.objects.get(pk=event_request.username) category.save()
def setUp(self): self.browser = webdriver.Firefox() category = Category(name="salary", type="income") category.save() account = Account(name="savings") account.save() income = Income(category=category, account=account, notes="October 2015 salary", amount=100000, date=datetime.datetime.now()) income.save()
def setUp(self): self.browser = webdriver.Firefox() self.browser.implicitly_wait(3) category = Category(name="others", type="expense") category.save() account = Account(name="savings") account.save() expense = Expense(category=category, account=account, notes="gift", amount=1000, date=datetime.datetime.now()) expense.save()
def save_products(self): """Save products and categories in Database.""" api_class = Api() products_and_categories_from_api = api_class.download_products() if products_and_categories_from_api is not None: print(" ✅ Les produits ont été téléchargé avec succès depuis l'API") else: print(" ❌ -> Les produits n'ont pas pu être téléchargés depuis l'API") for product in products_and_categories_from_api: name = product.get("product_name_fr")[:150].strip().lower().capitalize() brands = product.get("brands")[:150].strip().lower().capitalize() grade = product.get("nutriscore_grade")[0].upper() barcode = product.get("code")[:13].strip() url = product.get("url") picture = product.get("image_url") small_picture = product.get("image_small_url") categories = [ name.strip().lower().capitalize() for name in product["categories"].split(",") ] selected_nutriments = [ "energy_100g", "sugars_100g", "fat_100g", "salt_100g", ] nutriments_dict = {} for nutriment in selected_nutriments: nutriment_value = product.get("nutriments").get(nutriment) if isinstance(nutriment_value, float) is True: value = nutriment_value else: value = 0 nutriments_dict[nutriment] = value product_instance = Product( name=name, brand=brands, barcode=barcode, score=grade, url=url, image_url=picture, small_image_url=small_picture, kcal_100g=nutriments_dict.get("energy_100g"), sugar_100g=nutriments_dict.get("sugars_100g"), salt_100g=nutriments_dict.get("salt_100g"), fat_100g=nutriments_dict.get("fat_100g"), ) try: product_instance.save() saved_categories = [] for category in categories: category_instance = Category(name=category) if category not in saved_categories: saved_categories.append(category) try: category_instance.save() except IntegrityError: category_instance = Category.objects.get(name=category) # Link products to categories product_instance.categories.add(category_instance) product_instance.save() except IntegrityError: continue print("😎 -> LA BASE DE DONNÉES EST COMPLÉTÉE ! <- 😎")