-
Notifications
You must be signed in to change notification settings - Fork 0
/
base_user_params.py
37 lines (27 loc) · 1.24 KB
/
base_user_params.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os
from typing import Optional
from dotenv import load_dotenv
load_dotenv() # take environment variables from .env.
class UserParams:
def __init__(self) -> None:
# the root directory to work from
self.root_dir: Optional[str] = os.getenv("ROOT_DIR")
# the directory where the data is located
# if none the class will assume it is in: f'{self._root_dir}/{self._sdg_name}_data'
# e.g. 'C:/Users/name/Scripts/SDGs/sdg_15_1_1_data'
self.data_dir: Optional[str] = None
# the directory where the output is saved
# if none the class will assume it is in: f'{self._root_dir}/{self._sdg_name}_output'
# e.g. 'C:/Users/name/Scripts/SDGs/sdg_15_1_1_output'
self.output_dir: Optional[str] = None
# the starting year for multiple year exports
# the ONLY year for single year exports
self.year_start: int = 2017
# the ending year for multiple year exports
self.year_end: int = 2022
# if only testing a single year
# Note this requires all the file paths below to be populated
self.single_year_test: bool = False
def print_params(self) -> None:
for k, v in vars(self).items():
print(f"{k} = {v}")