Functions-d 1.2.4__tar.gz → 1.27__tar.gz

{functions_d-1.2.4 → functions_d-1.27}/Functions_d/Functions_d.py

@@ -148,6 +148,7 @@ class DataProcessingAndMessaging:
         self.logger.info('\n' * 10)
 
     # -------------------------- 企业微信消息发送 --------------------------
+    # 仅展示修改后的 uxin_wx 方法，其余代码不变
     def uxin_wx(self, name, message, mentioned_list=None):
         sender = WeComMsg.WeChatWorkSender(self.corpid, self.corpsecret, self.agentid)
         try:
@@ -158,6 +159,43 @@ class DataProcessingAndMessaging:
                 target_type = "群聊（Webhook）" if name.startswith("https://") else "单个用户"
                 self.logger.info(f"开始向{target_type}发送消息，目标：{name}")
 
+            # ========== 新增：Markdown消息识别与处理 ==========
+            # 识别规则：message以特定标识开头（如「MD:」），或显式传入markdown类型
+            is_markdown = False
+            md_content = ""
+            if isinstance(message, dict) and message.get("type") == "markdown":
+                # 支持字典格式传参：{"type": "markdown", "content": "markdown内容"}
+                is_markdown = True
+                md_content = message.get("content", "")
+            elif isinstance(message, str) and message.startswith("MD:"):
+                # 支持字符串前缀标识："MD:### 标题\n内容"
+                is_markdown = True
+                md_content = message[3:].strip()  # 去掉前缀"MD:"
+
+            if is_markdown:
+                # 处理Markdown消息
+                if isinstance(name, str) and name.startswith("https://"):
+                    # 群聊Webhook发送Markdown
+                    self.logger.info(f"发送群聊Markdown消息：内容={md_content}")
+                    result = sender.send_markdown_to_group(name, md_content)
+                else:
+                    # 个人/多用户发送Markdown
+                    receivers = name if isinstance(name, list) else [name]
+                    self.logger.info(f"发送个人Markdown消息：内容={md_content}，接收者：{receivers}")
+                    result = sender.send_markdown(receivers, md_content)
+
+                msg_id = result.get('msgid', '未知')
+                self.logger.info(
+                    f"Markdown消息发送结果：{'成功' if result.get('errcode') == 0 else '失败'}，错误信息：{result.get('errmsg')}，消息ID：{msg_id}")
+
+                if result.get('errcode') == 0:
+                    print(f"给 {target_type} 的Markdown消息发送成功，消息ID：{msg_id}")
+                else:
+                    print(
+                        f"给 {target_type} 的Markdown消息发送失败，错误码：{result.get('errcode')}，错误信息：{result.get('errmsg')}，消息ID：{msg_id}")
+                return
+
+            # ========== 原有逻辑（文本/文件/图片）保持不变 ==========
             if isinstance(name, str) and name.startswith("https://"):
                 if isinstance(message, str) and message.endswith(('.xlsx', '.docx', '.pdf', '.txt')) and os.path.isfile(
                         message):
@@ -319,26 +357,116 @@ class DataProcessingAndMessaging:
         return result
 
     # -------------------------- 发送邮件（旧版） --------------------------
-    def sende_email(self, name, contact_name, title, rec, file, cc=False, bcc=None):
-        yag = yagmail.SMTP(user='cc_yingxiao@xin.com', password='cw46pfeznNQx', host='mail.xin.com', port='587',
-                           smtp_ssl=False, smtp_starttls=True)
-        contents = f'{name} 好：\n \n ' \
-                   f'附件为{title}，请查收！\n \n' \
-                   f'如有疑问请联系{contact_name}，谢谢~'
+    def sende_email(self, name, contact_name, title, rec, file=None, cc=None, bcc=None, remarks=None):
+        """
+        简洁版邮件发送（支持重复调用，自动过滤无效参数）
+        :param name: 收件人称呼（如"各位"）
+        :param contact_name: 联系人姓名（如"董养"）
+        :param title: 邮件主题
+        :param rec: 收件人邮箱（单个字符串或列表）
+        :param file: 附件路径（单个字符串、列表，或None）
+        :param cc: 抄送邮箱（单个字符串、列表，或None）
+        :param bcc: 密送邮箱（单个字符串、列表，或None）
+        :param remarks: 附加备注（可选，默认None，传入文本字符串时追加到正文）
+        """
+
+        # 1. 内部工具：格式化邮箱（转列表+过滤无效值）
+        def format_email(emails):
+            if not emails:
+                return None
+            # 单个邮箱转列表，列表直接使用
+            email_list = [emails] if isinstance(emails, str) else emails
+            # 过滤：非空字符串 + 简单格式校验（含@和.）
+            valid_emails = [
+                e.strip() for e in email_list
+                if isinstance(e, str) and e.strip() and '@' in e.strip() and '.' in e.strip()
+            ]
+            return valid_emails if valid_emails else None
+
+        # 2. 格式化所有邮箱参数
+        to_emails = format_email(rec)
+        cc_emails = format_email(cc)
+        bcc_emails = format_email(bcc)
+
+        # 3. 核心参数校验（提前报错，避免无效连接）
+        if not to_emails:
+            err_msg = "邮件发送失败：无有效收件人邮箱"
+            self.logger.error(err_msg)
+            raise ValueError(err_msg)
+        if not title.strip():
+            err_msg = "邮件发送失败：邮件主题不能为空"
+            self.logger.error(err_msg)
+            raise ValueError(err_msg)
+
+        # 4. 附件处理（校验存在性，支持单个/列表）
+        attachments = None
+        if file:
+            file_list = [file] if isinstance(file, str) else file
+            attachments = []
+            for f in file_list:
+                f = f.strip()
+                if not os.path.exists(f):
+                    err_msg = f"邮件发送失败：附件不存在 -> {f}"
+                    self.logger.error(err_msg)
+                    raise FileNotFoundError(err_msg)
+                attachments.append(f)
+
+        # 5. SMTP固定配置（重复调用无需修改）
+        smtp_conf = {
+            "user": 'cc_yingxiao@xin.com',
+            "password": 'cw46pfeznNQx',
+            "host": 'mail.xin.com',
+            "port": 587,
+            "smtp_ssl": False,
+            "smtp_starttls": True
+        }
+
+        # 6. 邮件内容（新增remarks逻辑：有值则追加到"请查收！"下方，保持缩进一致）
+        email_content = f"{name} 好：\n附件为《{title}》，请查收！"
+        if remarks and isinstance(remarks, str) and remarks.strip():
+            formatted_remarks = remarks.strip().replace('\n', '\n')  # 处理多行备注
+            email_content += f"\n{formatted_remarks}"
+        email_content += f"\n\n如有疑问请联系{contact_name}，谢谢~"
+
         try:
-            if cc and bcc:
-                yag.send(rec, title, contents, file, cc, bcc)
-            elif cc:
-                yag.send(rec, title, contents, file, cc)
-            elif bcc:
-                yag.send(rec, title, contents, file, bcc)
-            else:
-                yag.send(rec, title, contents, file)
-            self.logger.info(f'邮件主题：{title} \n邮件附件：{file} 发送完成')
-            print(f'邮件主题：{title} \n邮件附件：{file} 发送完成')
+            # 上下文管理器：自动关闭连接，重复调用不泄露资源
+            with yagmail.SMTP(**smtp_conf) as yag:
+                yag.send(
+                    to=to_emails,
+                    subject=title.strip(),
+                    contents=email_content,
+                    attachments=attachments,
+                    cc=cc_emails,
+                    bcc=bcc_emails
+                )
+            # 简洁日志：关键信息+统计，便于排查
+            log_msg = f"邮件发送成功 | 主题：{title.strip()} | 收件人：{len(to_emails)}人"
+            if cc_emails:
+                log_msg += f" | 抄送：{len(cc_emails)}人"
+            if bcc_emails:
+                log_msg += f" | 密送：{len(bcc_emails)}人"
+            if attachments:
+                log_msg += f" | 附件：{len(attachments)}个"
+            if remarks and remarks.strip():
+                log_msg += f" | 包含备注：{remarks.strip()[:20]}..."  # 日志显示备注前20字（避免过长）
+            self.logger.info(log_msg)
+            print(log_msg)
+
+        # 7. 分类异常捕获（易排查问题）
+        except yagmail.SMTPAuthenticationError:
+            err_msg = "邮件发送失败：SMTP账号/密码/授权码错误"
+            self.logger.error(err_msg)
+            raise ValueError(err_msg)
+        except yagmail.SMTPConnectionError:
+            err_msg = "邮件发送失败：SMTP服务器连接失败（检查host/port/防火墙）"
+            self.logger.error(err_msg)
+            raise ConnectionError(err_msg)
+        except FileNotFoundError:
+            raise  # 附件错误已提前处理，直接抛出
         except Exception as e:
