From 04a1d7b357055f3b3db6d912585c33a87b832554 Mon Sep 17 00:00:00 2001 From: fredrikr79 Date: Sat, 10 Jan 2026 18:11:24 +0100 Subject: [PATCH] add languages to seeded test data --- src/worblehat/devscripts/seed_test_data.py | 19 ++++++++++--------- src/worblehat/services/bookcase_item.py | 3 +-- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/src/worblehat/devscripts/seed_test_data.py b/src/worblehat/devscripts/seed_test_data.py index ccb3144..4d81c15 100644 --- a/src/worblehat/devscripts/seed_test_data.py +++ b/src/worblehat/devscripts/seed_test_data.py @@ -1,17 +1,18 @@ import csv - from pathlib import Path from worblehat.models import ( Bookcase, BookcaseItem, BookcaseShelf, - MediaType, Language, + MediaType, ) - -CSV_FILE = Path(__file__).parent.parent.parent.parent / "data" / "arbeidsrom_smal_hylle_5.csv" +CSV_FILE = ( + Path(__file__).parent.parent.parent.parent / "data" / "arbeidsrom_smal_hylle_5.csv" +) +LANGUAGE_FILE = Path(__file__).parent.parent.parent.parent / "data" / "iso639_1.csv" def clear_db(sql_session): @@ -32,11 +33,11 @@ def main(sql_session): ) sql_session.add(media_type) - language = Language( - name="Norwegian", - iso639_1_code="no", - ) - sql_session.add(language) + with open(LANGUAGE_FILE, newline="") as langs: + t = csv.reader(langs, delimiter=",", quotechar="|") + for row in t: + language = Language(name=row[1], iso639_1_code=row[0]) + sql_session.add(language) seed_case = Bookcase( name="seed_case", diff --git a/src/worblehat/services/bookcase_item.py b/src/worblehat/services/bookcase_item.py index 5e8c3aa..b5cb9e6 100644 --- a/src/worblehat/services/bookcase_item.py +++ b/src/worblehat/services/bookcase_item.py @@ -1,14 +1,13 @@ import isbnlib - from sqlalchemy import select from sqlalchemy.orm import Session -from .metadata_fetchers import fetch_metadata_from_multiple_sources from ..models import ( Author, BookcaseItem, Language, ) +from .metadata_fetchers import fetch_metadata_from_multiple_sources def is_valid_pvv_isbn(isbn: str) -> bool: