mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-20 13:10:05 +08:00
26 lines
739 B
Python
Executable File
26 lines
739 B
Python
Executable File
from configparser import ConfigParser
|
|
import os,inspect
|
|
|
|
CF = ConfigParser()
|
|
__fnm = os.path.join(os.path.dirname(__file__), '../conf/sys.cnf')
|
|
if not os.path.exists(__fnm):__fnm = os.path.join(os.path.dirname(__file__), '../../conf/sys.cnf')
|
|
assert os.path.exists(__fnm), f"【EXCEPTION】can't find {__fnm}." + os.path.dirname(__file__)
|
|
if not os.path.exists(__fnm): __fnm = "./sys.cnf"
|
|
|
|
CF.read(__fnm)
|
|
|
|
class Config:
|
|
def __init__(self, env):
|
|
self.env = env
|
|
if env == "spark":CF.read("./cv.cnf")
|
|
|
|
def get(self, key, default=None):
|
|
global CF
|
|
return os.environ.get(key.upper(), \
|
|
CF[self.env].get(key, default)
|
|
)
|
|
|
|
def init(env):
|
|
return Config(env)
|
|
|