Skip to content

Commit 8bce842

Browse files
authored
docs: add securitycenter prefix to samples, wrap published samples and 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.
1 parent 21a7879 commit 8bce842

6 files changed

+82
-12
lines changed

securitycenter/snippets/snippets_findings.py

Lines changed: 40 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020
def create_source(organization_id):
2121
"""Create a new findings source. """
22+
# [START securitycenter_create_source]
2223
# [START create_source]
2324
from google.cloud import securitycenter
2425

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

4244

4345
def get_source(source_name):
4446
"""Gets an existing source."""
47+
# [START securitycenter_get_source]
4548
# [START get_source]
4649
from google.cloud import securitycenter
4750

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

5861
print("Source: {}".format(source))
5962
# [END get_source]
63+
# [END securitycenter_get_source]
6064
return source
6165

6266

6367
def update_source(source_name):
6468
"""Updates a source's display name."""
69+
# [START securitycenter_update_source]
6570
# [START update_source]
6671
from google.cloud import securitycenter
6772
from google.protobuf import field_mask_pb2
@@ -85,12 +90,14 @@ def update_source(source_name):
8590
)
8691
print("Updated Source: {}".format(updated))
8792
# [END update_source]
93+
# [END securitycenter_update_source]
8894
return updated
8995

9096

9197
def add_user_to_source(source_name):
9298
"""Gives a user findingsEditor permission to the source."""
9399
user_email = "[email protected]"
100+
# [START securitycenter_update_source_iam]
94101
# [START update_source_iam]
95102
from google.cloud import securitycenter
96103
from google.iam.v1 import policy_pb2
@@ -125,12 +132,14 @@ def add_user_to_source(source_name):
125132
print("Updated Policy: {}".format(updated))
126133

127134
# [END update_source_iam]
135+
# [END securitycenter_update_source_iam]
128136
return binding, updated
129137

130138

131139
def list_source(organization_id):
132140
"""Lists finding sources."""
133141
i = -1
142+
# [START securitycenter_list_sources]
134143
# [START list_sources]
135144
from google.cloud import securitycenter
136145

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

149159

150160
def create_finding(source_name):
151161
"""Creates a new finding."""
162+
# [START securitycenter_create_finding]
152163
# [START create_finding]
153164
from google.cloud import securitycenter
154165
from google.cloud.securitycenter_v1 import CreateFindingRequest, Finding
@@ -193,11 +204,13 @@ def create_finding(source_name):
193204
)
194205
print(created_finding)
195206
# [END create_finding]
207+
# [END securitycenter_create_finding]
196208
return created_finding
197209

198210

199211
def create_finding_with_source_properties(source_name):
200212
"""Demonstrate creating a new finding with source properties. """
213+
# [START securitycenter_create_finding_with_properties]
201214
# [START create_finding_with_properties]
202215
import datetime
203216

@@ -249,9 +262,11 @@ def create_finding_with_source_properties(source_name):
249262
)
250263
print(created_finding)
251264
# [END create_finding_with_properties]
265+
# [END securitycenter_create_finding_with_properties]
252266

253267

254268
def update_finding(source_name):
269+
# [START securitycenter_update_finding]
255270
# [START update_finding]
256271
import datetime
257272

@@ -295,10 +310,12 @@ def update_finding(source_name):
295310
)
296311
)
297312
# [END update_finding]
313+
# [END securitycenter_update_finding]
298314

299315

300316
def update_finding_state(source_name):
301317
"""Demonstrate updating only a finding state."""
318+
# [START securitycenter_update_finding_state]
302319
# [START update_finding_state]
303320
import datetime
304321

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

328346

329347
def trouble_shoot(source_name):
330348
"""Demonstrate calling test_iam_permissions to determine if the
331349
service account has the correct permisions."""
350+
# [START securitycenter_test_iam_permissions]
332351
# [START test_iam_permissions]
333352
from google.cloud import securitycenter
334353

@@ -355,7 +374,9 @@ def trouble_shoot(source_name):
355374
)
356375
)
357376
# [END test_iam_permissions]
377+
# [END securitycenter_test_iam_permissions]
358378
assert len(permission_response.permissions) > 0
379+
# [START securitycenter_test_iam_permissions]
359380
# [START test_iam_permissions]
360381
# Check for permissions necessary to call set_finding_state.
361382
permission_response = client.test_iam_permissions(
@@ -368,11 +389,13 @@ def trouble_shoot(source_name):
368389
"Permision to update state? {}".format(len(permission_response.permissions) > 0)
369390
)
370391
# [END test_iam_permissions]
392+
# [END securitycenter_test_iam_permissions]
371393
return permission_response
372394
assert len(permission_response.permissions) > 0
373395

