更新钉钉文档封装好的代码

import json
import requests

class OperateKnowledgeBaseExcel():
    robot_code = ''       #机器人
    agent_id = ''         
    app_key = ''          #机器人密钥  获取方法见我的其他博客
    app_secret = ''       #机器人密钥

    def __init__(self,union_id, workbook_id, worksheet_id=''):
        self.union_id = union_id
        self.workbook_id = workbook_id  # 获取方式:... > 文档信息 > 表格ID
        self.worksheet_id =  worksheet_id # 获取方式:self.get_workbook_sheet_names() 拿到所有sheet名称和id
        if not self.union_id:
            raise Exception('请先设置union_id')

        self.access_token = self.get_access_token()

    def get_access_token(self):
        """ 获取token: https://open.dingtalk.com/document/orgapp/obtain-orgapp-token """
        headers = {'Content-Type': "application/x-www-form-urlencoded"}
        url = f"https://oapi.dingtalk.com/gettoken/?appkey={self.app_key}&appsecret={self.app_secret}"
        resp = requests.get(url, headers=headers)
        # print(resp.json()['access_token'])
        return resp.json()['access_token']

    def get_workbook_sheet_names(self):
        """
        获取工作表信息,包含工作表名称、工作表id
        开发者文档:https://open.dingtalk.com/document/orgapp/obtain-all-worksheets
        """
        url = f'https://oapi.dingtalk.com/v1.0/doc/workbooks/{self.workbook_id}/sheets?operatorId={self.union_id}'
        headers = {
            'Content-Type': "application/json",
            'Host': 'api.dingtalk.com',
            'x-acs-dingtalk-access-token': self.access_token
        }
        resp = requests.get(url, headers=headers)
        # print(resp.json())
        return resp.json()

    def get_worksheet_describe(self):
        """
        获取工作表的描述
        开发者文档:https://open.dingtalk.com/document/orgapp/obtain-worksheet-properties
        """
        url = f'https://oapi.dingtalk.com/v1.0/doc/workbooks/{self.workbook_id}/sheets/{self.worksheet_id}?operatorId={self.union_id}'
        headers = {
            'Content-Type': "application/json",
            'Host': 'api.dingtalk.com',
            'x-acs-dingtalk-access-token': self.access_token
        }
        resp = requests.get(url, headers=headers)
        # print(resp.json())
        return resp.json()

    def get_all_cells_area(self):
        """ 获取表格的全部区域,如:A1:D4 """
        worksheet_describe = self.get_worksheet_describe()
        last_row = worksheet_describe['lastNonEmptyRow']
        last_column = worksheet_describe['lastNonEmptyColumn']
        range_address = f'A1:{get_column_letter(last_column + 1)}{last_row + 1}'
        return range_address

    def read(self, range_address=False):
        """
        读取工作表数据
        开发者文档:https://open.dingtalk.com/document/orgapp/get-cell-properties
        """
        if not range_address:  # 如果没有指定读取区域,则读取整个工作表
            range_address = self.get_all_cells_area()
        url = f'https://oapi.dingtalk.com/v1.0/doc/workbooks/{self.workbook_id}/sheets/{self.worksheet_id}/ranges/{range_address}?select=values&operatorId={self.union_id}'
        headers = {
            'Content-Type': "application/json",
            'Host': 'api.dingtalk.com',
            'x-acs-dingtalk-access-token': self.access_token
        }
        resp = requests.get(url, headers=headers)
        data = pd.DataFrame(columns=resp.json()['values'][0], data=resp.json()['values'][1:])
        data.replace('', pd.NA, inplace=True)
        return data

    def update(self, range_address, values):
        """
        更新工作表指定区域的数据
        开发者文档:https://open.dingtalk.com/document/orgapp/get-cell-properties
        :param range_address: 更新区域,如:A2:B3
        :param values: 更新的数据,如:[['张三', 18], ['李四', 19]]
        """
        url = f'https://oapi.dingtalk.com/v1.0/doc/workbooks/{self.workbook_id}/sheets/{self.worksheet_id}/ranges/{range_address}?operatorId={self.union_id}'
        headers = {
            'Content-Type': "application/json",
            'Host': 'api.dingtalk.com',
            'x-acs-dingtalk-access-token': self.access_token
        }
        data = {"values": values}
        resp = requests.put(url, data=json.dumps(data), headers=headers)
        print("更新工作表区域:", resp.json())

    def delete(self, range_address=False):
        """
        清空工作表指定区域的数据,仅清除数据,不清除格式
        开发者文档:https://open.dingtalk.com/document/orgapp/get-cell-properties
        """
        if not range_address:  # 如果没有指定读取区域,则清空整个工作表
            range_address = self.get_all_cells_area()

        url = f'https://oapi.dingtalk.com/v1.0/doc/workbooks/{self.workbook_id}/sheets/{self.worksheet_id}/ranges/{range_address}/clearData?operatorId={self.union_id}'
        headers = {
            'Content-Type': "application/json",
            'Host': 'api.dingtalk.com',
            'x-acs-dingtalk-access-token': self.access_token
        }
        resp = requests.post(url, headers=headers)
        print('清空工作表区域:', resp.json())


userid = ''    #钉钉账号
workid = ''    #文档id

def worksheetinsert(df,number=1,worksheetid=''):
    if df.empty:
        return
    # 转换列索引为Excel的列名称
    def col_num_to_letter(col_num):
        string = ""
        while col_num > 0:
            col_num, remainder = divmod(col_num-1, 26)
            string = chr(65 + remainder) + string
        return string
    # 获取行数和列数
    num_rows, num_cols = df.shape

    # 因为包括列名,所以行数需要加1
    range = f"A{number}:{col_num_to_letter(num_cols)}{num_rows + number}"
    print(range)

    # 将数据转换为列表形式
    values = df.values.tolist()
    values = [df.columns.tolist()] + [[str(item) for item in row] for row in values]

    # 构造字典对象
    data = values
    try:
        temp = OperateKnowledgeBaseExcel(userid,workid,worksheetid).read()
        OperateKnowledgeBaseExcel(userid,workid,worksheetid).delete()
    except:
        pass
    OperateKnowledgeBaseExcel(userid,workid,worksheetid).update(range,data)

df = 

excelsheet_json = OperateKnowledgeBaseExcel(userid,workid).get_workbook_sheet_names()
excelsheet = {item['name']: item['id'] for item in excelsheet_json['value']}


#指定表sheet名插入
worksheetinsert(df,number=1,worksheetid=excelsheet['XXX'])
  1. 获取访问令牌:通过设置机器人的密钥和密钥获取访问令牌。

  2. 获取工作表信息:获取工作表的名称和ID。

  3. 获取工作表描述:获取工作表的描述信息。

  4. 获取全部单元格区域:获取工作表的全部区域。

  5. 读取数据:读取指定区域的数据,并将其转换为DataFrame格式返回。

  6. 更新数据:更新指定区域的数据。

  7. 清空数据:清空指定区域的数据,但保留格式。

  8. 插入数据到指定表格:将DataFrame格式的数据插入到指定的工作表中。

你可能感兴趣的:(钉钉,python,开发语言)