Added query string as configurable (Docker) environment variable
This commit is contained in:
+5
-1
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user