Merge pull request #3 from yanshil/main

add level threshold while saving

Author: ProblemFactory

main.py

@@ -299,6 +299,7 @@ def to_stop():
 input(f'自动判断圣遗物背包有{art_rows}行{art_cols}列，请务必确认是否正确！！错误请退出并更改分辨率后尝试')
 input('运行期间请保持原神在前台，请勿遮挡窗口或操作鼠标，按鼠标中键停止。按回车继续')
 input('开始后将尝试自动对齐第一行以方便识别，若对齐结果有误，请立刻按中键停止。按回车开始执行')
+level_threshold = input('请输入圣遗物等级阈值(0-20)(比如：16，则仅将保存16级及以上的圣遗物信息)。直接按回车则默认保存所有圣遗物信息。')
 print('程序将于5秒后自动开始运行，若此条提示显示时未自动切换到原神窗口，请手动点击原神窗口切到前台')
 
 keyboard.press('alt')
@@ -332,11 +333,22 @@ def to_stop():
 except Exception as e:
     print()
     print(f"因为\"{e}\"而意外停止扫描，将保存已扫描的圣遗物信息")
+
+try:
+    level_threshold = int(level_threshold)
+except ValueError:
+    level_threshold = 0
+num_count = 0
+if level_threshold > 0:
+    for key in result.keys():
+        if key != 'version':
+            result[key] = [x for x in result[key] if x['level'] >= level_threshold]
+            num_count += len(result[key])
 if saved:
     with open('artifacts.genshinart.json', "wb") as f:
         s = json.dumps(result, ensure_ascii=False)
         f.write(s.encode('utf-8'))
-    print('总计扫描了{}个圣遗物，保存了{}个到artifacts.genshinart.json，未保存的则为识别结果无法理解，请到artifacts路径中查看'.format(art_id, saved))
+    print('总计扫描了{}/{}个圣遗物(无效识别结果请到artifacts路径中查看)，保存了{}个到artifacts.genshinart.json'.format(art_id, saved, num_count))
 else:
     print('总计扫描了{}个圣遗物，未保存任何圣遗物，未保存的则为识别结果无法理解，请到artifacts路径中查看'.format(art_id))
 input('已完成，按回车退出')