fix: Check result list before appending. Remove the need to clean final_data
This commit is contained in:
+3
-3
@@ -25,7 +25,8 @@ def getResources(region_name: str) -> list[list[str | int]]:
|
||||
for t in table_names:
|
||||
resp2 = client.describe_table(TableName=t)
|
||||
i = resp2['Table']
|
||||
return_data.append([i['TableName'], i['TableStatus'], i['TableSizeBytes'], i['ItemCount'], region_name])
|
||||
if i:
|
||||
return_data.append([i['TableName'], i['TableStatus'], i['TableSizeBytes'], i['ItemCount'], region_name])
|
||||
return return_data
|
||||
|
||||
# Main function
|
||||
@@ -42,8 +43,7 @@ def main() -> None:
|
||||
region_data = future.result()
|
||||
final_data.extend(region_data)
|
||||
|
||||
clean_data = [row for row in final_data if row]
|
||||
for row in clean_data:
|
||||
for row in final_data:
|
||||
ws.append(row)
|
||||
|
||||
wb.save('aws-inventory.xlsx')
|
||||
|
||||
Reference in New Issue
Block a user