Skip to content
This repository has been archived by the owner on Sep 5, 2023. It is now read-only.

Commit

Permalink
docs: add securitycenter prefix to samples, wrap published samples an…
Browse files Browse the repository at this point in the history
…d repl… (#85)

Standardizing Security Command Center samples to use 'securitycenter' prefixing. Wrapped existing samples to keep published doclinks unbroken, and fully replaced the ones that aren't published. Once this PR is through, published sample inclusions will be updated to use the new prefix, then I'll come through again and remove the unused block wraps.
  • Loading branch information
Strykrol committed Dec 15, 2020
1 parent f80a0d2 commit 553dfbb
Show file tree
Hide file tree
Showing 6 changed files with 82 additions and 12 deletions.
46 changes: 40 additions & 6 deletions samples/snippets/snippets_findings.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@

def create_source(organization_id):
"""Create a new findings source. """
# [START securitycenter_create_source]
# [START create_source]
from google.cloud import securitycenter

Expand All @@ -38,10 +39,12 @@ def create_source(organization_id):
)
print("Created Source: {}".format(created.name))
# [END create_source]
# [END securitycenter_create_source]


def get_source(source_name):
"""Gets an existing source."""
# [START securitycenter_get_source]
# [START get_source]
from google.cloud import securitycenter

Expand All @@ -57,11 +60,13 @@ def get_source(source_name):

print("Source: {}".format(source))
# [END get_source]
# [END securitycenter_get_source]
return source


def update_source(source_name):
"""Updates a source's display name."""
# [START securitycenter_update_source]
# [START update_source]
from google.cloud import securitycenter
from google.protobuf import field_mask_pb2
Expand All @@ -85,12 +90,14 @@ def update_source(source_name):
)
print("Updated Source: {}".format(updated))
# [END update_source]
# [END securitycenter_update_source]
return updated


def add_user_to_source(source_name):
"""Gives a user findingsEditor permission to the source."""
user_email = "csccclienttest@gmail.com"
# [START securitycenter_update_source_iam]
# [START update_source_iam]
from google.cloud import securitycenter
from google.iam.v1 import policy_pb2
Expand Down Expand Up @@ -125,12 +132,14 @@ def add_user_to_source(source_name):
print("Updated Policy: {}".format(updated))

# [END update_source_iam]
# [END securitycenter_update_source_iam]
return binding, updated


def list_source(organization_id):
"""Lists finding sources."""
i = -1
# [START securitycenter_list_sources]
# [START list_sources]
from google.cloud import securitycenter

Expand All @@ -144,11 +153,13 @@ def list_source(organization_id):
for i, source in enumerate(client.list_sources(request={"parent": org_name})):
print(i, source)
# [END list_sources]
# [END securitycenter_list_sources]
return i


def create_finding(source_name):
"""Creates a new finding."""
# [START securitycenter_create_finding]
# [START create_finding]
from google.cloud import securitycenter
from google.cloud.securitycenter_v1 import CreateFindingRequest, Finding
Expand Down Expand Up @@ -193,11 +204,13 @@ def create_finding(source_name):
)
print(created_finding)
# [END create_finding]
# [END securitycenter_create_finding]
return created_finding


def create_finding_with_source_properties(source_name):
"""Demonstrate creating a new finding with source properties. """
# [START securitycenter_create_finding_with_properties]
# [START create_finding_with_properties]
import datetime

Expand Down Expand Up @@ -249,9 +262,11 @@ def create_finding_with_source_properties(source_name):
)
print(created_finding)
# [END create_finding_with_properties]
# [END securitycenter_create_finding_with_properties]


def update_finding(source_name):
# [START securitycenter_update_finding]
# [START update_finding]
import datetime

Expand Down Expand Up @@ -295,10 +310,12 @@ def update_finding(source_name):
)
)
# [END update_finding]
# [END securitycenter_update_finding]


def update_finding_state(source_name):
"""Demonstrate updating only a finding state."""
# [START securitycenter_update_finding_state]
# [START update_finding_state]
import datetime

Expand All @@ -324,11 +341,13 @@ def update_finding_state(source_name):
)
print(f"New state: {new_finding.state}")
# [END update_finding_state]
# [END securitycenter_update_finding_state]


