bacow/AV_Data_Capture
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add config path search order

Browse Source
This commit is contained in:
lededev
2021-09-27 01:05:24 +08:00
parent d7197d8b8c
commit cc36562c17
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
config.py
View File

@@ -2,15 +2,29 @@ import os
import sys
import configparser
import codecs
from pathlib import Path
class Config:
def __init__(self, path: str = "config.ini"):
if os.path.exists(path):
path_search_order = [
path,
"./config.ini",
os.path.join(Path.home(), "avdc.ini"),
os.path.join(Path.home(), ".avdc.ini"),
os.path.join(Path.home(), ".avdc/config.ini"),
os.path.join(Path.home(), ".config/avdc/config.ini")
]
ini_path = None
for p in path_search_order:
if os.path.exists(p):
ini_path = p
break
if ini_path:
self.conf = configparser.ConfigParser()
try:
self.conf.read(path, encoding="utf-8-sig")
self.conf.read(ini_path, encoding="utf-8-sig")
except:
self.conf.read(path, encoding="utf-8")
self.conf.read(ini_path, encoding="utf-8")
else:
print("[-]Config file not found!")
sys.exit(2)