Added query string as configurable (Docker) environment variable

This commit is contained in:
Heiko Joerg Schick
2023-09-07 16:02:09 +02:00
parent 6a4b1176c6
commit 169db54df2
3 changed files with 9 additions and 3 deletions
+5 -1
View File
@@ -9,6 +9,7 @@ import time
import random
import logging
import argparse
import os
from aslite.arxiv import get_response, parse_response
from aslite.db import get_papers_db, get_metas_db
@@ -30,7 +31,10 @@ if __name__ == '__main__':
"""
# query string of papers to look for
q = 'cat:cs.CV+OR+cat:cs.LG+OR+cat:cs.CL+OR+cat:cs.AI+OR+cat:cs.NE+OR+cat:cs.RO+OR+cat:cs.AR+OR+cat:cs.DC'
q = os.environ.get('ARXIV_QUERY', '')
if not q:
print("No query string provided, will use default.")
q = 'cat:cs.CV+OR+cat:cs.LG+OR+cat:cs.CL+OR+cat:cs.AI+OR+cat:cs.NE+OR+cat:cs.RO+OR+cat:cs.AR+OR+cat:cs.DC'
pdb = get_papers_db(flag='c')
mdb = get_metas_db(flag='c')