import csv from vocab_test import create_app from vocab_test.models import Word from vocab_test.extensions import db app = create_app() with app.app_context(): with open('words.csv', 'rb') as csvfile: reader = csv.reader(csvfile, delimiter=',', quotechar='"') for row in reader: print row[0], row[1] q = Word.query.filter(Word.word == row[0].decode('cp1252')) if q.count > 0: word = q.first() word.definition = row[1].decode('cp1252') else: word = Word(word=row[0].decode('cp1252'), definition=row[1].decode('cp1252'), confidence=0) db.session.add(word) db.session.commit()
COV.start() if os.path.exists('.env'): print('Importing environment from .env...') for line in open('.env'): var = line.strip().split('=') if len(var) == 2: os.environ[var[0]] = var[1] from vocab_test import create_app from vocab_test.extensions import db from vocab_test.models import Word from flask.ext.script import Manager, Shell from flask.ext.migrate import Migrate, MigrateCommand app = create_app(os.getenv('FLASK_CONFIG') or 'default') manager = Manager(app) migrate = Migrate(app, db) def make_shell_context(): return dict(app=app, db=db, Word=Word) manager.add_command("shell", Shell(make_context=make_shell_context)) manager.add_command('db', MigrateCommand) @manager.command def test(coverage=False):