12345678910111213141516171819202122232425262728293031323334353637 |
- import sqlite3
- def convert(value):
- if value.startswith('~'):
- return value.strip('~')
- if not value:
- value = '0'
- return float(value)
- conn = sqlite3.connect('food.db')
- curs = conn.cursor()
- curs.execute('''
- CREATE TABLE food (
- id TEXT PRIMARY KEY,
- desc TEXT,
- water FLOAT,
- kcal FLOAT,
- protein FLOAT,
- fat FLOAT,
- ash FLOAT,
- carbs FLOAT,
- fiber FLOAT,
- sugar FLOAT
- )
- ''')
- query = 'INSERT INTO food VALUES (?,?,?,?,?,?,?,?,?,?)'
- field_count = 10
- for line in open('ABBREV.txt'):
- fields = line.split('^')
- vals = [convert(f) for f in fields[:field_count]]
- curs.execute(query, vals)
- conn.commit()
- conn.close()
|