Commit 8566c87a authored by Ondřej Borýsek's avatar Ondřej Borýsek
Browse files

Additional type hints

parent b0cb4d2c

scan2report.py

+5 −5
Original line number Diff line number Diff line
@@ -140,7 +140,7 @@ class Scan2ReportConfig: 
    lang: str = 'cs'

    ignore: List[str] = field(default_factory=list)

    ignorePluginOutput: bool = False

    include: list = field(default_factory=list)

    include: List[str] = field(default_factory=list)

    minSeverity: int = 0

    outfile: str = 'out.docx'

    outfileFormat: str = 'docx'
@@ -150,9 +150,9 @@ class Scan2ReportConfig: 
    resolveRedirects: bool = False

    skipServices: bool = False

    skipTcpWrapped: bool = False

    templateDoc: str = None

    boldStart = '**'

    boldEnd = '**'

    templateDoc: str = None  # filled using __post_init__

    boldStart: str = '**'

    boldEnd: str = '**'



    def __post_init__(self):

        self.templateDoc = os.path.join(self.rootDir, self.TEMPLATE_DOC_DEFAULT)
@@ -392,7 +392,7 @@ def produce_findings_json(s2r: Scan2ReportData): 




# todo: switch reinit_global_vars to True?

def main(argv, reinit_global_vars: bool = False, s2r_c: Scan2ReportConfig = None) -> Scan2ReportData:

def main(argv: List[str], reinit_global_vars: bool = False, s2r_c: Scan2ReportConfig = None) -> Scan2ReportData:

    if reinit_global_vars or s2r_c is None:

        s2r_c = Scan2ReportConfig()