Fix ovld display / export exception with ExtendAttr class

paperoni/cli_helper.py

@@ -173,6 +173,11 @@ class ExtendAttr():
     def __init__(self, search_result) -> None:
         self._search_result = search_result
 
+    # TODO: make this class fake it's type so it works with the Ovld package
+    @property
+    def __class__(self):
+        return self._search_result.__class__
+
     def __getattribute__(self, __name: str) -> Any:
         try:
             attr = super().__getattribute__(__name)
@@ -250,6 +255,7 @@ def query_papers(
     # [negate]
     allow_download: Option & bool = True,
 ):
+    excerpt = r"[^\w](Milaâ?|Quebec AI Institute|Montreal Institute for Learning Algorithms|Québec AI Institute Université de Montréal)[^\w]"
     yield from search(
         title=title,
         author=author,

paperoni/db/model_export.py

@@ -1,9 +1,9 @@
 from ovld import ovld
 
+from ..cli_helper import ExtendAttr
 from .. import model as M
 from . import schema as sch
 
-
 @ovld
 def export(paper: sch.Paper):
     return M.Paper(
@@ -86,3 +86,10 @@ def export(venue: sch.Venue):
         volume=venue.volume,
         quality=venue.quality,
     )
+
+
+@ovld
+def export(extattr: ExtendAttr):
+    attrs = {**extattr.__dict__}
+    del attrs["_search_result"]
+    export(extattr._search_result, **attrs)

paperoni/display.py

@@ -7,6 +7,7 @@
 from hrepr import H
 from ovld import ovld
 
+from .cli_helper import ExtendAttr
 from .db import schema as sch
 from .model import Author, DatePrecision, Paper, Venue, from_dict
 
@@ -91,7 +92,7 @@ def display(d: dict):
 
 
 @ovld
-def display(paper: Union[Paper, sch.Paper]):
+def display(paper: Union[Paper, sch.Paper], excerpt=None):
     """Print the paper in long form on the terminal.
 
     Long form includes abstract, affiliations, keywords, number of
@@ -119,7 +120,9 @@ def display(paper: Union[Paper, sch.Paper]):
         print(f"  {T.bold_green(typ)} {link}")
     print_field("Citations", paper.citation_count)
     if hasattr(paper, "excerpt"):
-        before, match, after = paper.excerpt
+        excerpt = paper.excerpt
+    if excerpt:
+        before, match, after = excerpt
         print_field("Excerpt", before + T.bold_red(match) + after)
 
 
@@ -157,6 +160,13 @@ def display(venue: Venue):
         print(f"  {T.bold_green(typ):20} {link}")
 
 
+@ovld
+def display(extattr: ExtendAttr):
+    attrs = {**extattr.__dict__}
+    del attrs["_search_result"]
+    display(extattr._search_result, **attrs)
+
+
 def join(elems, sep=", ", lastsep=None):
     """Create a list using the given separators.