8
8
import os
9
9
import sys
10
10
import platform
11
+ import re
11
12
from argparse import ArgumentParser , RawDescriptionHelpFormatter
12
13
from typing import List , Tuple
13
14
import os .path as path
@@ -679,25 +680,30 @@ async def main():
679
680
# TODO: tests
680
681
if recursive_search_enabled :
681
682
extracted_ids = extract_ids_from_results (results , db )
683
+ query_notify .warning (f'Extracted IDs: { extracted_ids } ' )
682
684
usernames .update (extracted_ids )
683
685
684
686
# reporting for a one username
685
687
if args .xmind :
688
+ username = username .replace ('/' , '_' )
686
689
filename = report_filepath_tpl .format (username = username , postfix = '.xmind' )
687
690
save_xmind_report (filename , username , results )
688
691
query_notify .warning (f'XMind report for { username } saved in { filename } ' )
689
692
690
693
if args .csv :
694
+ username = username .replace ('/' , '_' )
691
695
filename = report_filepath_tpl .format (username = username , postfix = '.csv' )
692
696
save_csv_report (filename , username , results )
693
697
query_notify .warning (f'CSV report for { username } saved in { filename } ' )
694
698
695
699
if args .txt :
700
+ username = username .replace ('/' , '_' )
696
701
filename = report_filepath_tpl .format (username = username , postfix = '.txt' )
697
702
save_txt_report (filename , username , results )
698
703
query_notify .warning (f'TXT report for { username } saved in { filename } ' )
699
704
700
705
if args .json :
706
+ username = username .replace ('/' , '_' )
701
707
filename = report_filepath_tpl .format (
702
708
username = username , postfix = f'_{ args .json } .json'
703
709
)
@@ -715,18 +721,21 @@ async def main():
715
721
username = report_context ['username' ]
716
722
717
723
if args .html :
724
+ username = username .replace ('/' , '_' )
718
725
filename = report_filepath_tpl .format (
719
726
username = username , postfix = '_plain.html'
720
727
)
721
728
save_html_report (filename , report_context )
722
729
query_notify .warning (f'HTML report on all usernames saved in { filename } ' )
723
730
724
731
if args .pdf :
732
+ username = username .replace ('/' , '_' )
725
733
filename = report_filepath_tpl .format (username = username , postfix = '.pdf' )
726
734
save_pdf_report (filename , report_context )
727
735
query_notify .warning (f'PDF report on all usernames saved in { filename } ' )
728
736
729
737
if args .graph :
738
+ username = username .replace ('/' , '_' )
730
739
filename = report_filepath_tpl .format (
731
740
username = username , postfix = '_graph.html'
732
741
)
0 commit comments