fix: Check result list before appending. Remove the need to clean final_data

This commit is contained in:
KenF
2026-01-22 20:01:20 +08:00
parent 0614cee9b2
commit 675f0f6ef3
10 changed files with 30 additions and 30 deletions
+2 -2
View File
@@ -25,6 +25,7 @@ def getResources(region_name: str) -> list[list[str | int]]:
for t in table_names: for t in table_names:
resp2 = client.describe_table(TableName=t) resp2 = client.describe_table(TableName=t)
i = resp2['Table'] i = resp2['Table']
if i:
return_data.append([i['TableName'], i['TableStatus'], i['TableSizeBytes'], i['ItemCount'], region_name]) return_data.append([i['TableName'], i['TableStatus'], i['TableSizeBytes'], i['ItemCount'], region_name])
return return_data return return_data
@@ -42,8 +43,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -27,6 +27,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
if tag['Key'] == "Name": if tag['Key'] == "Name":
name_tag = tag['Value'] name_tag = tag['Value']
break break
if i:
return_data.append([i['InstanceId'], name_tag, i.get('PlatformDetails'), i['InstanceType'], i['PrivateIpAddress'], i['Placement']['AvailabilityZone']]) return_data.append([i['InstanceId'], name_tag, i.get('PlatformDetails'), i['InstanceType'], i['PrivateIpAddress'], i['Placement']['AvailabilityZone']])
return return_data return return_data
@@ -44,8 +45,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -22,6 +22,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('elasticache', region_name=region_name) client = boto3.client('elasticache', region_name=region_name)
response = client.describe_cache_clusters() response = client.describe_cache_clusters()
for i in response['CacheClusters']: for i in response['CacheClusters']:
if i:
return_data.append([i['CacheClusterId'], i['CacheNodeType'], i['Engine'], i['EngineVersion'], i['NumCacheNodes'], i['PreferredAvailabilityZone']]) return_data.append([i['CacheClusterId'], i['CacheNodeType'], i['Engine'], i['EngineVersion'], i['NumCacheNodes'], i['PreferredAvailabilityZone']])
return return_data return return_data
@@ -39,8 +40,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -21,6 +21,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('emr', region_name=region_name) client = boto3.client('emr', region_name=region_name)
response = client.list_clusters() response = client.list_clusters()
for i in response['Clusters']: for i in response['Clusters']:
if i:
return_data.append([i['Name'], i['Status'], i['NormalizedInstanceHours'], region_name]) return_data.append([i['Name'], i['Status'], i['NormalizedInstanceHours'], region_name])
return return_data return return_data
@@ -38,8 +39,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -23,6 +23,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
for page in paginator.paginate(): for page in paginator.paginate():
functions = page.get('Functions', []) functions = page.get('Functions', [])
for i in functions: for i in functions:
if i:
return_data.append([i['FunctionName'], i['Runtime'], i['Architectures'][0], region_name]) return_data.append([i['FunctionName'], i['Runtime'], i['Architectures'][0], region_name])
return return_data return return_data
@@ -40,8 +41,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -22,6 +22,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('elbv2', region_name=region_name) client = boto3.client('elbv2', region_name=region_name)
response = client.describe_load_balancers() response = client.describe_load_balancers()
for i in response['LoadBalancers']: for i in response['LoadBalancers']:
if i:
return_data.append([i['LoadBalancerName'], i['Scheme'], i['Type'], region_name]) return_data.append([i['LoadBalancerName'], i['Scheme'], i['Type'], region_name])
return return_data return return_data
@@ -39,8 +40,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -21,6 +21,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('logs', region_name=region_name) client = boto3.client('logs', region_name=region_name)
response = client.describe_log_groups() response = client.describe_log_groups()
for logGroup in response['logGroups']: for logGroup in response['logGroups']:
if logGroup:
return_data.append([logGroup['logGroupName'], logGroup.get('retentionInDays'), logGroup['storedBytes'], region_name]) return_data.append([logGroup['logGroupName'], logGroup.get('retentionInDays'), logGroup['storedBytes'], region_name])
return return_data return return_data
@@ -38,8 +39,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -21,6 +21,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('opensearch', region_name=region_name) client = boto3.client('opensearch', region_name=region_name)
response = client.list_domain_names() response = client.list_domain_names()
for i in response['DomainNames']: for i in response['DomainNames']:
if i:
return_data.append([i['DomainName'], i['EngineType'], region_name]) return_data.append([i['DomainName'], i['EngineType'], region_name])
return return_data return return_data
@@ -38,8 +39,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -22,6 +22,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
client = boto3.client('rds', region_name=region_name) client = boto3.client('rds', region_name=region_name)
response = client.describe_db_instances() response = client.describe_db_instances()
for i in response['DBInstances']: for i in response['DBInstances']:
if i:
return_data.append([i['DBInstanceIdentifier'], i['DBInstanceClass'], i['Engine'], i['EngineVersion'], i['MultiAZ'], region_name]) return_data.append([i['DBInstanceIdentifier'], i['DBInstanceClass'], i['Engine'], i['EngineVersion'], i['MultiAZ'], region_name])
return return_data return return_data
@@ -39,8 +40,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')
+2 -2
View File
@@ -23,6 +23,7 @@ def printResources(region_name: str) -> list[list[str | int]]:
for i in response['Subnets']: for i in response['Subnets']:
default_vpc_check = client.describe_vpcs(VpcIds=[i['VpcId']]) default_vpc_check = client.describe_vpcs(VpcIds=[i['VpcId']])
for v in default_vpc_check['Vpcs']: for v in default_vpc_check['Vpcs']:
if i:
return_data.append([i['SubnetId'], i['VpcId'], i['CidrBlock'], i['AvailabilityZone'], v['IsDefault']]) return_data.append([i['SubnetId'], i['VpcId'], i['CidrBlock'], i['AvailabilityZone'], v['IsDefault']])
return return_data return return_data
@@ -40,8 +41,7 @@ def main() -> None:
region_data = future.result() region_data = future.result()
final_data.extend(region_data) final_data.extend(region_data)
clean_data = [row for row in final_data if row] for row in final_data:
for row in clean_data:
ws.append(row) ws.append(row)
wb.save('aws-inventory.xlsx') wb.save('aws-inventory.xlsx')