-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.py
42 lines (32 loc) · 1.28 KB
/
config.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
38
39
40
41
42
import os
import yaml
import logging
from larksuiteoapi import Config, DOMAIN_FEISHU, DefaultLogger, LEVEL_DEBUG
from larksuiteoapi.service.im.v1 import Service as ImService
def load_global_config(path):
if os.path.exists(path):
with open(path, "r") as f:
config = yaml.safe_load(f)
else:
config = dict()
config["APP_ID"] = os.environ.get('APP_ID')
config["APP_SECRET"] = os.environ.get("APP_SECRET")
config["APP_VERIFICATION_TOKEN"] = os.environ.get("APP_VERIFICATION_TOKEN")
config["APP_ENCRYPT_KEY"] = os.environ.get("APP_ENCRYPT_KEY")
return config
global_config = load_global_config("config.yaml")
def _get_sdk_config():
app_settings = Config.new_internal_app_settings(
app_id=global_config["APP_ID"],
app_secret=global_config["APP_SECRET"],
verification_token=global_config["APP_VERIFICATION_TOKEN"],
encrypt_key=global_config["APP_ENCRYPT_KEY"],
)
logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.DEBUG)
# TODO: log level use config file
sdk_config = Config.new_config_with_memory_store(
DOMAIN_FEISHU, app_settings, DefaultLogger(), LEVEL_DEBUG
)
return sdk_config
sdk_config = _get_sdk_config()
service = ImService(sdk_config)