逻辑

文件路径

同目录下“.jianou”文件，多个文件会提示，默认使用第一个（后续考虑改为修改时间最新的）。

配置文件设计

为了便于前台操作（复选框），改为MarkDown语法的编辑页。

显示

image.png

运行结果

image.png

数据合法有效时

image.png
image.png

!数据合法有效时

image.png
image.png
主要不同点在于，如果字符串类型的Key没有对应个数的Value，会将Value全部设为空字符串。

代码

🐍Python🐍

  1. # encoding = utf-8
  2. import os
  3. import pathlib
  4. import re
  5. from addict import Dict
  7. msgStart = r"——————————Python脚本：Start——————————"
  9. strN = '\n'
  10. strMarkCheck = '- ['
  12. title2 = '\n## '
  13. title3 = '### '
  15. strAnchorEnd = "\"></a>"
  17. strBR = r"<br />"
  18. strNull = ""
  20. msgEnd = r"——————————Python脚本：End——————————"
  23. # 读取配置文件字符串
  24. def ReadMarkDownText():
  25.     pathSelf = os.getcwd()
  26.     print("当前工作目录：", pathSelf)
  28.     listFile = list(pathlib.Path(pathSelf).glob('*.jianou'))
  29.     nJianou = len(listFile)
  30.     if nJianou == 0:
  31.         print("无配置文件“*.jianou”")
  32.     else:
  33.         pathConf = listFile[0]
  34.         if nJianou == 1:
  35.             print("使用配置文件：", pathConf)
  36.         else:
  37.             print("有多个文件：", listFile)
  38.             print("默认使用第一个：", pathConf)
  40.         with open(pathConf, 'r+', encoding='utf-8') as f:
  41.             textMarkDown = f.read()
  42.         return textMarkDown
  45. def ExtractConf(textMarkDown):
  46.     listLevel2 = textMarkDown.split(title2)
  47.     # print("2级标题列表：", listLevel2)
  49.     listValueAll = []
  50.     for strLevel2 in listLevel2:
  51.         # print("2级标题字符串：", strLevel2)
  53.         if title3 in strLevel2:
  54.             strTypeLevel2 = strLevel2.split(strN)[0]
  55.             # print("2级标题类型：", strTypeLevel2)
  57.             listValue2Filtered = []
  58.             listValue2 = re.split(r"\n|<br />", strLevel2)
  59.             # print("2级标题的值：", listValue2)
  60.             for iValue in listValue2:
  61.                 if strAnchorEnd in iValue:
  62.                     pass
  63.                 elif iValue == strNull:
  64.                     # 输入输出目录可能为空
  65.                     pass
  66.                 elif "//" in iValue:
  67.                     # 备注
  68.                     pass
  69.                 else:
  70.                     if strBR in iValue:
  71.                         iValue = iValue.strip(strBR)
  72.                     listValue2Filtered.append(iValue)
  73.             # print("过滤后的Level2：", listValue2Filtered)
  74.             listValueAll.append(listValue2Filtered)
  76.     # print("3级标题列表：", listValueAll)
  78.     listKey = []
  79.     listValue = []
  80.     dictConfAll = { }
  81.     for iLevel2 in listValueAll:
  82.         # print(iLevel2)
  83.         strTypeLevel2 = iLevel2[0]
  84.         listValueLevel2 = iLevel2[1:]
  86.         listConfKey = []
  87.         listConfValue = []
  88.         for iValueLevel2 in listValueLevel2:
  89.             if title3 in iValueLevel2:
  90.                 strLevel3 = iValueLevel2.split(title3)[1:][0].strip()
  91.                 # print(strLevel3)
  92.                 listConfKey.append(strLevel3)
  94.             elif strMarkCheck in iValueLevel2:
  95.                 if '[x]' in iValueLevel2:
  96.                     listConfValue.append(True)
  97.                 elif '[ ]' in iValueLevel2:
  98.                     listConfValue.append(False)
  100.             elif iValueLevel2 == strNull:
  101.                 # 如：文件夹
  102.                 listConfValue.append(iValueLevel2)
  104.             else:
  105.                 if CaseStrIn(iValueLevel2, "True"):
  106.                     listConfValue.append(True)
  107.                 elif CaseStrIn(iValueLevel2, "False"):
  108.                     listConfValue.append(False)
  109.                 else:
  110.                     # 如：文件夹
  111.                     listConfValue.append(iValueLevel2)
  113.         # print(listConfKey)
  114.         # print(listConfValue)
  115.         listKey.append(listConfKey)
  116.         listValue.append(listConfValue)
  118.         if len(listConfKey) != len(listConfValue):
  119.             print("字典的Key列表和Value列表个数不一致:")
  120.             print(listConfKey)
  122.             listConfValue = []
  123.             for i in range(len(listConfKey)):
  124.                 listConfValue.append(strNull)
  125.             print("Value列表值被置为空字符串")
  127.         zipConf = zip(listConfKey, listConfValue)
  128.         # print(listConf)
  129.         dictConf = dict(zipConf)
  130.         # print(dictConf)
  131.         dictConfAll.update(dictConf)
  133.     # print(dictConfAll)
  134.     return dictConfAll
  137. def CaseStrIn(str2CMP, strBeCMP):
  138.     if strBeCMP.lower() in str2CMP.lower():
  139.         bCMP = True
  140.     else:
  141.         bCMP = False
  142.     return bCMP
  145. def DictObj():
  146.     textConf = ReadMarkDownText()
  147.     dictConf = ExtractConf(textConf)
  148.     objCof = Dict(dictConf)
  149.     # print(objCof)
  150.     return objCof
  153. if __name__ == '__main__':
  154.     print(msgStart)
  156.     objCof = DictObj()
  158.     print(msgEnd)

