Flask: вставить некоторые тестовые данные при запуске? - PullRequest
Новая
       7

Flask: вставить некоторые тестовые данные при запуске?

0 голосов
/ 07 ноября 2018

Я пытаюсь вставить некоторые фиктивные данные в мою базу данных при первом запуске колбы. Я создаю свои таблицы с flask-migrate

Я бегу:

  • flask db init
  • flask db migrate
  • flask db upgrade

Я проверяю скрипт миграции и все выглядит хорошо.

Затем я запускаю Dbeaver и открываю свой файл test.db и проверяю, что таблицы из моих моделей действительно созданы.

Я даже запустил insert into keywords (keyword) values ('test'); в Dbeaver просто для проверки того, что я могу вставить некоторые данные.

Однако, когда я запускаю свое приложение через flask run, я получаю sqlalchemy.exc.OperationalError: (sqlite3.OperationalError) no such table: keywords ??

При трассировке возникает ошибка existing = Keyword.query.filter(Keyword.keyword == k.lower()).one_or_none() внутри моей функции create_app.

Вся функция create_app выглядит следующим образом: 

def create_app(config='metadata.settings'):
    app = Flask(__name__, instance_relative_config=True)

    app.config.from_object(config)
    load_extensions(app)  # sets db.init_app(app)

    @app.before_first_request
    def insert_test_data():
        test_record = {
            "metadata_version": "1.1",
            "name": "foo-package",
            "version": "0.1",
            "platforms": [
                "UNKNOWN"
            ],
            "summary": "Hello Foo Package",
            "keywords": "foo pandas numpy",
            "home_page": "https://gitlab.blah.com/my-examples/model-bar",
            "author": "John Smith",
            "author_email": "john.smith@gmail.com",
            "classifiers": [
                "Programming Language :: Python :: 3",
                "Operating System :: Microsoft :: Windows :: Windows 10",
                "Development Status :: 1 - Planning",
                "Topic :: Software Development :: Testing :: Mocking",
                "Topic :: Software Development :: Testing"
            ]
        }

        def add_record(test_record):
            rec = PackageMetadata()
            rec.name = test_record['name'],
            rec.version = test_record['version'],
            rec.desc = test_record['summary'],
            rec.home_page = test_record['home_page'],
            rec.author = test_record['author'],
            rec.author_email = test_record['author_email']

            keywords = test_record['keywords'].split(' ')
            for keyword in keywords:
                keyword_to_add = add_keyword(keyword)
                rec.keywords.append(keyword_to_add)

            classifiers = test_record['classifiers']
            for classifier in classifiers:
                classifier_to_add = add_classifier(classifier)
                rec.classifiers.append(classifier_to_add)

            try:
                db.session.add(rec)
                db.session.commit()
            except Exception as e:
                print(e)

        def add_keyword(k):
            existing = Keyword.query.filter(Keyword.keyword == k.lower()).one_or_none()
            if existing is not None:
                return existing
            else:
                new_keyword = Keyword()
                new_keyword.keyword = k.lower()
                return new_keyword

        def add_classifier(k):
            existing = Classifier.query.filter(Classifier.classifier == k).one_or_none()
            if existing is not None:
                return existing
            else:
                new_classifier = Classifier()
                new_classifier.classifier = k
                return new_classifier
        add_record(test_record)
    return app

1 Ответ

0 голосов
/ 07 ноября 2018

Таблица, созданная в БД, имеет имя, отличное от того, которое вы используете в файле модели. Вы должны проверить это в файле схемы.

...