def trouble_shoot(source_name):
"""Demonstrate calling test_iam_permissions to determine if the
service account has the correct permisions."""
# [START securitycenter_test_iam_permissions]
# [START test_iam_permissions]
from google.cloud import securitycenter

Expand All @@ -355,7 +374,9 @@ def trouble_shoot(source_name):
)
)
# [END test_iam_permissions]
# [END securitycenter_test_iam_permissions]
assert len(permission_response.permissions) > 0
# [START securitycenter_test_iam_permissions]
# [START test_iam_permissions]
# Check for permissions necessary to call set_finding_state.
permission_response = client.test_iam_permissions(
Expand All @@ -368,11 +389,13 @@ def trouble_shoot(source_name):
"Permision to update state? {}".format(len(permission_response.permissions) > 0)
)
# [END test_iam_permissions]
# [END securitycenter_test_iam_permissions]
return permission_response
assert len(permission_response.permissions) > 0


def list_all_findings(organization_id):
# [START securitycenter_list_all_findings]
# [START list_all_findings]
from google.cloud import securitycenter

Expand All @@ -393,10 +416,12 @@ def list_all_findings(organization_id):
)
)
# [END list_all_findings]
# [END securitycenter_list_all_findings]
return i


def list_filtered_findings(source_name):
# [START securitycenter_list_filtered_findings]
# [START list_filtered_findings]
from google.cloud import securitycenter

Expand All @@ -423,10 +448,12 @@ def list_filtered_findings(source_name):
)
)
# [END list_filtered_findings]
# [END securitycenter_list_filtered_findings]
return i


def list_findings_at_time(source_name):
# [START securitycenter_list_findings_at_a_time]
# [START list_findings_at_a_time]
from google.cloud import securitycenter
from datetime import timedelta, datetime
Expand All @@ -444,7 +471,9 @@ def list_findings_at_time(source_name):
# source_name = "organizations/111122222444/sources/-"
five_days_ago = str(datetime.now() - timedelta(days=5))
# [END list_findings_at_a_time]
# [END securitycenter_list_findings_at_a_time]
i = -1
# [START securitycenter_list_findings_at_a_time]
# [START list_findings_at_a_time]

finding_result_iterator = client.list_findings(
Expand All @@ -457,11 +486,13 @@ def list_findings_at_time(source_name):
)
)
# [END list_findings_at_a_time]
# [END securitycenter_list_findings_at_a_time]
return i


def get_iam_policy(source_name):
"""Gives a user findingsEditor permission to the source."""
# [START securitycenter_get_source_iam]
# [START get_source_iam]
from google.cloud import securitycenter

Expand All @@ -477,11 +508,13 @@ def get_iam_policy(source_name):
policy = client.get_iam_policy(request={"resource": source_name})
print("Policy: {}".format(policy))
# [END get_source_iam]
# [END securitycenter_get_source_iam]


def group_all_findings(organization_id):
"""Demonstrates grouping all findings across an organization."""
i = 0
# [START securitycenter_group_all_findings]
# [START group_all_findings]
from google.cloud import securitycenter

Expand All @@ -500,13 +533,14 @@ def group_all_findings(organization_id):
for i, group_result in enumerate(group_result_iterator):
print((i + 1), group_result)
# [END group_all_findings]
# [END securitycenter_group_all_findings]
return i


def group_filtered_findings(source_name):
"""Demonstrates grouping all findings across an organization."""
i = 0
# [START group_filtered_findings]
# [START securitycenter_group_filtered_findings]
from google.cloud import securitycenter

# Create a client.
Expand All @@ -528,15 +562,15 @@ def group_filtered_findings(source_name):
)
for i, group_result in enumerate(group_result_iterator):
print((i + 1), group_result)
# [END group_filtered_findings]
# [END securitycenter_group_filtered_findings]
return i


def group_findings_at_time(source_name):
"""Demonstrates grouping all findings across an organization as of
a specific time."""
i = -1
# [START group_findings_at_time]
# [START securitycenter_group_findings_at_time]
from datetime import datetime, timedelta
from google.cloud import securitycenter

Expand All @@ -562,15 +596,15 @@ def group_findings_at_time(source_name):
)
for i, group_result in enumerate(group_result_iterator):
print((i + 1), group_result)
# [END group_findings_at_time]
# [END securitycenter_group_findings_at_time]
return i


