import requests import time from urllib.parse import urlencode from config.using_config import sha256_token, sha256_account from model.UserAuthUtils import get_g_token_sha256 from model.DateUtils import DateUtils from model.DataBaseUtils import * from model.UserAuthUtils import super_auth du = DateUtils() def user_channel(arg): g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = '/erp/api/resource/search/3' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) channel_list = [] for _ in rsp.json()['data']: channel_list.append({'id': _['id'], 'nick_name': _['resourceName']}) return channel_list def user_channel_group(arg): db = MysqlUtils() g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = '/erp/api/resource/search/3' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) channel_dict = {} for _ in rsp.json()['data']: channel_dict[str(_['id'])] = _['resourceName'] user_id = arg.get('user_id') if user_id: sql = f"""SELECT id,name,channel_ids from channel_group where id in (select channel_group_id from user_channel_group_auth where user_id={user_id})""" else: # 不传user_id 返回所有组 sql = "select id,name,channel_ids from channel_group" data = db.quchen_text.getData_json(sql) for i in data: sr = '' for j in i['channel_ids'].split(','): sr += channel_dict[j] + ',' i['channel_names'] = sr return data def user(arg): # 在2021-9月之后不被前端使用 # 返回所有用户 g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = '/erp/api/user/search/3' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) user_list = [] for _ in rsp.json()['data']: user_list.append({'user_id': _['userId'], 'nick_name': _['nickName']}) return user_list def group_add(arg): db = MysqlUtils() name = arg.get("name") ids = arg.get("channel_ids") if name and ids: sql = "insert into channel_group (name,channel_ids) values (%s,%s)" db.quchen_text.execute(sql, (name, ids)) def group_del(arg): db = MysqlUtils() id = arg.get("id") if id: sql = f"delete from channel_group where id={id}" db.quchen_text.execute(sql) def group_auth_add(arg): db = MysqlUtils() user_id = arg.get("user_id") group_id = arg.get("group_id") if user_id and group_id: sql = "insert into user_channel_group_auth (user_id,channel_group_id) values (%s,%s)" db.quchen_text.execute(sql, (user_id, group_id)) def group_auth_del(arg): db = MysqlUtils() user_id = arg.get("user_id") group_id = arg.get("group_id") if user_id and group_id: sql = f"delete from user_channel_group_auth where user_id={user_id} and channel_group_id={group_id}" db.quchen_text.execute(sql) def get_channel(arg): db = MysqlUtils() user_id = arg.get('user_id') name_cahnnel = arg.get('name') pitcher_id = arg.get('pitcher_id') # 获取公众号列表 g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = '/erp/api/resource/search/3' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) channel_list_all = [] channel_dict = {} for _ in rsp.json()['data']: # channel_list.append((_['id'], _['nickName'])) if name_cahnnel and name_cahnnel in _['resourceName']: channel_list_all.append({'id': _['id'], 'nick_name': _['resourceName']}) if not name_cahnnel: channel_list_all.append({'id': _['id'], 'nick_name': _['resourceName']}) channel_dict[_['id']] = _['resourceName'] if user_id in super_auth() and not pitcher_id: # return channel_list_all[:100] return channel_list_all select_id = user_id # 通过pitcher_id,来添加 if pitcher_id: select_id = pitcher_id # 普通权限------java,获取本人对应拥有的公众号 g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = f'/erp/api/resource/subUserResourceList/3/{user_id}/投手,运营' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) # 防止urlencode自动换行 params_other_urlencode = params_other_urlencode.replace('%0A', '') url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) data1 = [] for _ in rsp.json()['data']: # for可替换这个，下面的for循环未找到原因为何这样写，暂时以下面为主 # if name_cahnnel in _['resourceName']: # if name_cahnnel and name_cahnnel in _['resourceName']: # print("1----------->" + _['resourceName']) # data1.append({'id': _['id'], 'nick_name': _['resourceName']}) # if not name_cahnnel: # print("2----------->" + _['resourceName']) # data1.append({'id': _['id'], 'nick_name': _['resourceName']}) if name_cahnnel and name_cahnnel in _['resourceName']: if not name_cahnnel in _['resourceName']: data1.append({'id': _['id'], 'nick_name': _['resourceName']}) elif name_cahnnel in _['nickName']: data1.append({'id': _['id'], 'nick_name': _['resourceName']}) if not name_cahnnel: data1.append({'id': _['id'], 'nick_name': _['resourceName']}) # # 加上配置的号 # sql2 = f"""select GROUP_CONCAT(channel_ids) from user_channel_group_auth a # left join channel_group b on a.channel_group_id=b.id # where user_id={select_id}""" # data2 = db.quchen_text.getOne(sql2) # data3 = [] # if data2: # data2_2 = tuple(data2.split(',')) # for _ in data2_2: # data3.append({'id': int(_), 'nick_name': channel_dict[int(_)]}) # data4=[] # data4 = data1 + data3 return data1 def get_pitcher(arg): user_id = arg.get('user_id') if user_id in super_auth(): user_list = [] g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = '/erp/api/user/search/3' params_others = '?resourceTags=投手' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + params_others + '&' + params_other_urlencode rsp = requests.get(url) for _ in rsp.json()['data']: user_list.append({'user_id': _['userId'], 'nick_name': _['nickName']}) return user_list user_list = [] g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = f'/erp/api/user/subUser/3/{user_id}' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) for _ in rsp.json()['data']: user_list.append({'user_id': _['userId'], 'nick_name': _['nickName']}) return user_list def get_user_name_by_id(user_id): g_time = int(time.time()) base_url = 'https://api.zanxiangnet.com' params = f'/erp/api/user/subUser/3/{user_id}' g_token = get_g_token_sha256(g_time, params) params_other_urlencode = urlencode({'g_time': g_time, 'g_token': g_token, 'g_account': sha256_account}) url = base_url + params + '?' + params_other_urlencode rsp = requests.get(url) nick_name = None for _ in rsp.json()['data']: if str(_['userId']) == str(user_id): nick_name = _['nickName'] return nick_name if __name__ == '__main__': print(get_channel({'user_id': 78})) # print(get_user_name_by_id({'user_id':78,'name':''})) # print(get_user_name_by_id(78))