mirror of https://github.com/skygpu/skynet.git
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
#!/usr/bin/python
|
|
|
|
import os
|
|
import json
|
|
|
|
from pathlib import Path
|
|
from configparser import ConfigParser
|
|
from re import sub
|
|
|
|
from .constants import DEFAULT_CONFIG_PATH
|
|
|
|
|
|
def load_skynet_ini(
|
|
file_path=DEFAULT_CONFIG_PATH
|
|
):
|
|
config = ConfigParser()
|
|
config.read(file_path)
|
|
|
|
return config
|
|
|
|
|
|
def init_env_from_config(
|
|
hf_token: str | None = None,
|
|
hf_home: str | None = None,
|
|
tg_token: str | None = None,
|
|
file_path=DEFAULT_CONFIG_PATH
|
|
):
|
|
config = load_skynet_ini(file_path=file_path)
|
|
|
|
if 'HF_TOKEN' in os.environ:
|
|
hf_token = os.environ['HF_TOKEN']
|
|
|
|
elif 'skynet.dgpu' in config:
|
|
sub_config = config['skynet.dgpu']
|
|
if 'hf_token' in sub_config:
|
|
hf_token = sub_config['hf_token']
|
|
os.environ['HF_TOKEN'] = hf_token
|
|
|
|
if 'HF_HOME' in os.environ:
|
|
hf_home = os.environ['HF_HOME']
|
|
|
|
elif 'skynet.dgpu' in config:
|
|
sub_config = config['skynet.dgpu']
|
|
if 'hf_home' in sub_config:
|
|
hf_home = sub_config['hf_home']
|
|
os.environ['HF_HOME'] = hf_home
|
|
|
|
if 'TG_TOKEN' in os.environ:
|
|
tg_token = os.environ['TG_TOKEN']
|
|
elif 'skynet.telegram' in config:
|
|
sub_config = config['skynet.telegram']
|
|
if 'token' in sub_config:
|
|
tg_token = sub_config['token']
|
|
|
|
return hf_home, hf_token, tg_token
|
|
|
|
|
|
def load_account_info(
|
|
_type: str,
|
|
key: str | None = None,
|
|
account: str | None = None,
|
|
permission: str | None = None,
|
|
file_path=DEFAULT_CONFIG_PATH
|
|
):
|
|
config = load_skynet_ini(file_path=file_path)
|
|
|
|
type_key = f'skynet.{_type}'
|
|
|
|
if type_key in config:
|
|
sub_config = config[type_key]
|
|
if not key and 'key' in sub_config:
|
|
key = sub_config['key']
|
|
|
|
if not account and 'name' in sub_config:
|
|
account = sub_config['name']
|
|
|
|
if not permission and 'permission' in sub_config:
|
|
permission = sub_config['permission']
|
|
|
|
return key, account, permission
|
|
|
|
|
|
def load_endpoint_info(
|
|
_type: str,
|
|
node_url: str | None = None,
|
|
hyperion_url: str | None = None,
|
|
ipfs_url: str | None = None,
|
|
file_path=DEFAULT_CONFIG_PATH
|
|
):
|
|
config = load_skynet_ini(file_path=file_path)
|
|
|
|
type_key = f'skynet.{_type}'
|
|
|
|
if type_key in config:
|
|
sub_config = config[type_key]
|
|
if not node_url and 'node_url' in sub_config:
|
|
node_url = sub_config['node_url']
|
|
|
|
if not hyperion_url and 'hyperion_url' in sub_config:
|
|
hyperion_url = sub_config['hyperion_url']
|
|
|
|
if not ipfs_url and 'ipfs_url' in sub_config:
|
|
ipfs_url = sub_config['ipfs_url']
|
|
|
|
return node_url, hyperion_url, ipfs_url
|