初始提交

function/func_chengyu.py

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+
+import os
+import random
+import logging
+
+import pandas as pd
+
+# 获取模块级 logger
+logger = logging.getLogger(__name__)
+
+
+class Chengyu(object):
+    def __init__(self) -> None:
+        root = os.path.dirname(os.path.abspath(__file__))
+        self.df = pd.read_csv(f"{root}/chengyu.csv", delimiter="\t")
+        self.cys, self.zis, self.yins = self._build_data()
+
+    def _build_data(self):
+        df = self.df.copy()
+        df["shouzi"] = df["chengyu"].apply(lambda x: x[0])
+        df["mozi"] = df["chengyu"].apply(lambda x: x[-1])
+
+        df["shouyin"] = df["pingyin"].apply(lambda x: x.split(" ")[0])
+        df["moyin"] = df["pingyin"].apply(lambda x: x.split(" ")[-1])
+
+        cys = dict(zip(df["chengyu"], df["moyin"]))
+        zis = df.groupby("shouzi").agg({"chengyu": set})["chengyu"].to_dict()
+        yins = df.groupby("shouyin").agg({"chengyu": set})["chengyu"].to_dict()
+
+        return cys, zis, yins
+
+    def isChengyu(self, cy: str) -> bool:
+        return self.cys.get(cy, None) is not None
+
+    def getNext(self, cy: str, tongyin: bool = True) -> str:
+        """获取下一个成语
+            cy: 当前成语
+            tongyin: 是否允许同音字
+        """
+        zi = cy[-1]
+        ansers = list(self.zis.get(zi, {}))
+        try:
+            ansers.remove(cy)  # 移除当前成语
+        except Exception as e:
+            pass  # Just ignore...
+
+        if ansers:
+            return random.choice(ansers)
+
+        # 如果找不到同字，允许同音
+        if tongyin:
+            yin = self.cys.get(cy)
+            ansers = list(self.yins.get(yin, {}))
+
+        try:
+            ansers.remove(cy)  # 移除当前成语
+        except Exception as e:
+            pass  # Just ignore...
+
+        if ansers:
+            return random.choice(ansers)
+
+        return None
+
+    def getMeaning(self, cy: str) -> str:
+        ress = self.df[self.df["chengyu"] == cy].to_dict(orient="records")
+        if ress:
+            res = ress[0]
+            rsp = res["chengyu"] + "\n" + res["pingyin"] + "\n" + res["jieshi"]
+            if res["chuchu"] and res["chuchu"] != "无":
+                rsp += "\n出处：" + res["chuchu"]
+            if res["lizi"] and res["lizi"] != "无":
+                rsp += "\n例子：" + res["lizi"]
+            return rsp
+        return None
+
+
+cy = Chengyu()
+
+if __name__ == "__main__":
+    # 设置测试用的日志配置
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(asctime)s - %(levelname)s - %(name)s - %(message)s'
+    )
+    
+    answer = cy.getNext("便宜行事")
+    logger.info(answer)