Skip to content

Commit 898e7f3

Browse files
committed
Lint errors.
1 parent 748dcb1 commit 898e7f3

File tree

2 files changed

+27
-22
lines changed

2 files changed

+27
-22
lines changed

dlp/stored_info_types.py

Lines changed: 23 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,10 @@
2222

2323

2424
# [START dlp_create_stored_info_type]
25-
def create_stored_info_type_from_gcs_files(project, gcs_input_file_path,
26-
gcs_output_path, stored_info_type_id=None,
27-
display_name=None, description=None):
25+
def create_stored_info_type_from_gcs_files(
26+
project, gcs_input_file_path,
27+
gcs_output_path, stored_info_type_id=None,
28+
display_name=None, description=None):
2829
"""Creates a scheduled Data Loss Prevention API stored infoType from a set
2930
of GCS files.
3031
Args:
@@ -46,15 +47,15 @@ def create_stored_info_type_from_gcs_files(project, gcs_input_file_path,
4647
'output_path': {'path': gcs_output_path},
4748
'cloud_storage_file_set': {'url': gcs_input_file_path},
4849
}
49-
create_stored_info_type(project, dictionary_config,
50-
stored_info_type_id=stored_info_type_id,
50+
create_stored_info_type(
51+
project, dictionary_config, stored_info_type_id=stored_info_type_id,
5152
display_name=display_name, description=description)
5253

53-
def create_stored_info_type_from_bq_table(project, bq_input_project_id,
54-
bq_input_dataset_id, bq_input_table_id,
55-
bq_input_table_field, gcs_output_path,
56-
stored_info_type_id=None, display_name=None,
57-
description=None):
54+
55+
def create_stored_info_type_from_bq_table(
56+
project, bq_input_project_id, bq_input_dataset_id,
57+
bq_input_table_id, bq_input_table_field, gcs_output_path,
58+
stored_info_type_id=None, display_name=None, description=None):
5859
"""Creates a scheduled Data Loss Prevention API stored infoType from a
5960
column of a BigQuery.
6061
Args:
@@ -88,12 +89,13 @@ def create_stored_info_type_from_bq_table(project, bq_input_project_id,
8889
}
8990
}
9091
create_stored_info_type(project, dictionary_config,
91-
stored_info_type_id=stored_info_type_id,
92-
display_name=display_name, description=description)
92+
stored_info_type_id=stored_info_type_id,
93+
display_name=display_name, description=description)
94+
9395

9496
def create_stored_info_type(project, dictionary_config,
95-
stored_info_type_id=None, display_name=None,
96-
description=None):
97+
stored_info_type_id=None, display_name=None,
98+
description=None):
9799
"""Creates a scheduled Data Loss Prevention API stored infoType from a
98100
column of a BigQuery.
99101
Args:
@@ -220,8 +222,9 @@ def delete_stored_info_type(project, stored_info_type_id):
220222
dest='action', help='Select which action to perform.')
221223
subparsers.required = True
222224

223-
parser_create = subparsers.add_parser('create',
224-
help='Create a stored infoType.')
225+
parser_create = subparsers.add_parser(
226+
'create',
227+
help='Create a stored infoType.')
225228
parser_create.add_argument(
226229
'--gcs_input_file_path',
227230
help='GCS path of the input files containing the dictionary words.')
@@ -256,14 +259,16 @@ def delete_stored_info_type(project, stored_info_type_id):
256259
help='The Google Cloud project id to use as a parent resource.',
257260
default=default_project)
258261

259-
parser_list = subparsers.add_parser('list',
262+
parser_list = subparsers.add_parser(
263+
'list',
260264
help='List all stored infoTypes.')
261265
parser_list.add_argument(
262266
'--project',
263267
help='The Google Cloud project id to use as a parent resource.',
264268
default=default_project)
265269

266-
parser_delete = subparsers.add_parser('delete',
270+
parser_delete = subparsers.add_parser(
271+
'delete',
267272
help='Delete a stored infoType.')
268273
parser_delete.add_argument(
269274
'stored_info_type_id',

dlp/stored_info_types_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ def test_create_list_and_delete_stored_info_type(bucket, capsys):
7272

7373
except google.api_core.exceptions.InvalidArgument:
7474
# Stored infoType already exists, perhaps due to a previous test.
75-
stored_info_types.delete_stored_info_type(GCLOUD_PROJECT,
76-
TEST_STORED_INFO_TYPE_ID)
75+
stored_info_types.delete_stored_info_type(
76+
GCLOUD_PROJECT, TEST_STORED_INFO_TYPE_ID)
7777

7878
out, _ = capsys.readouterr()
7979
assert TEST_STORED_INFO_TYPE_ID in out
@@ -92,8 +92,8 @@ def test_create_list_and_delete_stored_info_type(bucket, capsys):
9292
out, _ = capsys.readouterr()
9393
assert TEST_STORED_INFO_TYPE_ID in out
9494

95-
stored_info_types.delete_stored_info_type(GCLOUD_PROJECT,
96-
TEST_STORED_INFO_TYPE_ID)
95+
stored_info_types.delete_stored_info_type(
96+
GCLOUD_PROJECT, TEST_STORED_INFO_TYPE_ID)
9797

9898
out, _ = capsys.readouterr()
9999
assert TEST_STORED_INFO_TYPE_ID in out

0 commit comments

Comments
 (0)