CSV added as parameter in run_get_all_documents_py

MaRDI4NFDI · Feb 8, 2024 · 084adff · 084adff
1 parent 3bee580
commit 084adff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/zbmath_rest2oai/get_all_de.py b/src/zbmath_rest2oai/get_all_de.py
@@ -6,10 +6,9 @@
 from zbmath_rest2oai.writeOai import write_oai
 
 # from https://stackoverflow.com/a/38677619/9215209
-CSV_URL = 'https://github.com/MaRDI4NFDI/python-zbMathRest2Oai/releases/download/test/all_de_240115.csv'
 
 
-def get_all_de(api_source):
+def get_all_de(api_source,CSV_URL):
     with requests.get(CSV_URL, stream=True) as r:
         lines = (line.decode('utf-8') for line in r.iter_lines())
         for row in csv.reader(lines):

diff --git a/src/zbmath_rest2oai/run_get_all_de_documents.py b/src/zbmath_rest2oai/run_get_all_de_documents.py
@@ -3,4 +3,4 @@
 CSV_URL = 'https://github.com/MaRDI4NFDI/python-zbMathRest2Oai/releases/download/test/all_de_240115.csv'
 
 if __name__ == '__main__':
-    get_all_de('https://api.zbmath.org/v1/document/')
+    get_all_de('https://api.zbmath.org/v1/document/',CSV_URL)