diff --git a/src/app.py b/src/app.py index f8564335b8f08c2acdd3edfa68c95ad69f1f06d9..4884517cf63f3f79d27d4a77c03d9eb72e49e545 100644 --- a/src/app.py +++ b/src/app.py @@ -1,6 +1,4 @@ -from flask import Flask, jsonify -from flask import request -from flask import Response +from flask import Flask, Response, jsonify, request from flask_cors import CORS import config diff --git a/src/load_templates.py b/src/load_templates.py index 17e132b9bec5e697732118fb5aa3f742efe09608..7f6def5e87c28639cbd1431ae734796400c48077 100644 --- a/src/load_templates.py +++ b/src/load_templates.py @@ -1,13 +1,12 @@ -from wiktextract.wiktionary import parse_wiktionary -from wiktextract_context import get_wiktextract_context - import os import re - import threading import time from typing import List +from wiktextract.wiktionary import parse_wiktionary + +from wiktextract_context import get_wiktextract_context DUMPS_DIR = "dumps" diff --git a/src/wiktextract_context.py b/src/wiktextract_context.py index 79d08e0a0c8f2f56eb80e9ac5bfec60d07f7bfff..2cbabf19989958a292f5a35eaccd72ec34641ab8 100644 --- a/src/wiktextract_context.py +++ b/src/wiktextract_context.py @@ -1,11 +1,8 @@ -from wiktextract import ( - WiktextractContext, - WiktionaryConfig, -) -from wikitextprocessor import Wtp, Page - from typing import Optional +from wikitextprocessor import Page, Wtp +from wiktextract import WiktextractContext, WiktionaryConfig + from get_wikicode import get_wikicode @@ -31,7 +28,9 @@ class CustomWtp(Wtp): return original_result -def get_wiktextract_context(wiktlang: str, wordlang: Optional[str] = None, mock_get_page: bool = True): +def get_wiktextract_context( + wiktlang: str, wordlang: Optional[str] = None, mock_get_page: bool = True +): db_path = f"./sqlite-{wiktlang}.db" config = WiktionaryConfig( dump_file_lang_code=wiktlang, @@ -47,7 +46,11 @@ def get_wiktextract_context(wiktlang: str, wordlang: Optional[str] = None, mock_ capture_inflections=True, ) config.load_edition_settings() - wtp = CustomWtp(db_path=db_path, lang_code=wiktlang) if mock_get_page else Wtp(db_path=db_path, lang_code=wiktlang) + wtp = ( + CustomWtp(db_path=db_path, lang_code=wiktlang) + if mock_get_page + else Wtp(db_path=db_path, lang_code=wiktlang) + ) wxr = WiktextractContext(wtp, config) return wxr diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py index 0f072ec1d0a51548829adba9ccdd30f1915fddd0..79f78b2b28777f81acec6ffe026f0afb304cb23d 100644 --- a/src/wiktextract_wrapper.py +++ b/src/wiktextract_wrapper.py @@ -1,7 +1,7 @@ from wiktextract.page import parse_page from wiktextract_context import get_wiktextract_context - + db_path = "./sqlite.db" DEFAULT_PAGE_VALUES = { @@ -16,7 +16,7 @@ class Wiktextract: self.wordlang = wordlang # self.page_handler = page_handler - # self.page_handler.wxr : WiktextractContext = + # self.page_handler.wxr : WiktextractContext = self.wxr = get_wiktextract_context(wiktlang, wordlang) def parse_page(self, title: str, wikicode: str):