Merge pull request #999 from TeachMeTW/Fix-Result-Limit

Fix `DB_RESULT_LIMIT` Parsing to Allow Integer Configuration
e-mission · Jan 4, 2025 · e19af35 · e19af35
2 parents 6ab5a5c + 4150b22
commit e19af35
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/emission/core/get_database.py b/emission/core/get_database.py
@@ -23,7 +23,10 @@
         db_config[key] = None
 print("Retrieved config: %s" % db_config)
 url = config.get("DB_HOST", "localhost")
-result_limit = config.get("DB_RESULT_LIMIT", 250000)
+try:
+    result_limit = int(config.get("DB_RESULT_LIMIT", 250000))
+except ValueError:
+    result_limit = 250000
 
 try:
     parsed=pymongo.uri_parser.parse_uri(url)