Files
peko-java/python/scan_exception.py
2024-04-18 18:33:49 +08:00

57 lines
1.5 KiB
Python

import random
import hashlib
import requests
import time
import os
def spider(filepath, suffix):
code_files = []
for root, dirs, files in os.walk(filepath, topdown=False):
for f in files:
file = os.path.join(root, f)
if file.endswith(suffix):
code_files.append(file.replace("\\", "/"))
return code_files
def message(code_files):
msgs = []
for i in code_files:
fp = open(i, 'r', encoding='utf-8')
data = fp.readlines()
for line in data:
curr_code = line.strip('\n')
if curr_code is None:
curr_code = line.strip('\r\n')
msgs.append(curr_code)
return msgs
if __name__ == '__main__':
code_files = spider('/Users/liaozetao/IdeaProjects/peko-java/accompany-business', '.java')
msgs = message(code_files)
str_arr = []
for m in msgs:
try:
if len(m) == 0:
continue
if not ('Exception("' in m):
continue
arr = m.split('Exception("')
if len(arr) == 0:
continue
if not ('Service' in arr[0]):
continue
message = arr[1].replace('");', '')
message = message.replace(');', '')
message = message.replace('")', '')
message = message.replace('", e', '')
str_arr.append(message)
except Exception as e:
print(e)
continue
str_set = list(set(str_arr))
for s in str_set:
print(s)