374396

375397
def list_all_findings(organization_id):
398+
# [START securitycenter_list_all_findings]
376399
# [START list_all_findings]
377400
from google.cloud import securitycenter
378401

@@ -393,10 +416,12 @@ def list_all_findings(organization_id):
393416
)
394417
)
395418
# [END list_all_findings]
419+
# [END securitycenter_list_all_findings]
396420
return i
397421

398422

399423
def list_filtered_findings(source_name):
424+
# [START securitycenter_list_filtered_findings]
400425
# [START list_filtered_findings]
401426
from google.cloud import securitycenter
402427

@@ -423,10 +448,12 @@ def list_filtered_findings(source_name):
423448
)
424449
)
425450
# [END list_filtered_findings]
451+
# [END securitycenter_list_filtered_findings]
426452
return i
427453

428454

429455
def list_findings_at_time(source_name):
456+
# [START securitycenter_list_findings_at_a_time]
430457
# [START list_findings_at_a_time]
431458
from google.cloud import securitycenter
432459
from datetime import timedelta, datetime
@@ -444,7 +471,9 @@ def list_findings_at_time(source_name):
444471
# source_name = "organizations/111122222444/sources/-"
445472
five_days_ago = str(datetime.now() - timedelta(days=5))
446473
# [END list_findings_at_a_time]
474+
# [END securitycenter_list_findings_at_a_time]
447475
i = -1
476+
# [START securitycenter_list_findings_at_a_time]
448477
# [START list_findings_at_a_time]
449478

450479
finding_result_iterator = client.list_findings(
@@ -457,11 +486,13 @@ def list_findings_at_time(source_name):
457486
)
458487
)
459488
# [END list_findings_at_a_time]
489+
# [END securitycenter_list_findings_at_a_time]
460490
return i
461491

462492

463493
def get_iam_policy(source_name):
464494
"""Gives a user findingsEditor permission to the source."""
495+
# [START securitycenter_get_source_iam]
465496
# [START get_source_iam]
466497
from google.cloud import securitycenter
467498

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

481513

482514
def group_all_findings(organization_id):
483515
"""Demonstrates grouping all findings across an organization."""
484516
i = 0
517+
# [START securitycenter_group_all_findings]
485518
# [START group_all_findings]
486519
from google.cloud import securitycenter
487520

@@ -500,13 +533,14 @@ def group_all_findings(organization_id):
500533
for i, group_result in enumerate(group_result_iterator):
501534
print((i + 1), group_result)
502535
# [END group_all_findings]
536+
# [END securitycenter_group_all_findings]
503537
return i
504538

505539

506540
def group_filtered_findings(source_name):
507541
"""Demonstrates grouping all findings across an organization."""
508542
i = 0
509-
# [START group_filtered_findings]
543+
# [START securitycenter_group_filtered_findings]
510544
from google.cloud import securitycenter
511545

512546
# Create a client.
@@ -528,15 +562,15 @@ def group_filtered_findings(source_name):
528562
)
529563
for i, group_result in enumerate(group_result_iterator):
530564
print((i + 1), group_result)
531-
# [END group_filtered_findings]
565+
# [END securitycenter_group_filtered_findings]
532566
return i
533567

534568

535569
def group_findings_at_time(source_name):
536570
"""Demonstrates grouping all findings across an organization as of
537571
a specific time."""
538572
i = -1
539-
# [START group_findings_at_time]
573+
# [START securitycenter_group_findings_at_time]
540574
from datetime import datetime, timedelta
541575
from google.cloud import securitycenter
542576

@@ -562,15 +596,15 @@ def group_findings_at_time(source_name):
562596
)
563597
for i, group_result in enumerate(group_result_iterator):
564598
print((i + 1), group_result)
565-
# [END group_findings_at_time]
599+
# [END securitycenter_group_findings_at_time]
566600
return i
567601

568602

569603
def group_findings_and_changes(source_name):
570604
"""Demonstrates grouping all findings across an organization and
571605
associated changes."""
572606
i = 0
573-
# [START group_findings_with_changes]
607+
# [START securitycenter_group_findings_with_changes]
574608
from datetime import timedelta
575609

576610
from google.cloud import securitycenter
@@ -597,5 +631,5 @@ def group_findings_and_changes(source_name):
597631
)
598632
for i, group_result in enumerate(group_result_iterator):
599633
print((i + 1), group_result)
600-
# [END group_findings_with_changes]
634+
# [END securitycenter_group_findings_with_changes]]
601635
return i