-            self.logger.error(f"邮件发送失败: {str(e)}", exc_info=True)
-            raise
+            err_msg = f"邮件发送失败：{str(e)}"
+            self.logger.error(err_msg, exc_info=True)
+            raise RuntimeError(err_msg)
 
     # 新增：提取Hive核心错误信息方法
     def extract_core_hive_error(self, err_msg):
@@ -938,15 +1066,55 @@ class DataProcessingAndMessaging:
                 self._wechat_doc_log(error_msg)
                 raise Exception(error_msg)
             docid = result.get("docid")
-            self._wechat_doc_log(f"智能表格：{sheet_name} 创建成功，docid={docid}")
-            print(f"智能表格：{sheet_name} 创建成功，docid={docid}")
-            return {"智能表格": sheet_name, "docid": docid, "url": result.get("url")}
+            # ========== 新增：添加创建时间 ==========
+            # 获取当前时间（北京时间，格式：YYYY-MM-DD HH:MM:SS）
+            create_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+            # ========== 核心新增：创建成功后自动查询子sheet ==========
+            self._wechat_doc_log(f"开始查询新创建表格[{docid}]的子表信息")
+            # 然后修改wx_create_table中调用wx_get_sheets的代码，传入send_notice=False：
+            sheet_list = self.wx_get_sheets(docid, send_notice=False)
+
+            # ========== 构造包含子sheet的完整消息内容 ==========
+            # 基础信息
+            msg_content = f"✅ 智能表格创建成功\n表格名称：{sheet_name}\n创建时间：{create_time}\ndocid：{docid}\n\n子表信息："
+            # 子sheet信息
+            if sheet_list:
+                for idx, sheet in enumerate(sheet_list, 1):
+                    sheet_visibility = '可见' if sheet['is_visible'] else '不可见'
+                    msg_content += f"\n{idx}. ID: {sheet['sheet_id']} | 标题: {sheet['title']} | 类型: {sheet['type']} | 可见性: {sheet_visibility}"
+            else:
+                msg_content += "\n暂无子表信息"
+
+            self._wechat_doc_log(
+                f"智能表格：{sheet_name} 创建成功，docid={docid}，创建时间={create_time}，子表数量：{len(sheet_list)}")
+            print(f"智能表格：{sheet_name} 创建成功，docid={docid}，创建时间={create_time}")
+            print(f"该表格的子表信息：")
+            if sheet_list:
+                for idx, sheet in enumerate(sheet_list, 1):
+                    sheet_visibility = '可见' if sheet['is_visible'] else '不可见'
+                    print(
+                        f"  {idx}. ID: {sheet['sheet_id']} | 标题: {sheet['title']} | 类型: {sheet['type']} | 可见性: {sheet_visibility}")
+            else:
+                print("  暂无子表信息")
+
+            # 调用uxin_wx发送包含子sheet的完整消息
+            self.uxin_wx('dongyang', msg_content)
+
+            return {
+                "智能表格": sheet_name,
+                "docid": docid,
+                "url": result.get("url"),
+                "创建时间": create_time,
+                "子表列表": sheet_list  # 返回值新增子sheet列表
+            }
         except Exception as e:
             self._wechat_doc_log(f"创建异常: {str(e)}")
             raise
 
     # -------------------------- 企微文档：查询子表 --------------------------
