itam-assistant 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

main/lark_client.py

@@ -0,0 +1,159 @@
+import json
+
+import lark_oapi as lark
+from lark_oapi.api.sheets.v3 import *
+
+
+# SDK 使用说明: https://open.feishu.cn/document/uAjLw4CM/ukTMukTMukTM/server-side-sdk/python--sdk/preparations-before-development
+# 以下示例代码默认根据文档示例值填充，如果存在代码问题，请在 API 调试台填上相关必要参数后再复制代码使用
+
+class LarkdocsClient:
+    def __init__(self):
+        """
+        初始化 Client 实例,tenant_access_token 会在 Client 初始化时自动获取
+        https://github.com/larksuite/oapi-sdk-python
+        """
+        self.app_id = "cli_a48be9fd54a5900d"
+        self.app_secret = "pfiNeWKbkfbRvUunX3TrKdCCnftlUbxl"
+        # 创建 Lark 客户端
+        self.lark_client = lark.Client.builder().app_id(self.app_id).app_secret(self.app_secret).build()
+
+    def get_the_worksheet(self, spreadsheet_token):
+
+        # 获取工作表
+        # 创建client
+        # 使用 user_access_token 需开启 token 配置, 并在 request_option 中配置 token
+        # 构造请求对象
+        try:
+            request: QuerySpreadsheetSheetRequest = QuerySpreadsheetSheetRequest.builder() \
+                .spreadsheet_token(spreadsheet_token) \
+                .build()
+
+            # 发起请求
+            # option = self.lark_client.RequestOption.builder().user_access_token(
+            # "u-hQ.Vthwt5blbznh92YFKhflkkfsAl5R3hW20l5cy07ag").build()
+            response: QuerySpreadsheetSheetResponse = self.lark_client.sheets.v3.spreadsheet_sheet.query(request)
+
+            # 处理失败返回
+            if not response.success():
+                lark.logger.error(
+                    f"client.sheets.v3.spreadsheet_sheet.query failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+                return
+            return response.data
+            # 处理业务结果
+            lark.logger.info(lark.JSON.marshal(response.data, indent=4))
+        except Exception as e:
+            lark.logger.error(f"[lark]get user id by email failed, err: {e}")
+            return None, e
+
+    def get_plaintextcontent(self, ranges, spreadsheet_token, sheet_id):
+        # 创建client
+        # 使用 user_access_token 需开启 token 配置, 并在 request_option 中配置 token
+        # 构造请求对象
+        # 构造请求对象
+        # json_str = "{\"ranges\":[\"459f7e!A1:A1\"]}"
+        body = ranges
+        request: lark.BaseRequest = lark.BaseRequest.builder() \
+            .http_method(lark.HttpMethod.POST) \
+            .uri(f"/open-apis/sheets/v3/spreadsheets/{spreadsheet_token}/sheets/{str(sheet_id)}/values/batch_get_plain") \
+            .token_types({lark.AccessTokenType.TENANT}) \
+            .body(body) \
+            .build()
+
+        # 发起请求
+        response: lark.BaseResponse = self.lark_client.request(request)
+
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.request failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+        # 处理业务结果
+        lark.logger.info(str(response.raw.content, lark.UTF_8))
+        return str(response.raw.content, lark.UTF_8)
+
+
+    def createsheets(self, spreadsheet_token, title):
+        # 创建工作表
+        # json_str = "{\"index\":0,\"title\":\"abc\"}"
+        # 构造请求对象
+        body = title
+        request: lark.BaseRequest = lark.BaseRequest.builder() \
+            .http_method(lark.HttpMethod.POST) \
+            .uri(f"/open-apis/sheets/v3/spreadsheets/{spreadsheet_token}/sheets") \
+            .token_types({lark.AccessTokenType.TENANT}) \
+            .body(body) \
+            .build()
+
+        # 发起请求
+        response: lark.BaseResponse = self.lark_client.request(request)
+
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.request failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+        # 处理业务结果
+        lark.logger.info(str(response.raw.content, lark.UTF_8))
+        return json.loads(response.raw.content)['data']['sheet']
+
+
+    def writesheets(self, spreadsheet_token, sheet_id, data):
+        # 创建client
+        # 使用 user_access_token 需开启 token 配置, 并在 request_option 中配置 token
+        # 构造请求对象
+        # 构造请求对象
+        json_str = "{\"value_ranges\":[{\"range\":\"HEJb8z!C1:C1\",\"values\":[[[{\"text\":{\"text\":\"abc\"},\"type\":\"text\"}]]]}]}"
+        body = json.loads(json_str)
+        body = data
+        request: lark.BaseRequest = lark.BaseRequest.builder() \
+            .http_method(lark.HttpMethod.POST) \
+            .uri(
+            f"/open-apis/sheets/v3/spreadsheets/{spreadsheet_token}/sheets/{sheet_id}/values/batch_update?user_id_type=open_id") \
+            .token_types({lark.AccessTokenType.TENANT}) \
+            .body(body) \
+            .build()
+
+        # 发起请求
+        response: lark.BaseResponse = self.lark_client.request(request)
+
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.request failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+        # 处理业务结果
+        lark.logger.info(str(response.raw.content, lark.UTF_8))
+
+
+if __name__ == '__main__':
+    spreadsheet_token = ""
+    sheets = LarkdocsClient().get_the_worksheet(spreadsheet_token)
+    for i in sheets.sheets:
+
+        column_count = i.grid_properties.column_count
+        row_count = i.grid_properties.row_count
+        sheet_id = i.sheet_id
+        title = i.title
+
+        #print(column_count, row_count, sheet_id, title)
+        json_str = {"ranges": ["459f7e!A1:A1"]}
+        json_str = {"ranges": [sheet_id + "!A1:A" + str(row_count)]}
+        test = LarkdocsClient().get_plaintextcontent(json_str, spreadsheet_token, sheet_id)
+        test = json.loads(test)
+        userinput = test['data']['value_ranges'][0]['values']
+        print(f"表头为{userinput[0]}")
+        i = 0
+        for i in range(1,row_count):
+            if userinput[i][0]:
+                print(userinput[i][0])
+            else:
+                break
+
+
+
+
+

main/logger.py

@@ -0,0 +1,53 @@
+import os
+import logging.config
+
+config = {
+    'version': 1,
+    'disable_existing_loggers': False,
+    'formatters': {
+        'default': {
+            'format': '%(asctime)s %(levelname)s %(message)s'
+        }
+    },
+    'loggers': {
+        'playground': {
+            'handlers': ['log_agent', 'console', 'kafka_agent'],
+            'propagate': False,
+            'level': 'INFO',
+        },
+        'root': {
+            'handlers': ['log_agent', 'console'],
+            'level': 'INFO'
+        }
+    },
+    'handlers': {
+        'log_agent': {
+            'level': 'INFO',
+            'class': 'bytedlogger.StreamLogHandler',
+            'tags': {
+                'customtag': 'devops_boe',
+            }
+        },
+        'kafka_agent': {
+            'level': 'INFO',
+            'class': 'kafkalogger.handlers.KafkaLogHandler',
+            'formatter': 'default',
+        },
+        'console': {
+            'level': 'INFO',
+            'class': 'logging.StreamHandler',
+        }
+    },
+
+}
+if os.getenv("DEBUG") == "true":
+    config["handlers"].pop("kafka_agent")
+    config["handlers"].pop("log_agent")
+    config["loggers"]["playground"]["handlers"].remove("kafka_agent")
+    config["loggers"]["playground"]["handlers"].remove("log_agent")
+    config['loggers']['root']['handlers'].remove("log_agent")
+logging.config.dictConfig(config)
+logger = logging.getLogger('playground')
+logging.getLogger("kafka").setLevel(logging.CRITICAL)
+logging.getLogger("bytedkafka.common.config").setLevel(logging.ERROR)
+logging.getLogger("kafka.producer.kafka").setLevel(logging.ERROR)

itam_assistant-0.1.1.dist-info/METADATA

@@ -1,25 +0,0 @@
-Metadata-Version: 2.4
-Name: itam_assistant
-Version: 0.1.1
-Summary: 新增线上设备型号接口
-Home-page: https://github.com/liujunmeiD/it_assistant
-Author: liujunmeiD
-Author-email: 1105030421@qq.com
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-Requires-Dist: requests>=2.25.1
-Requires-Dist: numpy>=1.20.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-详见https://bytedance.feishu.cn/docs/doccnSOL0o2qtnElftgd2rir2Dc

itam_assistant-0.1.1.dist-info/RECORD

@@ -1,13 +0,0 @@
-it_assistant/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-it_assistant/ailyapp_client.py,sha256=_6cLF80a8SrO7LCujfHe4re7xZyt6GBbDlmApgqbK6A,6326
-it_assistant/config.py,sha256=L_5grm2ZWwg_S1JTOU6XLhQ5GPeYJaBFBLwzP2JA5Ys,5498
-it_assistant/do_ai.py,sha256=FRuP4yZLqBEES0PCLholdHVCbqMRlYssxT631gXRrRk,19953
-it_assistant/intent_detail.py,sha256=xqHELgYmlZIdz6odN1AyRoE5TIPr78zCUKgiqILApOY,16914
-it_assistant/lark_client.py,sha256=ZVDsdt7ucQvJN1-khC3TR-0IMeYaQXfdWF5DxRrkBH4,6665
-it_assistant/logger.py,sha256=W3-tbQ4J-QKSrQSpaNBruzoyMuWYrUkrIY0uBZ1o-lw,1583
-it_assistant/openapi.py,sha256=AJbhdbsB8PXfMjC0hFy2Nnu52dN1EGIXqoloGGD84Ig,29196
-it_assistant/test.py,sha256=qGSihMYmtZkz_vifwEOz579f_wPBJAx5_rz1sABUyo4,581
-itam_assistant-0.1.1.dist-info/METADATA,sha256=NMHaNgIjpYBMzFWdJDoUK-EVWBuPqqnhm6A0myX2A_Q,738
-itam_assistant-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-itam_assistant-0.1.1.dist-info/top_level.txt,sha256=3g528EjzDs06L1brP8ZvVIvWFyS9K5b8gA6eRoRGdt0,13
-itam_assistant-0.1.1.dist-info/RECORD,,