securitycenter/snippets/snippets_list_assets.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
def list_all_assets(organization_id):
2121
"""Demonstrate listing and printing all assets."""
2222
i = 0
23+
# [START securitycenter_demo_list_all_assets]
2324
# [START demo_list_all_assets]
2425
from google.cloud import securitycenter
2526

@@ -33,12 +34,14 @@ def list_all_assets(organization_id):
3334
for i, asset_result in enumerate(asset_iterator):
3435
print(i, asset_result)
3536
# [END demo_list_all_assets]
37+
# [END securitycenter_demo_list_all_assets]
3638
return i
3739

3840

3941
def list_assets_with_filters(organization_id):
4042
"""Demonstrate listing assets with a filter."""
4143
i = 0
44+
# [START securitycenter_demo_list_assets_with_filter]
4245
# [START demo_list_assets_with_filter]
4346
from google.cloud import securitycenter
4447

@@ -59,12 +62,14 @@ def list_assets_with_filters(organization_id):
5962
for i, asset_result in enumerate(asset_iterator):
6063
print(i, asset_result)
6164
# [END demo_list_assets_with_filter]
65+
# [END securitycenter_demo_list_assets_with_filter]
6266
return i
6367

6468

6569
def list_assets_with_filters_and_read_time(organization_id):
6670
"""Demonstrate listing assets with a filter."""
6771
i = 0
72+
# [START securitycenter_demo_list_assets_with_filter_and_time]
6873
# [START demo_list_assets_with_filter_and_time]
6974
from datetime import datetime, timedelta
7075

@@ -95,12 +100,14 @@ def list_assets_with_filters_and_read_time(organization_id):
95100
for i, asset_result in enumerate(asset_iterator):
96101
print(i, asset_result)
97102
# [END demo_list_assets_with_filter_and_time]
103+
# [END securitycenter_demo_list_assets_with_filter_and_time]
98104
return i
99105

100106

101107
def list_point_in_time_changes(organization_id):
102108
"""Demonstrate listing assets along with their state changes."""
103109
i = 0
110+
# [START securitycenter_demo_list_assets_changes]
104111
# [START demo_list_assets_changes]
105112
from datetime import timedelta
106113

@@ -131,13 +138,14 @@ def list_point_in_time_changes(organization_id):
131138
print(i, asset)
132139

133140
# [END demo_list_assets_changes]
141+
# [END securitycenter_demo_list_assets_changes]
134142
return i
135143

136144

137145
def group_assets(organization_id):
138146
"""Demonstrates grouping all assets by type. """
139147
i = 0
140-
# [START group_all_assets]
148+
# [START securitycenter_group_all_assets]
141149
from google.cloud import securitycenter
142150

143151
client = securitycenter.SecurityCenterClient()
@@ -153,14 +161,14 @@ def group_assets(organization_id):
153161
)
154162
for i, result in enumerate(result_iterator):
155163
print((i + 1), result)
156-
# [END group_all_assets]
164+
# [END securitycenter_group_all_assets]
157165
return i
158166

159167

160168
def group_filtered_assets(organization_id):
161169
"""Demonstrates grouping assets by type with a filter. """
162170
i = 0
163-
# [START group_all_assets_with_filter]
171+
# [START securitycenter_group_all_assets_with_filter]
164172
from google.cloud import securitycenter
165173

166174
client = securitycenter.SecurityCenterClient()
@@ -179,15 +187,15 @@ def group_filtered_assets(organization_id):
179187
)
180188
for i, result in enumerate(result_iterator):
181189
print((i + 1), result)
182-
# [END group_all_assets_with_filter]
190+
# [END securitycenter_group_all_assets_with_filter]
183191
# only one asset type is a project
184192
return i
185193

186194

187195
def group_assets_by_changes(organization_id):
188196
"""Demonstrates grouping assets by their changes over a period of time."""
189197
i = 0
190-
# [START group_all_assets_by_change]
198+
# [START securitycenter_group_all_assets_by_change]
191199
from datetime import timedelta
192200

193201
from google.cloud import securitycenter
@@ -208,5 +216,5 @@ def group_assets_by_changes(organization_id):
208216
)
209217
for i, result in enumerate(result_iterator):
210218
print((i + 1), result)
211-
# [END group_all_assets_by_change]
219+
# [END securitycenter_group_all_assets_by_change]
212220
return i

0 commit comments

Comments
 (0)