diff --git a/factorygame/data/fetch.py b/factorygame/data/fetch.py index b60fa06..ca0f759 100755 --- a/factorygame/data/fetch.py +++ b/factorygame/data/fetch.py @@ -13,8 +13,9 @@ from ..helper import click_prompt @click.command() @click.option("--debug", is_flag=True) @click.option("--refetch", is_flag=True) +@click.option("--ignore-factories", type=list[str], default=["A.I. Fluid Packer", "Craft Bench"]) @click.argument("search") -def main(debug: bool, refetch: bool, search: str): +def main(debug: bool, refetch: bool, ignore_factories: list[str], search: str): engine = create_engine("sqlite:///file.db", echo=debug) Base.metadata.create_all(bind=engine) if not search: @@ -25,7 +26,7 @@ def main(debug: bool, refetch: bool, search: str): resource = chose_resource(session=session, resource_label=search, prompt=click_prompt).result() exists_in_db = resource is not None - with SatisfactoryPlus(debug=debug) as data_provider: + with SatisfactoryPlus(ignore_factories=ignore_factories, debug=debug) as data_provider: if resource is None: ret = data_provider.search_for_resource(session=session, search=search) if ret is None: diff --git a/factorygame/data/sfp.py b/factorygame/data/sfp.py index fd87db9..35472ab 100644 --- a/factorygame/data/sfp.py +++ b/factorygame/data/sfp.py @@ -15,9 +15,13 @@ from ..helper import click_prompt class SatisfactoryPlus(RecipeProvider, AbstractContextManager): _browser: Optional[Firefox] = None + ignore_factories: list[str] = [] - def __init__(self, debug: bool = False): + def __init__(self, ignore_factories: list[str] = None, debug: bool = False): super().__init__() + if ignore_factories: + for factory in ignore_factories: + self.ignore_factories.append(factory.casefold()) self.debug = debug def _init_browser(self) -> Firefox: @@ -109,6 +113,9 @@ class SatisfactoryPlus(RecipeProvider, AbstractContextManager): factory_label = factory_html_elem.text.strip() assert factory_label, "factory label is missing (a[text])" + if factory_label.casefold() in self.ignore_factories: + return + # re-use existing Factory or create new factory = session.scalars(Factory.by_label(factory_label)).one_or_none() if factory is None: @@ -117,6 +124,9 @@ class SatisfactoryPlus(RecipeProvider, AbstractContextManager): factory = Factory(label=factory_label, uri=self._normalize_url(href=factory_href)) session.add(factory) recipe_factories.append(factory) + if not recipe_factories: + # ignore recipe when no applicable factories exist + return def find_or_create_resource(resource_label: str, resource_uri_getter: Callable) -> Resource: db_resource = session.scalars(Resource.by_label(resource_label)).one_or_none()