sij/sijapi
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Auto-update: Mon Sep 23 09:54:46 PDT 2024

Browse source
This commit is contained in:
sanj 2024-09-23 09:54:46 -07:00
parent 3fc0f9d30c
commit 7abaa9d75c
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
sijapi/helpers/db/repl.py
View file

@ -8,8 +8,9 @@ from tqdm import tqdm
def load_config(): def load_config():
script_dir = os.path.dirname(os.path.abspath(__file__)) script_dir = os.path.dirname(os.path.abspath(__file__))
sys_config_path = os.path.join(script_dir, '..', 'config', 'sys.yaml') project_root = os.path.abspath(os.path.join(script_dir, '..', '..'))
gis_config_path = os.path.join(script_dir, '..', 'config', 'gis.yaml') sys_config_path = os.path.join(project_root, 'config', 'sys.yaml')
gis_config_path = os.path.join(project_root, 'config', 'gis.yaml')
with open(sys_config_path, 'r') as f: with open(sys_config_path, 'r') as f:
sys_config = yaml.safe_load(f) sys_config = yaml.safe_load(f)
@ -19,6 +20,7 @@ def load_config():
return sys_config, gis_config return sys_config, gis_config
def get_table_size(server, table_name): def get_table_size(server, table_name):
env = os.environ.copy() env = os.environ.copy()
env['PGPASSWORD'] = server['db_pass'] env['PGPASSWORD'] = server['db_pass']