1919
2020def 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
4345def 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
6367def 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
9197def add_user_to_source (source_name ):
9298 """Gives a user findingsEditor permission to the source."""
9399 user_email = "csccclienttest@gmail.com"
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
131139def 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
150160def 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
199211def 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
254268def 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
300316def 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
329347def 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
375397def 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
399423def 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
429455def 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
463493def 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
482514def 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
506540def 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
535569def 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
569603def 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
0 commit comments