mirror of
https://github.com/Steffo99/unimore-bda-6.git
synced 2024-11-22 07:54:19 +00:00
44 lines
850 B
Python
44 lines
850 B
Python
import cfig
|
|
|
|
config = cfig.Configuration()
|
|
|
|
|
|
@config.optional()
|
|
def MONGO_HOST(val: str | None) -> str:
|
|
"""
|
|
The hostname of the MongoDB database to connect to.
|
|
"""
|
|
return val or "127.0.0.1"
|
|
|
|
|
|
@config.optional()
|
|
def MONGO_PORT(val: str | None) -> int:
|
|
"""
|
|
The port of the MongoDB database to connect to.
|
|
"""
|
|
if not val:
|
|
return 27017
|
|
try:
|
|
return int(val)
|
|
except ValueError:
|
|
raise cfig.InvalidValueError("Not an int.")
|
|
|
|
|
|
@config.optional()
|
|
def TRAINING_SET_SIZE(val: str | None) -> int:
|
|
"""
|
|
The number of reviews from each category to fetch for the training set.
|
|
"""
|
|
if not val:
|
|
return 1000
|
|
try:
|
|
return int(val)
|
|
except ValueError:
|
|
raise cfig.InvalidValueError("Not an int.")
|
|
|
|
|
|
__all__ = (
|
|
"config",
|
|
"MONGO_HOST",
|
|
"MONGO_PORT",
|
|
)
|