Merge pull request #40 from arXiv/ARXIVCE-4176-limits-set-by-env-var

get limits from environment variables

Author: kyokukou

oaipmh/config.py

@@ -19,6 +19,9 @@ class Settings(arxiv_base.Settings):
     FLASKS3_FORCE_MIMETYPE: bool = True
     FLASKS3_ACTIVE: bool = False
 
+    RECORDS_LIMIT: int=2500
+    IDENTIFIERS_LIMIT: int=20000 
+
     def check(self) -> None:
         """A check and fix up of a settings object."""
         if 'sqlite' in self.CLASSIC_DB_URI:

oaipmh/data/oai_config.py

@@ -1,8 +1,10 @@
 from datetime import datetime, timezone
 from oaipmh.data.oai_properties import MetadataFormat
+from oaipmh.config import Settings
 
-RECORDS_LIMIT=2500
-IDENTIFIERS_LIMIT=20000 
+settings = Settings()  
+RECORDS_LIMIT = settings.RECORDS_LIMIT
+IDENTIFIERS_LIMIT=settings.IDENTIFIERS_LIMIT
 
 SUPPORTED_METADATA_FORMATS={
     "oai_dc":MetadataFormat(

oaipmh/factory.py

@@ -4,10 +4,9 @@
 from flask import Flask, render_template
 from flask_s3 import FlaskS3
 from flask.logging import default_handler
-from werkzeug.exceptions import HTTPException
 
 from arxiv.base import Base
-from arxiv.db import config_query_timing, configure_db
+from arxiv.db import configure_db
 from arxiv.integration.fastly.headers import add_surrogate_key
 
 from oaipmh.data.oai_errors import OAIException