fix: update scraping code (#6)

osl-incubator · Feb 4, 2024 · 220f2bf · 220f2bf
1 parent 0b3480c
commit 220f2bf
Show file tree

Hide file tree

Showing 6 changed files with 266 additions and 44 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,6 +17,7 @@ exclude = [
 [tool.poetry.dependencies]
 python = ">3.8.1,<4"
 requests = ">=2.20.0"
+lxml = "^5.1.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = ">=7.3.2"
@@ -41,6 +42,7 @@ mkdocstrings-python = ">=1.1.2"
 jupyterlab = ">=4.0.5"
 makim = "1.12.0"
 requests-cache = ">=1"
+types-lxml = "^2023.10.21"
 
 [tool.pytest.ini_options]
 testpaths = [

diff --git a/src/pymedx/api.py b/src/pymedx/api.py
@@ -1,12 +1,13 @@
 """API module for PubMed."""
 import datetime
 import itertools
-import xml.etree.ElementTree as xml
 
 from typing import Any, Dict, Iterable, List, Union, cast
 
 import requests
 
+from lxml import etree as xml
+
 from .article import PubMedArticle
 from .book import PubMedBookArticle
 from .helpers import batches