📝MarkDown📝

有一些锚点，懒得删了，在代码里过滤了

  1. # 配置文件
  2. **命名**：*.jianou<br />**格式**：以MarkDown文本格式<br />**路径**：exe同目录<br />**选项**：复选框以及大小写不敏感的True和False
  3. <a name="CP9TX"></a>
  4. ## 显示：窗口消息
  5. <a name="xAWsM"></a>
  6. ### 单个组织
  8. - [ ] <br />
  9. <a name="ASPlp"></a>
  10. ### 全部组织
  12. - [ ] <br />
  13. <a name="MeEtg"></a>
  14. ## 异常提醒
  15. <a name="Dm1HK"></a>
  16. ### Table数据异常
  18. - [x] <br />
  19. <a name="Nn8vz"></a>
  20. ### Code非常见类型
  22. - [x] <br />
  23. <a name="V2qh5"></a>
  24. ### Code有异常符号
  26. - [x] <br />
  27. <a name="iFsK0"></a>
  28. ## 导出文件
  29. <a name="b3NRZ"></a>
  30. ### IoCs表格
  32. - [ ] <br />
  33. <a name="qBbRD"></a>
  34. ### 完成后打开IoCs表格
  36. - [ ] <br />
  37. <a name="fLncF"></a>
  38. ### 规则和代码
  40. - [x] <br />
  41. <a name="wCdf9"></a>
  42. ### 原文链接
  44. - [x] <br />
  45. <a name="WRv52"></a>
  46. ### 所有组织的IoCs数量统计
  48. - [x] <br />
  49. <a name="pM7jE"></a>
  50. ### 所有需要提取IoCs的MarkDown文件路径
  51. 支持大写的Bool值：TRUE
  52. <a name="xQn4E"></a>
  53. ### 代码有异常符号
  55. - [x] 支持复选框
  56. <a name="igKf7"></a>
  57. ### 不符合IoCs标题的文件
  58. 也支持小写的Bool值：false
  59. <a name="Ks1xA"></a>
  60. ## 文件夹
  61. <a name="fYnvY"></a>
  62. ### 输入
  63. C:\Users\Administrator\Desktop
  64. <a name="wzTIn"></a>
  65. ### 输出
  66. C:\Users\Administrator\Desktop\APT情报输出<br />// 缺省：exe同目录

参考

虽然没有采用，但是下次比较急的时候我应该会直接用现成的库和其对应的配置文件格式。这次自己构建自己的方式也是当练练Python语法。

python读取配置文件&&简单封装