def setUp(self):
self.token_patcher = mock.patch(
'thirdparty_data_sdk.dingding.dingsdk.accesstoken_manager')
self.request_patcher = mock.patch(
'thirdparty_data_sdk.dingding.dingsdk.request_manager')
self.mock_token_manager = self.token_patcher.start()
self.mock_request_manager = self.request_patcher.start()
self.department_manager = DepartmentManager(
AccessTokenManager(TEST_APP_KEY, TEST_APP_SECRET))
self.department_manager.token_manager = self.mock_token_manager
self.department_manager.request_manager = self.mock_request_manager
self.department_manager.token = lambda *args, **kwargs: TEST_TOKEN
def entrypoint():
'''
import dingding data
'''
ding_config = DingConfig.get_current()
token_manager = AccessTokenManager(ding_config.app_key, ding_config.app_secret, DINGDING_APP_VERSION)
department_manager = DepartmentManager(token_manager)
user_manager = UserManager(token_manager)
role_manager = RoleManager(token_manager)
db_executer = cli_factory(EXECUTERS)(User.objects.get(username='******'))
root_dep = Dept.objects.get(uid=DEPARTMENT_ROOT_UID)
build_department_user_rawdata(db_executer, user_manager, department_manager, DEPARTMENT_ROOT_ID, root_dep)
build_group_rawdata(db_executer, role_manager)
def entrypoint():
'''
override dingding data
'''
ding_config = DingConfig.get_current()
token_manager = AccessTokenManager(ding_config.app_key,
ding_config.app_secret,
DINGDING_APP_VERSION)
department_manager = DepartmentManager(token_manager)
user_manager = UserManager(token_manager)
override_ding_dept(department_manager)
override_ding_role(department_manager)
override_ding_user(user_manager)
def __init__(self, app_key=None, app_secret=None, app_version=None, agent_id=None):
"""
初始化钉钉相关参数
"""
if app_key and app_secret and app_version and agent_id:
self.token_manager = AccessTokenManager(app_key, app_secret, app_version)
else:
ding_config = DingConfig.get_current()
self.token_manager = AccessTokenManager(ding_config.app_key, ding_config.app_secret, DINGDING_APP_VERSION)
self.user_manager = UserManager(self.token_manager)
self.dept_manager = DepartmentManager(self.token_manager)
self.role_manager = RoleManager(self.token_manager)
self.agent_id = AGENT_ID
def __init__(self,
app_key=None,
app_secret=None,
app_version=None,
agent_id=None):
"""
初始化钉钉相关参数
"""
if app_key and app_secret and app_version and agent_id:
self.token_manager = AccessTokenManager(app_key, app_secret,
app_version)
else:
ding_config = DingConfig.get_current()
self.token_manager = AccessTokenManager(
ding_config.app_key, ding_config.app_secret,
TOKEN_FROM_APPKEY_APPSECRET)
self.user_manager = UserManager(self.token_manager)
self.dept_manager = DepartmentManager(self.token_manager)
self.role_manager = RoleManager(self.token_manager)