14
14
from dataclasses import asdict , dataclass
15
15
from datetime import datetime
16
16
from pathlib import Path
17
+ import typing
17
18
from typing import Any
18
19
19
20
import xlsxwriter # type: ignore[reportMissingTypeStubs]
@@ -86,19 +87,20 @@ def populate(self):
86
87
self .add_attack_matrix ()
87
88
self .add_rule_details (self .deprecated_rules , "Deprecated Rules" )
88
89
90
+ @typing .no_type_check
89
91
def add_summary (self ):
90
92
"""Add the summary worksheet."""
91
- worksheet = self .add_worksheet ("Summary" ) # type: ignore[reportUnknownMemberType]
92
- worksheet .freeze_panes (1 , 0 ) # type: ignore[reportUnknownMemberType]
93
- worksheet .set_column (0 , 0 , 25 ) # type: ignore[reportUnknownMemberType]
94
- worksheet .set_column (1 , 1 , 10 ) # type: ignore[reportUnknownMemberType]
93
+ worksheet = self .add_worksheet ("Summary" )
94
+ worksheet .freeze_panes (1 , 0 )
95
+ worksheet .set_column (0 , 0 , 25 )
96
+ worksheet .set_column (1 , 1 , 10 )
95
97
96
98
row = 0
97
- worksheet .merge_range (row , 0 , row , 1 , "SUMMARY" , self .bold_center ) # type: ignore[reportUnknownMemberType]
99
+ worksheet .merge_range (row , 0 , row , 1 , "SUMMARY" , self .bold_center )
98
100
row += 1
99
101
100
- _ = worksheet .write (row , 0 , "Package Name" ) # type: ignore[reportUnknownMemberType]
101
- _ = worksheet .write (row , 1 , self .package .name , self .right_align ) # type: ignore[reportUnknownMemberType]
102
+ _ = worksheet .write (row , 0 , "Package Name" )
103
+ _ = worksheet .write (row , 1 , self .package .name , self .right_align )
102
104
row += 1
103
105
104
106
tactic_counts : dict [str , int ] = defaultdict (int )
@@ -108,29 +110,29 @@ def add_summary(self):
108
110
for entry in threat :
109
111
tactic_counts [entry .tactic .name ] += 1
110
112
111
- _ = worksheet .write (row , 0 , "Total Production Rules" ) # type: ignore[reportUnknownVariableType]
112
- _ = worksheet .write (row , 1 , len (self .production_rules )) # type: ignore[reportUnknownVariableType]
113
+ _ = worksheet .write (row , 0 , "Total Production Rules" )
114
+ _ = worksheet .write (row , 1 , len (self .production_rules ))
113
115
row += 2
114
116
115
- _ = worksheet .write (row , 0 , "Total Deprecated Rules" ) # type: ignore[reportUnknownVariableType]
116
- _ = worksheet .write (row , 1 , len (self .deprecated_rules )) # type: ignore[reportUnknownVariableType]
117
+ _ = worksheet .write (row , 0 , "Total Deprecated Rules" )
118
+ _ = worksheet .write (row , 1 , len (self .deprecated_rules ))
117
119
row += 1
118
120
119
- _ = worksheet .write (row , 0 , "Total Rules" ) # type: ignore[reportUnknownVariableType]
120
- _ = worksheet .write (row , 1 , len (self .package .rules )) # type: ignore[reportUnknownVariableType]
121
+ _ = worksheet .write (row , 0 , "Total Rules" )
122
+ _ = worksheet .write (row , 1 , len (self .package .rules ))
121
123
row += 2
122
124
123
- worksheet .merge_range (row , 0 , row , 3 , f"MITRE { attack_tm } TACTICS" , self .bold_center ) # type: ignore[reportUnknownVariableType]
125
+ worksheet .merge_range (row , 0 , row , 3 , f"MITRE { attack_tm } TACTICS" , self .bold_center )
124
126
row += 1
125
127
126
128
for tactic in tactics :
127
- _ = worksheet .write (row , 0 , tactic ) # type: ignore[reportUnknownVariableType]
128
- _ = worksheet .write (row , 1 , tactic_counts [tactic ]) # type: ignore[reportUnknownVariableType]
129
+ _ = worksheet .write (row , 0 , tactic )
130
+ _ = worksheet .write (row , 1 , tactic_counts [tactic ])
129
131
num_techniques = len (self ._coverage [tactic ])
130
132
total_techniques = len (matrix [tactic ])
131
133
percent = float (num_techniques ) / float (total_techniques )
132
- _ = worksheet .write (row , 2 , percent , self .percent ) # type: ignore[reportUnknownVariableType]
133
- _ = worksheet .write (row , 3 , f"{ num_techniques } /{ total_techniques } " , self .right_align ) # type: ignore[reportUnknownVariableType]
134
+ _ = worksheet .write (row , 2 , percent , self .percent )
135
+ _ = worksheet .write (row , 3 , f"{ num_techniques } /{ total_techniques } " , self .right_align )
134
136
row += 1
135
137
136
138
def add_rule_details (
0 commit comments