def group_findings_and_changes(source_name):
"""Demonstrates grouping all findings across an organization and
associated changes."""
i = 0
# [START group_findings_with_changes]
# [START securitycenter_group_findings_with_changes]
from datetime import timedelta

from google.cloud import securitycenter
Expand All @@ -597,5 +631,5 @@ def group_findings_and_changes(source_name):
)
for i, group_result in enumerate(group_result_iterator):
print((i + 1), group_result)
# [END group_findings_with_changes]
# [END securitycenter_group_findings_with_changes]]
return i
20 changes: 14 additions & 6 deletions samples/snippets/snippets_list_assets.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
def list_all_assets(organization_id):
"""Demonstrate listing and printing all assets."""
i = 0
# [START securitycenter_demo_list_all_assets]
# [START demo_list_all_assets]
from google.cloud import securitycenter

Expand All @@ -33,12 +34,14 @@ def list_all_assets(organization_id):
for i, asset_result in enumerate(asset_iterator):
print(i, asset_result)
# [END demo_list_all_assets]
# [END securitycenter_demo_list_all_assets]
return i


def list_assets_with_filters(organization_id):
"""Demonstrate listing assets with a filter."""
i = 0
# [START securitycenter_demo_list_assets_with_filter]
# [START demo_list_assets_with_filter]
from google.cloud import securitycenter

Expand All @@ -59,12 +62,14 @@ def list_assets_with_filters(organization_id):
for i, asset_result in enumerate(asset_iterator):
print(i, asset_result)
# [END demo_list_assets_with_filter]
# [END securitycenter_demo_list_assets_with_filter]
return i


def list_assets_with_filters_and_read_time(organization_id):
"""Demonstrate listing assets with a filter."""
i = 0
# [START securitycenter_demo_list_assets_with_filter_and_time]
# [START demo_list_assets_with_filter_and_time]
from datetime import datetime, timedelta

Expand Down Expand Up @@ -95,12 +100,14 @@ def list_assets_with_filters_and_read_time(organization_id):
for i, asset_result in enumerate(asset_iterator):
print(i, asset_result)
# [END demo_list_assets_with_filter_and_time]
# [END securitycenter_demo_list_assets_with_filter_and_time]
return i


def list_point_in_time_changes(organization_id):
"""Demonstrate listing assets along with their state changes."""
i = 0
# [START securitycenter_demo_list_assets_changes]
# [START demo_list_assets_changes]
from datetime import timedelta

Expand Down Expand Up @@ -131,13 +138,14 @@ def list_point_in_time_changes(organization_id):
print(i, asset)

# [END demo_list_assets_changes]
# [END securitycenter_demo_list_assets_changes]
return i


def group_assets(organization_id):
"""Demonstrates grouping all assets by type. """
i = 0
# [START group_all_assets]
# [START securitycenter_group_all_assets]
from google.cloud import securitycenter

client = securitycenter.SecurityCenterClient()
Expand All @@ -153,14 +161,14 @@ def group_assets(organization_id):
)
for i, result in enumerate(result_iterator):
print((i + 1), result)
# [END group_all_assets]
# [END securitycenter_group_all_assets]
return i


def group_filtered_assets(organization_id):
"""Demonstrates grouping assets by type with a filter. """
i = 0
# [START group_all_assets_with_filter]
# [START securitycenter_group_all_assets_with_filter]
from google.cloud import securitycenter

client = securitycenter.SecurityCenterClient()
Expand All @@ -179,15 +187,15 @@ def group_filtered_assets(organization_id):
)
for i, result in enumerate(result_iterator):
print((i + 1), result)
# [END group_all_assets_with_filter]
# [END securitycenter_group_all_assets_with_filter]
# only one asset type is a project
return i


def group_assets_by_changes(organization_id):
"""Demonstrates grouping assets by their changes over a period of time."""
i = 0
# [START group_all_assets_by_change]
# [START securitycenter_group_all_assets_by_change]
from datetime import timedelta

from google.cloud import securitycenter
Expand All @@ -208,5 +216,5 @@ def group_assets_by_changes(organization_id):
)
for i, result in enumerate(result_iterator):
print((i + 1), result)
# [END group_all_assets_by_change]
# [END securitycenter_group_all_assets_by_change]
return i
Loading

0 comments on commit 553dfbb

Please sign in to comment.