Create basic recipe stuff.

beer/models.py

@@ -37,9 +37,162 @@ class Batch(CustomModel):
         # Return a string that represents the instance
         return 'BF #{num}: {name}'.format(name=self.brewfather_name, num=self.brewfather_num)
 
+#-----------------------------------------------------------------------
+# Recipe Stuff
+#-----------------------------------------------------------------------
+class Unit(CustomModel):
+    unit_types = {
+        'WT': 'Weight',
+        'VL': 'Volume',
+    }
+
+    """ Recipe to be stored with a batch."""
+    name = models.CharField(max_length=50)
+    unit_type = models.CharField(max_length=3, choices=unit_types, default='WT')
+
+class Supplier(CustomModel):
+    name = models.CharField(max_length=50)
+    
+class CustomIngredient(CustomModel):
+    """ Custom model class with default fields to use. """
+    created_date = models.DateTimeField(default=timezone.now)
+    name = models.CharField(max_length=50)
+    units = models.ForeignKey(Unit, on_delete=models.PROTECT)
+    unit_cost = models.DecimalField(max_digits=6, decimal_places=2, null=True, blank=True)
+    supplier = models.ForeignKey(Supplier, on_delete=models.PROTECT, null=True, blank=True)
+    notes = models.TextField(max_length=500, blank=True, null=True)
+    user_notes = models.TextField(max_length=500, blank=True, null=True)
+
+    class Meta:
+        abstract = True
+        
 class BatchRecipe(CustomModel):
     """ Recipe to be stored with a batch."""
     name = models.CharField(max_length=50)
     batch_recipe = models.BooleanField(null=True)
     recipe_json = models.TextField(null=True, blank=True)
-   
+    mash = models.ForeignKey('Mash' , on_delete=models.PROTECT, null=True, blank=True)
+    efficiency = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+
+    def __str__(self):
+        return self.name
+
+class Fermentable(CustomIngredient):
+    categories = {
+        1: 'Base',
+        2: 'Wheat/Oat',
+        3: 'Crystal',
+        4: 'Roasted',
+        5: 'Acid',
+    }
+
+    types = {
+        1: 'Grain',
+        2: 'Adjunct',
+    }
+
+    grain_category = models.IntegerField(choices=categories, default=1)
+    fermentable_type = models.IntegerField(choices=types, default=1)
+    diastatic_power = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    potential = models.DecimalField(max_digits=6, decimal_places=4)
+    protein = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    attenuation = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    lovibond = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    max_in_batch = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    moisture = models.DecimalField(max_digits=6, decimal_places=4, null=True, blank=True)
+    non_fermentable = models.BooleanField(null=True, blank=True)
+    ibu_per_unit = models.DecimalField(max_digits=6, decimal_places=4, default=0)
+
+    def __str__(self):
+        return self.name
+
+class RecipeFermentable(CustomModel):
+    recipe = models.ForeignKey(BatchRecipe, on_delete=models.CASCADE)
+    fermentable = models.ForeignKey(Fermentable, on_delete=models.CASCADE)
+    quantity = models.DecimalField(max_digits=6, decimal_places=4)
+
+class Hop(CustomIngredient):
+    uses = {
+        1: 'Bittering',
+        2: 'Aroma',
+        3: 'Both',
+    }
+
+    types = {
+        1: 'Pellet',
+        2: 'Leaf',
+        3: 'Cryo',
+        4: 'CO2 Extract',
+    }
+
+    ibu = models.DecimalField(max_digits=6, decimal_places=4, default=0)
+    use = models.IntegerField(choices=uses, default=1)
+    hop_type = models.IntegerField(choices=types, default=1)
+    alpha = models.DecimalField(max_digits=6, decimal_places=2, default=0)
+
+    def __str__(self):
+        return self.name
+
+class RecipeHop(CustomModel):
+    recipe = models.ForeignKey(BatchRecipe, on_delete=models.CASCADE)
+    hop = models.ForeignKey(Hop, on_delete=models.CASCADE)
+    quantity = models.DecimalField(max_digits=6, decimal_places=4)
+
+class Misc(CustomIngredient):
+    uses = {
+        1: 'Mash',
+        2: 'Sparge',
+        3: 'Boil',
+        4: 'Flamout',
+        5: 'Primary',
+        6: 'Secondary',
+        7: 'Cold Crash',
+        8: 'Bottling',
+    }
+
+    types = {
+        1: 'Spice',
+        2: 'Fining',
+        3: 'Water Agent',
+        4: 'Herb',
+        5: 'Flavor',
+        6: 'Other',
+    }
+
+    use = models.IntegerField(choices=uses, default=1)
+    misc_type = models.IntegerField(choices=types, default=1)
+    water_adjustment = models.BooleanField(null=True, blank=True)
+
+    def __str__(self):
+        return self.name
+
+class RecipeMisc(CustomModel):
+    recipe = models.ForeignKey(BatchRecipe, on_delete=models.CASCADE)
+    misc = models.ForeignKey(Misc, on_delete=models.CASCADE)
+    quantity = models.DecimalField(max_digits=6, decimal_places=4)
+
+class RecipeYeast(CustomModel):
+    recipe = models.ForeignKey(BatchRecipe, on_delete=models.CASCADE)
+    yeast = models.ForeignKey('yeast.Strain', on_delete=models.CASCADE)
+
+class Mash(CustomModel):
+    name = models.CharField(max_length=50)
+
+    def __str__(self):
+        return self.name
+
+class MashStep(CustomModel):
+    step_types = {
+        1: 'infusion',
+        2: 'temperature',
+        3: 'decoction',
+    }
+    name = models.CharField(max_length=50)
+    step_temp = models.DecimalField(max_digits=6, decimal_places=2)
+    ramp_time = models.DecimalField(max_digits=6, decimal_places=2)
+    step_time = models.DecimalField(max_digits=6, decimal_places=2)
+    step_type = models.IntegerField(choices=step_types, default=1)
+    parent_mash = models.ForeignKey(Mash, on_delete=models.CASCADE)
+
+    def __str__(self):
+        return self.name