GranularFireplace commited on
Commit
305adbb
·
verified ·
1 Parent(s): 80f734e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -89,11 +89,13 @@ def preprocess_yara_rules(repo_path: Path) -> Path:
89
  if match:
90
  original_name = match.group(1)
91
  if original_name in seen_rules:
 
 
92
  seen_rules[original_name] += 1
93
  new_name = f"{original_name}_{seen_rules[original_name]}"
94
  line = line.replace(original_name, new_name, 1)
95
- rule_counter += 1
96
  else:
 
97
  seen_rules[original_name] = 0 # First occurrence
98
  new_content.append(line)
99
 
@@ -107,8 +109,8 @@ def preprocess_yara_rules(repo_path: Path) -> Path:
107
  def compile_yara_rules(repo_path: Path) -> Optional[yara.Rules]:
108
  """Compile YARA rules from repository with error handling"""
109
  try:
110
- # processed_dir = preprocess_yara_rules(repo_path)
111
- yara_files = list(repo_path.glob("**/*.yara"))
112
 
113
  if not yara_files:
114
  logger.warning("No YARA files found in repository")
 
89
  if match:
90
  original_name = match.group(1)
91
  if original_name in seen_rules:
92
+ logger.info(f"{original_name} is unseen")
93
+ rule_counter += 1
94
  seen_rules[original_name] += 1
95
  new_name = f"{original_name}_{seen_rules[original_name]}"
96
  line = line.replace(original_name, new_name, 1)
 
97
  else:
98
+ logger.warn(f"{original_name} is SEEN")
99
  seen_rules[original_name] = 0 # First occurrence
100
  new_content.append(line)
101
 
 
109
  def compile_yara_rules(repo_path: Path) -> Optional[yara.Rules]:
110
  """Compile YARA rules from repository with error handling"""
111
  try:
112
+ processed_dir = preprocess_yara_rules(repo_path)
113
+ yara_files = list(processed_dir.glob("**/*.yara"))
114
 
115
  if not yara_files:
116
  logger.warning("No YARA files found in repository")