Skip to content

Commit b76d0c9

Browse files
committed
fix: X and Youtube module's capitalization issue
1 parent 988c1ab commit b76d0c9

1 file changed

Lines changed: 14 additions & 20 deletions

File tree

user_scanner/core/orchestrator.py

Lines changed: 14 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -103,40 +103,34 @@ def run_user_full(username: str, configs: ScanConfig) -> List[Result]:
103103
display_name = cat_name.capitalize()
104104
for m in modules:
105105
all_modules.append(m)
106-
module_to_cat[get_site_name(m)] = display_name
106+
site_key = get_site_name(m).capitalize()
107+
module_to_cat[site_key] = display_name
107108

108109
with ThreadPoolExecutor(max_workers=60) as executor:
109110
exec_map = executor.map(
110111
lambda m: _worker_single(m, username, configs), all_modules
111112
)
112113
for result in exec_map:
113-
site_name = result.site_name
114-
cat_name = (
115-
module_to_cat.get(site_name, "Unknown") if site_name else "Unknown"
116-
)
114+
cat_name = module_to_cat.get(result.site_name, "Unknown")
117115

118116
result.update(category=cat_name)
119117
results.append(result)
120118

121-
if configs.only_found:
122-
if result.is_found():
123-
if cat_name not in printed_categories:
124-
print(
125-
f"\n{Fore.MAGENTA}== {cat_name.upper()} SITES =={Style.RESET_ALL}"
126-
)
127-
printed_categories.add(cat_name)
128-
result.show(configs)
129-
else:
130-
if cat_name not in printed_categories:
131-
print(
132-
f"\n{Fore.MAGENTA}== {cat_name.upper()} SITES =={Style.RESET_ALL}"
133-
)
134-
printed_categories.add(cat_name)
135-
result.show(configs)
119+
if configs.only_found and not result.is_found():
120+
continue
121+
122+
if cat_name not in printed_categories:
123+
print(f"\n{Fore.MAGENTA}== {cat_name.upper()} SITES =={Style.RESET_ALL}")
124+
printed_categories.add(cat_name)
125+
126+
result.show(configs)
136127

137128
return results
138129

139130

131+
132+
133+
140134
def make_request(url: str, **kwargs) -> httpx.Response:
141135
"""Simple wrapper to **httpx.get** that predefines headers and timeout"""
142136
if "headers" not in kwargs:

0 commit comments

Comments
 (0)