extra columns for off

openfoodfacts · Nov 6, 2024 · ec75d1e · ec75d1e
1 parent 8dc55c4
commit ec75d1e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/generate_dump_for_offline_apps.py b/scripts/generate_dump_for_offline_apps.py
@@ -15,7 +15,11 @@ def main():
         os.makedirs(off_public_data_dir + '/offline')
 
     df = pandas.read_csv(off_public_data_dir + '/en.' + product_opener_flavor + '.org.products.csv', sep='\t', low_memory=False)
-    colnames = ['code','product_name','quantity','brands','nutriscore_grade','nova_group','ecoscore_grade']
+    colnames = ['code','product_name','quantity','brands']
+    # add 'nutriscore_grade','nova_group','ecoscore_grade' columns if the flavor is off
+    if product_opener_flavor_short == 'off':
+        colnames = colnames + ['nutriscore_grade','nova_group','ecoscore_grade']
+
     df.rename(columns={'nutriscore_grade': 'nutrition_grade_fr'}).to_csv(off_public_data_dir + '/offline/en.' + product_opener_flavor + '.org.products.small.csv', columns = colnames,sep='\t',index=False)
 
 if __name__ == '__main__':