|
|
@@ -15,7 +15,13 @@ NUTRIENT_MAP = {
|
|
|
'205': 'carbs_g',
|
|
|
'291': 'fiber_g',
|
|
|
'269': 'sugar_g',
|
|
|
- '307': 'sodium_mg'
|
|
|
+ '307': 'sodium_mg',
|
|
|
+ '301': 'calcium_mg',
|
|
|
+ '303': 'iron_mg',
|
|
|
+ '306': 'potassium_mg',
|
|
|
+ '318': 'vitamin_a_iu',
|
|
|
+ '401': 'vitamin_c_mg',
|
|
|
+ '601': 'cholesterol_mg'
|
|
|
}
|
|
|
|
|
|
def parse_usda_line(line):
|
|
|
@@ -86,8 +92,10 @@ def run_seeding():
|
|
|
|
|
|
cursor.execute('''
|
|
|
INSERT INTO foods (
|
|
|
- name, category, calories, protein_g, fat_g, carbs_g, fiber_g, sugar_g, sodium_mg, source
|
|
|
- ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
|
+ name, category, calories, protein_g, fat_g, carbs_g, fiber_g, sugar_g,
|
|
|
+ sodium_mg, calcium_mg, iron_mg, potassium_mg, vitamin_a_iu, vitamin_c_mg,
|
|
|
+ cholesterol_mg, source
|
|
|
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
|
''', (
|
|
|
info['name'],
|
|
|
info['category'],
|
|
|
@@ -98,6 +106,12 @@ def run_seeding():
|
|
|
macros.get('fiber_g', 0.0),
|
|
|
macros.get('sugar_g', 0.0),
|
|
|
macros.get('sodium_mg', 0.0),
|
|
|
+ macros.get('calcium_mg', 0.0),
|
|
|
+ macros.get('iron_mg', 0.0),
|
|
|
+ macros.get('potassium_mg', 0.0),
|
|
|
+ macros.get('vitamin_a_iu', 0.0),
|
|
|
+ macros.get('vitamin_c_mg', 0.0),
|
|
|
+ macros.get('cholesterol_mg', 0.0),
|
|
|
f"USDA-{ndb_no}"
|
|
|
))
|
|
|
count += 1
|