-    def wx_get_sheets(self, docid):
+    # 修改wx_get_sheets方法的定义，新增send_notice参数
+    def wx_get_sheets(self, docid, send_notice=True):
         self._wechat_doc_log(f"查询文档[{docid}]的子表信息")
         self._wechat_doc_refresh_token_if_needed()
 
@@ -962,10 +1130,30 @@ class DataProcessingAndMessaging:
         if result.get("errcode") != 0:
             error_msg = f"查询子表失败: {result.get('errmsg')}"
             self._wechat_doc_log(error_msg)
+            # 失败时仍发送通知
+            fail_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            fail_content = f"⚠️ 企微文档子表查询失败\n查询时间：{fail_time}\n文档ID：{docid}\n错误信息：{result.get('errmsg')}"
+            self.uxin_wx('dongyang', fail_content)
             raise Exception(error_msg)
 
         sheet_list = result.get("sheet_list", [])
+        query_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         self._wechat_doc_log(f"查询文档[{docid}]子表完成，共找到{len(sheet_list)}个子表")
+
+        # 构造通知内容
+        msg_content = f"✅ 企微文档子表查询完成\n查询时间：{query_time}\n文档ID：{docid}\n子表总数：{len(sheet_list)}\n\n子表详细信息："
+        if sheet_list:
+            for idx, sheet in enumerate(sheet_list, 1):
+                sheet_visibility = '可见' if sheet['is_visible'] else '不可见'
+                msg_content += f"\n{idx}. ID: {sheet['sheet_id']} | 标题: {sheet['title']} | 类型: {sheet['type']} | 可见性: {sheet_visibility}"
+        else:
+            msg_content += "\n暂无子表信息"
+
+        # 仅当send_notice为True时发送通知
+        if send_notice:
+            self.uxin_wx('dongyang', msg_content)
+
+        # 原有日志和打印逻辑
         for sheet in sheet_list:
             sheet_info = (f"子表信息 - ID: {sheet['sheet_id']}, 标题: {sheet['title']}, "
                           f"类型: {sheet['type']}, 可见性: {'可见' if sheet['is_visible'] else '不可见'}")
@@ -1119,8 +1307,12 @@ class DataProcessingAndMessaging:
             self._wechat_doc_log(error_msg)
             raise Exception(error_msg)
 
-        self._wechat_doc_log(f"文档删除成功：docid={docid}")
-        print(f"文档删除成功：docid={docid}")
+        # 获取当前时间（北京时间，格式：YYYY-MM-DD HH:MM:SS）
+        create_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        msg_content = f"企微文档删除成功：docid={docid} \n 删除时间：{create_time}"
+        self._wechat_doc_log(msg_content)
+        print(msg_content)
+        self.uxin_wx('dongyang', msg_content)
         return True
 
     # -------------------------- 辅助方法：检查文件是否被占用 --------------------------

{functions_d-1.2.4 → functions_d-1.27}/Functions_d/__init__.py

@@ -1,4 +1,4 @@
-# WeComMsg/__init__.py
+# Functions_d/__init__.py
 
 # 从核心代码文件中导入需要对外暴露的类/函数
 from .Functions_d import DataProcessingAndMessaging

{functions_d-1.2.4 → functions_d-1.27}/Functions_d.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Functions_d
-Version: 1.2.4
+Version: 1.27
 Summary: 包含数据处理、Hive交互、企业微信消息发送、Excel操作等功能的工具类库
 Author: DongYang
 Author-email: 649898871@qq.com

{functions_d-1.2.4 → functions_d-1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Functions_d
-Version: 1.2.4
+Version: 1.27
 Summary: 包含数据处理、Hive交互、企业微信消息发送、Excel操作等功能的工具类库
 Author: DongYang
 Author-email: 649898871@qq.com

{functions_d-1.2.4 → functions_d-1.27}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = Functions_d
-version = 1.2.4
+version = 1.27
 author = DongYang
 author_email = 649898871@qq.com
 description = 包含数据处理、Hive交互、企业微信消息发送、Excel操作等功能的工具类库